| 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 |