diff --git a/Vostok.ServiceDiscovery.Tests/ServiceLocatorStorage/EnvironmentsStorage_Tests.cs b/Vostok.ServiceDiscovery.Tests/ServiceLocatorStorage/EnvironmentsStorage_Tests.cs index 2945e43..c39c9d4 100644 --- a/Vostok.ServiceDiscovery.Tests/ServiceLocatorStorage/EnvironmentsStorage_Tests.cs +++ b/Vostok.ServiceDiscovery.Tests/ServiceLocatorStorage/EnvironmentsStorage_Tests.cs @@ -191,6 +191,8 @@ public void Should_not_delete_environment_from_cache_when_observation_of_deleted ShouldReturn(storage, "default", expectedInfo); Ensemble.Start(); + storage.UpdateAll(); + ShouldReturn(storage, "default", expectedInfo); } } diff --git a/Vostok.ServiceDiscovery.Tests/TestsBase.cs b/Vostok.ServiceDiscovery.Tests/TestsBase.cs index 7b36691..2365e17 100644 --- a/Vostok.ServiceDiscovery.Tests/TestsBase.cs +++ b/Vostok.ServiceDiscovery.Tests/TestsBase.cs @@ -29,7 +29,6 @@ internal abstract class TestsBase protected ZooKeeperEnsemble Ensemble; protected ZooKeeperClient ZooKeeperClient; protected IServiceDiscoveryManager ServiceDiscoveryManager; - private static readonly TimeSpan DefaultZkClientTimeout = 8.Seconds(); [OneTimeSetUp] public void OneTimeSetUp() @@ -65,7 +64,7 @@ protected Task KillSession(ZooKeeperClient client) => protected ZooKeeperClient GetZooKeeperClient() { - var settings = new ZooKeeperClientSettings(Ensemble.ConnectionString) {Timeout = DefaultZkClientTimeout}; + var settings = new ZooKeeperClientSettings(Ensemble.ConnectionString) {Timeout = DefaultTimeout}; return new ZooKeeperClient(settings, Log); }