| Index: telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py
|
| diff --git a/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py b/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py
|
| index 51ebe05d73521b02ce19cc8b6207ed8f73c4d8f9..7a17c870cd8a0a2e610f6a58003fead9e854a9d8 100644
|
| --- a/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py
|
| +++ b/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py
|
| @@ -4,10 +4,11 @@
|
|
|
| import unittest
|
|
|
| -from telemetry.core import exceptions
|
| from telemetry.internal import forwarders
|
| from telemetry.internal.forwarders import do_nothing_forwarder
|
|
|
| +import py_utils
|
| +
|
|
|
| class TestDoNothingForwarder(do_nothing_forwarder.DoNothingForwarder):
|
| """Override _WaitForConnect to avoid actual socket connection."""
|
| @@ -24,7 +25,7 @@ class TestErrorDoNothingForwarder(do_nothing_forwarder.DoNothingForwarder):
|
| """Simulate a connection error."""
|
|
|
| def _WaitForConnectionEstablished(self, address, timeout):
|
| - raise exceptions.TimeoutException
|
| + raise py_utils.TimeoutException
|
|
|
|
|
| class CheckPortPairsTest(unittest.TestCase):
|
|
|