Add integration test with DataStore