1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
from unittest import TestCase
from kazoo.hosts import collect_hosts
class HostsTestCase(TestCase):
def test_ipv4(self):
hosts, chroot = collect_hosts('127.0.0.1:2181, 192.168.1.2:2181, \
132.254.111.10:2181')
self.assertEquals([('127.0.0.1', 2181),
('192.168.1.2', 2181),
('132.254.111.10', 2181)], hosts)
self.assertEquals(None, chroot)
hosts, chroot = collect_hosts(['127.0.0.1:2181',
'192.168.1.2:2181',
'132.254.111.10:2181'])
self.assertEquals([('127.0.0.1', 2181),
('192.168.1.2', 2181),
('132.254.111.10', 2181)], hosts)
self.assertEquals(None, chroot)
def test_ipv6(self):
hosts, chroot = collect_hosts('[fe80::200:5aee:feaa:20a2]:2181')
self.assertEquals([('fe80::200:5aee:feaa:20a2', 2181)], hosts)
self.assertEquals(None, chroot)
hosts, chroot = collect_hosts(['[fe80::200:5aee:feaa:20a2]:2181'])
self.assertEquals([('fe80::200:5aee:feaa:20a2', 2181)], hosts)
self.assertEquals(None, chroot)
def test_hosts_list(self):
hosts, chroot = collect_hosts('zk01:2181, zk02:2181, zk03:2181')
expected1 = [('zk01', 2181), ('zk02', 2181), ('zk03', 2181)]
self.assertEquals(expected1, hosts)
self.assertEquals(None, chroot)
hosts, chroot = collect_hosts(['zk01:2181', 'zk02:2181', 'zk03:2181'])
self.assertEquals(expected1, hosts)
self.assertEquals(None, chroot)
expected2 = '/test'
hosts, chroot = collect_hosts('zk01:2181, zk02:2181, zk03:2181/test')
self.assertEquals(expected1, hosts)
self.assertEquals(expected2, chroot)
hosts, chroot = collect_hosts(['zk01:2181',
'zk02:2181',
'zk03:2181', '/test'])
self.assertEquals(expected1, hosts)
self.assertEquals(expected2, chroot)
|