public async Task RegisterAsync(ConsulRegistration registration)
        {
            if (_logger.IsEnabled(LogLevel.Information))
            {
                _logger.LogInformation("Registering service with consul: " + registration.ServiceId);
            }

            try
            {
                await _client.Agent.ServiceRegister(registration.AgentServiceRegistration);

                if (_heartbeatOptions.Enable)
                {
                    _ttlScheduler?.Add(registration.InstanceId);
                }
            }
            catch (Exception e)
            {
                if (_consulDiscoveryOptions.FailFast)
                {
                    _logger.LogError(e, "Error registering service with consul: " + registration.ServiceId);
                    throw;
                }

                _logger.LogWarning(e, "Failfast is false. Error registering service with consul: " + registration.ServiceId);
            }
        }
        public void Add_Throws_Invalid_InstanceId()
        {
            var clientMoq = new Mock <IConsulClient>();
            var client    = clientMoq.Object;
            var opts      = new ConsulDiscoveryOptions();
            var sch       = new TtlScheduler(opts, client);

            Assert.Throws <ArgumentException>(() => sch.Add(string.Empty));
        }
        public void Add_DoesNothing_NoHeartbeatOptionsConfigured()
        {
            var clientMoq = new Mock <IConsulClient>();
            var client    = clientMoq.Object;
            var opts      = new ConsulDiscoveryOptions()
            {
                Heartbeat = null
            };
            var sch = new TtlScheduler(opts, client);

            sch.Add("foobar");
            Assert.Empty(sch._serviceHeartbeats);
        }
        public void Add_AddsTimer()
        {
            var clientMoq = new Mock <IConsulClient>();
            var agentMoq  = new Mock <IAgentEndpoint>();

            clientMoq.Setup(c => c.Agent).Returns(agentMoq.Object);
            var client = clientMoq.Object;
            var opts   = new ConsulDiscoveryOptions()
            {
                Heartbeat = new ConsulHeartbeatOptions()
            };
            var sch = new TtlScheduler(opts, client);

            sch.Add("foobar");
            Assert.NotEmpty(sch._serviceHeartbeats);
            Assert.True(sch._serviceHeartbeats.TryRemove("foobar", out Timer timer));
            Assert.NotNull(timer);
            timer.Dispose();
        }
        public void Timer_CallsPassTtl()
        {
            var clientMoq = new Mock <IConsulClient>();
            var agentMoq  = new Mock <IAgentEndpoint>();

            clientMoq.Setup(c => c.Agent).Returns(agentMoq.Object);
            var client = clientMoq.Object;
            var opts   = new ConsulDiscoveryOptions()
            {
                Heartbeat = new ConsulHeartbeatOptions()
                {
                    TtlValue = 2
                }
            };
            var sch = new TtlScheduler(opts, client);

            sch.Add("foobar");
            Thread.Sleep(5000);
            agentMoq.Verify(a => a.PassTTL("service:foobar", "ttl", default(CancellationToken)), Times.AtLeastOnce);
            sch.Remove("foobar");
        }