| 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, base_barrier | 8 from autotest_lib.client.common_lib import error, base_barrier |
| 9 barrier = base_barrier | 9 barrier = base_barrier |
| 10 from autotest_lib.client.common_lib.test_utils import mock | 10 from autotest_lib.client.common_lib.test_utils import mock |
| (...skipping 29 matching lines...) Expand all Loading... |
| 40 | 40 |
| 41 def test_initialize(self): | 41 def test_initialize(self): |
| 42 b = barrier.barrier('127.0.0.1#', 'testtag', 100, 11921) | 42 b = barrier.barrier('127.0.0.1#', 'testtag', 100, 11921) |
| 43 self.assertEqual(b._hostid, '127.0.0.1#') | 43 self.assertEqual(b._hostid, '127.0.0.1#') |
| 44 self.assertEqual(b._tag, 'testtag') | 44 self.assertEqual(b._tag, 'testtag') |
| 45 self.assertEqual(b._timeout_secs, 100) | 45 self.assertEqual(b._timeout_secs, 100) |
| 46 self.assertEqual(b._port, 11921) | 46 self.assertEqual(b._port, 11921) |
| 47 | 47 |
| 48 | 48 |
| 49 def test_get_host_from_id(self): | 49 def test_get_host_from_id(self): |
| 50 b = barrier.barrier('127.0.0.1#', 'testgethost', 100) | 50 hostname = barrier.get_host_from_id('my_host') |
| 51 | |
| 52 hostname = b._get_host_from_id('my_host') | |
| 53 self.assertEqual(hostname, 'my_host') | 51 self.assertEqual(hostname, 'my_host') |
| 54 | 52 |
| 55 hostname = b._get_host_from_id('my_host#') | 53 hostname = barrier.get_host_from_id('my_host#') |
| 56 self.assertEqual(hostname, 'my_host') | 54 self.assertEqual(hostname, 'my_host') |
| 57 | 55 |
| 58 self.assertRaises(error.BarrierError, b._get_host_from_id, '#my_host') | 56 self.assertRaises(error.BarrierError, barrier.get_host_from_id, '#my_hos
t') |
| 59 | 57 |
| 60 | 58 |
| 61 def test_update_timeout(self): | 59 def test_update_timeout(self): |
| 62 b = barrier.barrier('127.0.0.1#', 'update', 100) | 60 b = barrier.barrier('127.0.0.1#', 'update', 100) |
| 63 b._update_timeout(120) | 61 b._update_timeout(120) |
| 64 self.assertEqual(b._timeout_secs, 120) | 62 self.assertEqual(b._timeout_secs, 120) |
| 65 | 63 |
| 66 | 64 |
| 67 def test_remaining(self): | 65 def test_remaining(self): |
| 68 b = barrier.barrier('127.0.0.1#', 'remain', 100) | 66 b = barrier.barrier('127.0.0.1#', 'remain', 100) |
| (...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 194 Test that reusing the same listen server object works. | 192 Test that reusing the same listen server object works. |
| 195 """ | 193 """ |
| 196 server = barrier.listen_server() | 194 server = barrier.listen_server() |
| 197 self.rendezvous_test(10, listen_server=server) | 195 self.rendezvous_test(10, listen_server=server) |
| 198 self.rendezvous_test(10, listen_server=server) | 196 self.rendezvous_test(10, listen_server=server) |
| 199 self.rendezvous_test(10, listen_server=server) | 197 self.rendezvous_test(10, listen_server=server) |
| 200 | 198 |
| 201 | 199 |
| 202 if __name__ == "__main__": | 200 if __name__ == "__main__": |
| 203 unittest.main() | 201 unittest.main() |
| OLD | NEW |