def test_randomize(self):
uri = 'failover:tcp://remote1:61616,tcp://localhost:61616,tcp://127.0.0.1:61615,tcp://remote2:61616?priorityBackup=true,randomize=true,startupMaxReconnectAttempts=3'
protocol = StompFailoverTransport(uri)
localShuffled = remoteShuffled = 0
localHosts = ['localhost', '127.0.0.1']
remoteHosts = ['remote1', 'remote2']
while (localShuffled * remoteShuffled) == 0:
protocol = StompFailoverTransport(uri)
hosts = [broker['host'] for (broker, _) in itertools.islice(protocol, 4)]
self.assertEqual(set(hosts[:2]), set(localHosts))
if (hosts[:2] != localHosts):
localShuffled += 1
self.assertEqual(set(hosts[2:]), set(remoteHosts))
if (hosts[2:] != remoteHosts):
remoteShuffled += 1
def test_priority_backup_broken_localhost_lookup(self, mock_gethostbyname):
local_ip = '1.2.3.4'
uri = 'failover:tcp://remote1:61616,tcp://localhost:61616,tcp://127.0.0.1:61615,tcp://%s:61616?startupMaxReconnectAttempts=3,priorityBackup=true,randomize=false' % local_ip
protocol = StompFailoverTransport(uri)
def _broken_gethostbyname(host):
raise socket.gaierror()
mock_gethostbyname.side_effect = _broken_gethostbyname
self._test_failover(iter(protocol), [
(0,
Broker(**{
'host': 'localhost',
'protocol': 'tcp',
'port': 61616
})),
(0.01,
Broker(**{
'host': '127.0.0.1',
'protocol': 'tcp',
'port': 61615
})),
(0.02,
Broker(**{
'host': 'remote1',
'protocol': 'tcp',
'port': 61616
})),
(0.04,
Broker(**{
'host': local_ip,
'protocol': 'tcp',
'port': 61616
})),
])
def test_priority_backup(self):
uri = 'failover:tcp://remote1:61616,tcp://localhost:61616,tcp://127.0.0.1:61615,tcp://remote2:61616?startupMaxReconnectAttempts=3,priorityBackup=true,randomize=false'
protocol = StompFailoverTransport(uri)
self._test_failover(
iter(protocol),
[(0,
Broker(**{
'host': 'localhost',
'protocol': 'tcp',
'port': 61616
})),
(0.01,
Broker(**{
'host': '127.0.0.1',
'protocol': 'tcp',
'port': 61615
})),
(0.02,
Broker(**{
'host': 'remote1',
'protocol': 'tcp',
'port': 61616
})),
(0.04,
Broker(**{
'host': 'remote2',
'protocol': 'tcp',
'port': 61616
}))])
def test_jitter(self):
uri = 'failover:tcp://remote1:61616?useExponentialBackOff=false,startupMaxReconnectAttempts=1,reconnectDelayJitter=4'
for j in itertools.count():
protocol = iter(StompFailoverTransport(uri))
protocol.next()
_, delay = protocol.next()
self.assertTrue(abs(delay - 0.01) < 0.004)
if (j > 10) and (abs(delay - 0.01) > 0.003):
break
def test_time_scales_and_reconnect_attempts(self):
uri = 'failover:tcp://remote1:61615,tcp://localhost:61616,tcp://remote2:61617?randomize=false,startupMaxReconnectAttempts=3,initialReconnectDelay=7,backOffMultiplier=3.0,maxReconnectAttempts=1'
protocol = StompFailoverTransport(uri)
expectedDelaysAndBrokers = [
(0, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615}),
(0.007, {'host': 'localhost', 'protocol': 'tcp', 'port': 61616}),
(0.021, {'host': 'remote2', 'protocol': 'tcp', 'port': 61617}),
(0.063, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615})
]
self._test_failover(iter(protocol), expectedDelaysAndBrokers)
expectedDelaysAndBrokers = [
(0, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615}),
(0.007, {'host': 'localhost', 'protocol': 'tcp', 'port': 61616})
]
self._test_failover(iter(protocol), expectedDelaysAndBrokers)
uri = 'failover:(tcp://remote1:61615,tcp://localhost:61616)?randomize=false,startupMaxReconnectAttempts=3,initialReconnectDelay=7,maxReconnectDelay=8,maxReconnectAttempts=0'
protocol = StompFailoverTransport(uri)
expectedDelaysAndBrokers = [
(0, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615}),
(0.007, {'host': 'localhost', 'protocol': 'tcp', 'port': 61616}),
(0.008, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615}),
(0.008, {'host': 'localhost', 'protocol': 'tcp', 'port': 61616})
]
self._test_failover(iter(protocol), expectedDelaysAndBrokers)
expectedDelaysAndBrokers = [
(0, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615})
]
self._test_failover(iter(protocol), expectedDelaysAndBrokers)
uri = 'failover:(tcp://remote1:61615,tcp://localhost:61616)?randomize=false,startupMaxReconnectAttempts=2,initialReconnectDelay=3,useExponentialBackOff=false'
protocol = StompFailoverTransport(uri)
expectedDelaysAndBrokers = [
(0, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615}),
(0.003, {'host': 'localhost', 'protocol': 'tcp', 'port': 61616}),
(0.003, {'host': 'remote1', 'protocol': 'tcp', 'port': 61615})
]
self._test_failover(iter(protocol), expectedDelaysAndBrokers)
def test_priority_backup_localhost_lookup(self, mock_gethostbyname):
local_ip = '1.2.3.4'
uri = 'failover:tcp://remote1:61616,tcp://localhost:61616,tcp://127.0.0.1:61615,tcp://%s:61616?startupMaxReconnectAttempts=3,priorityBackup=true,randomize=false' % local_ip
protocol = StompFailoverTransport(uri)
mock_gethostbyname.side_effect = lambda *_args, **_kwargs: local_ip
self._test_failover(iter(protocol), [
(0, {'host': 'localhost', 'protocol': 'tcp', 'port': 61616}),
(0.01, {'host': '127.0.0.1', 'protocol': 'tcp', 'port': 61615}),
(0.02, {'host': local_ip, 'protocol': 'tcp', 'port': 61616}),
(0.04, {'host': 'remote1', 'protocol': 'tcp', 'port': 61616}),
])