def test_preserveDeadHostStats(self):
"""
"""
time = None
ht1 = HostTracking(self.group1)
s1 = schedulerFactory(ht1.group.lbType, ht1)
# let's cycle through all the hosts twice
fakeSenderFactories = [object() for x in xrange(8)]
sequence = [ht1.getHost(x) for x in fakeSenderFactories]
# check the stats
stats = ht1.getStats()
self.assertEquals(len(stats['bad'].keys()), 0)
self.assertEquals(sum(stats['openconns'].values()), 8)
self.assertEquals(sum(stats['totals'].values()), 0)
# finish them up
[ht1.doneHost(x) for x in fakeSenderFactories]
# check the clenaup stats
stats = ht1.getStats()
self.assertEquals(len(stats['bad'].keys()), 0)
self.assertEquals(sum(stats['openconns'].values()), 0)
self.assertEquals(sum(stats['totals'].values()), 8)
# now let's do a bad host
fakeSenderFactory = object()
host = ht1.getHost(fakeSenderFactory)
stats = ht1.getStats()
# make sure that the bad host is listed in the open connections
self.assertEquals(sum(stats['openconns'].values()), 1)
ht1.openconns[fakeSenderFactory] = (time, host)
ht1.deadHost(fakeSenderFactory, doLog=False)
stats = ht1.getStats()
# make sure that the previous stats for the bad host are not deleted
self.assertEquals(sum(stats['totals'].values()), 8)
def test_initialStats(self):
"""
We can't use self.ht1, because it might be contaminated by other tests.
"""
ht1 = HostTracking(self.group1)
s1 = schedulerFactory(ht1.group.lbType, ht1)
stats = ht1.getStats()
self.assertEquals(len(stats['bad'].keys()), 0)
self.assertEquals(sum(stats['openconns'].values()), 0)
self.assertEquals(sum(stats['totals'].values()), 0)
def setUp(self):
"""
"""
group1 = model.ProxyGroup('test_group1', roundr, True)
group1.addHost(model.ProxyHost('goodhost1', 'ip1', 1))
group1.addHost(model.ProxyHost('goodhost2', 'ip1', 2))
group1.addHost(model.ProxyHost('badhost1', 'ip1', 3))
group1.addHost(model.ProxyHost('badhost2', 'ip1', 4))
self.group1 = group1
group2 = model.ProxyGroup('test_group2', leastc, True)
group2.addHost(model.ProxyHost('goodhost3', 'ip1', 5))
group2.addHost(model.ProxyHost('goodhost4', 'ip1', 6))
group2.addHost(model.ProxyHost('badhost3', 'ip1', 7))
group2.addHost(model.ProxyHost('badhost4', 'ip1', 8))
self.group2 = group2
self.ht1 = HostTracking(self.group1)
self.ht2 = HostTracking(self.group2)
self.s1 = schedulerFactory(self.ht1.group.lbType, self.ht1)
self.s2 = schedulerFactory(self.ht2.group.lbType, self.ht2)
def test_deadHostStats(self):
"""
"""
badHost = ('ip1', 3)
time = None
fakeSenderFactory = object()
ht1 = HostTracking(self.group1)
s1 = schedulerFactory(ht1.group.lbType, ht1)
ht1.openconns[fakeSenderFactory] = (time, badHost)
ht1.deadHost(fakeSenderFactory, doLog=False)
stats = ht1.getStats()
self.assertEquals(len(stats['bad'].keys()), 1)
self.assertEquals(sum(stats['openconns'].values()), 0)
self.assertEquals(sum(stats['totals'].values()), 0)