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