| Index: client/common_lib/base_barrier_unittest.py
|
| diff --git a/client/common_lib/base_barrier_unittest.py b/client/common_lib/base_barrier_unittest.py
|
| index 52d8e1734dae08eb36684aa9ba6bc48ad101ac75..71ea53842f958c489c9cff59852b2b186a73f6cc 100755
|
| --- a/client/common_lib/base_barrier_unittest.py
|
| +++ b/client/common_lib/base_barrier_unittest.py
|
| @@ -5,7 +5,7 @@ __author__ = """Ashwin Ganti (aganti@google.com)"""
|
| import os, sys, socket, errno, unittest, threading
|
| from time import time, sleep
|
| import common
|
| -from autotest_lib.client.common_lib import error, barrier
|
| +from autotest_lib.client.common_lib import error, barrier, base_barrier
|
| from autotest_lib.client.common_lib.test_utils import mock
|
|
|
|
|
| @@ -46,15 +46,14 @@ class barrier_test(unittest.TestCase):
|
|
|
|
|
| def test_get_host_from_id(self):
|
| - b = barrier.barrier('127.0.0.1#', 'testgethost', 100)
|
| -
|
| - hostname = b._get_host_from_id('my_host')
|
| + hostname = base_barrier.get_host_from_id('my_host')
|
| self.assertEqual(hostname, 'my_host')
|
|
|
| - hostname = b._get_host_from_id('my_host#')
|
| + hostname = base_barrier.get_host_from_id('my_host#')
|
| self.assertEqual(hostname, 'my_host')
|
|
|
| - self.assertRaises(error.BarrierError, b._get_host_from_id, '#my_host')
|
| + self.assertRaises(error.BarrierError,
|
| + base_barrier.get_host_from_id, '#my_host')
|
|
|
|
|
| def test_update_timeout(self):
|
|
|