OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 | 2 |
3 __author__ = """Ashwin Ganti (aganti@google.com)""" | 3 __author__ = """Ashwin Ganti (aganti@google.com)""" |
4 | 4 |
5 import os, sys, socket, errno, unittest, threading | 5 import os, sys, socket, errno, unittest, threading |
6 from time import time, sleep | 6 from time import time, sleep |
7 import common | 7 import common |
8 from autotest_lib.client.common_lib import error, barrier, base_barrier | 8 from autotest_lib.client.common_lib import error, barrier |
9 from autotest_lib.client.common_lib.test_utils import mock | 9 from autotest_lib.client.common_lib.test_utils import mock |
10 | 10 |
11 | 11 |
12 class listen_server_test(unittest.TestCase): | 12 class listen_server_test(unittest.TestCase): |
13 | 13 |
14 def test_init(self): | 14 def test_init(self): |
15 server = barrier.listen_server() | 15 server = barrier.listen_server() |
16 server.close() | 16 server.close() |
17 | 17 |
18 | 18 |
(...skipping 20 matching lines...) Expand all Loading... |
39 | 39 |
40 def test_initialize(self): | 40 def test_initialize(self): |
41 b = barrier.barrier('127.0.0.1#', 'testtag', 100, 11921) | 41 b = barrier.barrier('127.0.0.1#', 'testtag', 100, 11921) |
42 self.assertEqual(b._hostid, '127.0.0.1#') | 42 self.assertEqual(b._hostid, '127.0.0.1#') |
43 self.assertEqual(b._tag, 'testtag') | 43 self.assertEqual(b._tag, 'testtag') |
44 self.assertEqual(b._timeout_secs, 100) | 44 self.assertEqual(b._timeout_secs, 100) |
45 self.assertEqual(b._port, 11921) | 45 self.assertEqual(b._port, 11921) |
46 | 46 |
47 | 47 |
48 def test_get_host_from_id(self): | 48 def test_get_host_from_id(self): |
49 hostname = base_barrier.get_host_from_id('my_host') | 49 b = barrier.barrier('127.0.0.1#', 'testgethost', 100) |
| 50 |
| 51 hostname = b._get_host_from_id('my_host') |
50 self.assertEqual(hostname, 'my_host') | 52 self.assertEqual(hostname, 'my_host') |
51 | 53 |
52 hostname = base_barrier.get_host_from_id('my_host#') | 54 hostname = b._get_host_from_id('my_host#') |
53 self.assertEqual(hostname, 'my_host') | 55 self.assertEqual(hostname, 'my_host') |
54 | 56 |
55 self.assertRaises(error.BarrierError, | 57 self.assertRaises(error.BarrierError, b._get_host_from_id, '#my_host') |
56 base_barrier.get_host_from_id, '#my_host') | |
57 | 58 |
58 | 59 |
59 def test_update_timeout(self): | 60 def test_update_timeout(self): |
60 b = barrier.barrier('127.0.0.1#', 'update', 100) | 61 b = barrier.barrier('127.0.0.1#', 'update', 100) |
61 b._update_timeout(120) | 62 b._update_timeout(120) |
62 self.assertEqual(b._timeout_secs, 120) | 63 self.assertEqual(b._timeout_secs, 120) |
63 | 64 |
64 | 65 |
65 def test_remaining(self): | 66 def test_remaining(self): |
66 b = barrier.barrier('127.0.0.1#', 'remain', 100) | 67 b = barrier.barrier('127.0.0.1#', 'remain', 100) |
(...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
192 Test that reusing the same listen server object works. | 193 Test that reusing the same listen server object works. |
193 """ | 194 """ |
194 server = barrier.listen_server() | 195 server = barrier.listen_server() |
195 self.rendezvous_test(10, listen_server=server) | 196 self.rendezvous_test(10, listen_server=server) |
196 self.rendezvous_test(10, listen_server=server) | 197 self.rendezvous_test(10, listen_server=server) |
197 self.rendezvous_test(10, listen_server=server) | 198 self.rendezvous_test(10, listen_server=server) |
198 | 199 |
199 | 200 |
200 if __name__ == "__main__": | 201 if __name__ == "__main__": |
201 unittest.main() | 202 unittest.main() |
OLD | NEW |