public async Task Run_Disconnects_When_Monitor_Returns()
{
var timeout = TimeSpan.FromSeconds(2);
var grpcHost = await CreateGrpcServer();
var grpcClient = grpcHost.CreateFixedClient();
var stopTokenSource = new CancellationTokenSource(timeout);
var connectionWasLost = false;
var eventBus = new EventBus();
eventBus.Subscribe <EndpointConnectionEvents.Disconnected>(args =>
{
connectionWasLost = true;
stopTokenSource.Cancel();
});
var endpointConnectionManager = new EndpointConnectionManager(
eventBus, new DelegateClientFactory(endpoint => grpcClient),
new AlwaysYesServerVerifier(), _loggerFactory.CreateLogger <EndpointConnectionManager>()
);
await endpointConnectionManager.Run(
new[] { StaticApiServer.AnyOnUri("http://localhost/") },
new[] { new DisconnectsImmediately() },
stopTokenSource.Token
);
await grpcHost.StopAsync();
Assert.IsTrue(connectionWasLost);
}
public async Task Run_Reconnects_Immediately_After_Disconnect()
{
var timeout = TimeSpan.FromSeconds(2);
var grpcHost = await CreateGrpcServer();
var grpcClient = grpcHost.CreateFixedClient();
var stopTokenSource = new CancellationTokenSource(timeout);
var connectionCount = 0;
var eventBus = new EventBus();
eventBus.Subscribe <EndpointConnectionEvents.Connected>(args =>
{
connectionCount++;
if (connectionCount == 2)
{
stopTokenSource.Cancel();
}
});
var endpointConnectionManager = new EndpointConnectionManager(
eventBus, new DelegateClientFactory(endpoint => grpcClient),
new AlwaysYesServerVerifier(), _loggerFactory.CreateLogger <EndpointConnectionManager>()
);
await endpointConnectionManager.Run(
new[] { StaticApiServer.AnyOnUri("http://localhost/") },
new[] { new DisconnectsImmediately() },
stopTokenSource.Token
);
await grpcHost.StopAsync();
Assert.AreEqual(2, connectionCount);
}
public async Task Run_Restarts_Endpoint_Provider_After_Throwing_Exception()
{
var timeout = TimeSpan.FromSeconds(2);
var grpcHost = await CreateGrpcServer();
var grpcClient = grpcHost.CreateFixedClient();
var stopTokenSource = new CancellationTokenSource(timeout);
var connectionWasEstablished = false;
var eventBus = new EventBus();
eventBus.Subscribe <EndpointConnectionEvents.Connected>(args =>
{
connectionWasEstablished = true;
stopTokenSource.Cancel();
});
var endpointConnectionManager = new EndpointConnectionManager(
eventBus, new DelegateClientFactory(endpoint => grpcClient),
new AlwaysYesServerVerifier(), _loggerFactory.CreateLogger <EndpointConnectionManager>()
);
await endpointConnectionManager.Run(
new IServerEndpointProvider[]
{
new ThrowsExceptionFirstEndpointProvider(
StaticApiServer.AnyOnUri("http://localhost/"))
},
new[] { new NeverDisconnects() },
stopTokenSource.Token
);
await grpcHost.StopAsync();
Assert.IsTrue(connectionWasEstablished);
}
public async Task Run_Connects_After_Cooldown_If_Connection_Fails()
{
var timeout = TimeSpan.FromSeconds(7);
var grpcHost = await CreateGrpcServer();
var grpcClient = grpcHost.CreateFixedClient();
var stopTokenSource = new CancellationTokenSource(timeout);
var verifier = new VerifiesOnSecondAttemptVerifier();
var connectionWasEstablished = false;
var eventBus = new EventBus();
eventBus.Subscribe <EndpointConnectionEvents.Connected>(args =>
{
connectionWasEstablished = true;
stopTokenSource.Cancel();
});
var endpointConnectionManager = new EndpointConnectionManager(
eventBus, new DelegateClientFactory(endpoint => grpcClient),
verifier, _loggerFactory.CreateLogger <EndpointConnectionManager>()
);
await endpointConnectionManager.Run(
new[] { StaticApiServer.AnyOnUri("http://localhost/") },
new[] { new DisconnectsImmediately() },
stopTokenSource.Token
);
await grpcHost.StopAsync();
Assert.IsTrue(connectionWasEstablished);
// built-in cooldown is 5 seconds, make the check a little fuzzy
Assert.IsTrue(verifier.LastAttemptDelta.Value > TimeSpan.FromSeconds(4));
}
public HostServer(Host host,
EndpointConnectionManager connectionManager, EventBus eventBus,
ILogger <HostServer> logger,
HostManager hostManager,
ResourceStateManager resourceStateManager) :
base(connectionManager, eventBus, logger)
{
Host = host;
_hostManager = hostManager;
_resourceStateManager = resourceStateManager;
}
public EndpointConnectionRunner(
HostServer host,
EndpointConnectionManager connectionManager,
ILogger <ConnectionManager> logger,
EventBus eventBus,
ILoggerFactory loggerFactory)
{
HostServer = host;
_connectionManager = connectionManager;
_logger = logger;
_eventBus = eventBus;
_loggerFactory = loggerFactory;
EndpointProvider = new HostServerEndpointProvider(new Uri(Host.Endpoint));
}
public HomeCtlHostService(
EventBus eventBus,
ApiServer apiServer,
EndpointConnectionManager connectionManager,
IEnumerable <IServerEndpointProvider> serverEndpointProviders,
IEnumerable <IServerLivelinessMonitor> livelinessMonitors,
IEnumerable <IDeviceProvider> deviceProviders,
IEnumerable <StartupService> startupServices // ensure startup services are constructed before this service runs
)
{
_eventBus = eventBus;
_apiServer = apiServer;
_connectionManager = connectionManager;
_serverEndpointProviders = serverEndpointProviders;
_livelinessMonitors = livelinessMonitors;
_deviceProviders = deviceProviders;
SubscribeToEvents();
}
public async Task Run_Persists_Endpoint_Providers_Across_Failover()
{
var timeout = TimeSpan.FromSeconds(2);
var grpcHost = await CreateGrpcServer();
var grpcClient = grpcHost.CreateFixedClient();
var stopTokenSource = new CancellationTokenSource(timeout);
var verifier = new VerifiesOnSecondAttemptVerifier();
var connectionWasEstablished = false;
var connectedEndpoint = default(ServerEndpoint);
var eventBus = new EventBus();
eventBus.Subscribe <EndpointConnectionEvents.Connected>(args =>
{
connectedEndpoint = args.ServerEndpoint;
connectionWasEstablished = true;
stopTokenSource.Cancel();
});
var countingEndpointProvider = new CountingProxyEndpointProvider(
StaticApiServer.AnyOnUri("http://localhost/"));
var endpointConnectionManager = new EndpointConnectionManager(
eventBus, new DelegateClientFactory(endpoint => grpcClient),
verifier, _loggerFactory.CreateLogger <EndpointConnectionManager>()
);
await endpointConnectionManager.Run(
new IServerEndpointProvider[]
{
StaticApiServer.AnyOnUri("http://localhost/"),
countingEndpointProvider
},
new[] { new DisconnectsImmediately() },
stopTokenSource.Token
);
await grpcHost.StopAsync();
Assert.IsTrue(connectionWasEstablished);
Assert.AreEqual(1, countingEndpointProvider.CallCount);
}
public async Task Run_Uses_2nd_Endpoint_Immediately_When_1st_Throws_Exception()
{
var timeout = TimeSpan.FromSeconds(2);
var grpcHost = await CreateGrpcServer();
var grpcClient = grpcHost.CreateFixedClient();
var stopTokenSource = new CancellationTokenSource(timeout);
var verifier = new VerifiesOnSecondAttemptVerifier();
var connectionWasEstablished = false;
var connectedEndpoint = default(ServerEndpoint);
var eventBus = new EventBus();
eventBus.Subscribe <EndpointConnectionEvents.Connected>(args =>
{
connectedEndpoint = args.ServerEndpoint;
connectionWasEstablished = true;
stopTokenSource.Cancel();
});
var endpointConnectionManager = new EndpointConnectionManager(
eventBus, new DelegateClientFactory(endpoint => grpcClient),
verifier, _loggerFactory.CreateLogger <EndpointConnectionManager>()
);
await endpointConnectionManager.Run(
new IServerEndpointProvider[]
{
new ThrowsExceptionFirstEndpointProvider(StaticApiServer.AnyOnUri("http://localhost-first/")),
StaticApiServer.AnyOnUri("http://localhost-second/")
},
new[] { new NeverDisconnects() },
stopTokenSource.Token
);
await grpcHost.StopAsync();
Assert.IsTrue(connectionWasEstablished);
Assert.AreEqual("http://localhost-second/", connectedEndpoint.Uri.OriginalString);
}
public Disconnected(EndpointConnectionManager endpointConnectionManager,
ServerEndpoint serverEndpoint)
{
EndpointConnectionManager = endpointConnectionManager;
ServerEndpoint = serverEndpoint;
}
