introduce replica assignment change listener
diff --git a/testing/in_memory_env.go b/testing/in_memory_env.go
index ff6485f..e2aace3 100644
--- a/testing/in_memory_env.go
+++ b/testing/in_memory_env.go
@@ -50,7 +50,9 @@
return nil, err
}
server := grpc.NewServer()
- api.RegisterChunkStorageServer(server, chunk.NewChunkServer(&chunk.InMemoryChunkFactory{}))
+ api.RegisterChunkStorageServer(server, chunk.NewChunkServer(
+ &chunk.InMemoryChunkFactory{},
+ &chunk.NonChangingReplicaAssignment{}))
go server.Serve(lis)
env.c[i] = server
}