| Index: tools/telemetry/telemetry/core/extension_unittest.py
|
| diff --git a/tools/telemetry/telemetry/core/extension_unittest.py b/tools/telemetry/telemetry/core/extension_unittest.py
|
| index bc5fd78ded9ab7e69a92fe78338af99f014f7bf7..f17d029a0549294ad8de4326ce8f54263397759e 100644
|
| --- a/tools/telemetry/telemetry/core/extension_unittest.py
|
| +++ b/tools/telemetry/telemetry/core/extension_unittest.py
|
| @@ -11,7 +11,6 @@ import unittest
|
| from telemetry.core import browser_finder
|
| from telemetry.core import extension_to_load
|
| from telemetry.core import util
|
| -from telemetry.core.backends.chrome import extension_dict_backend
|
| from telemetry.unittest import options_for_unittests
|
|
|
|
|
| @@ -69,8 +68,7 @@ class NonExistentExtensionTest(unittest.TestCase):
|
| browser_to_create = browser_finder.FindBrowser(options)
|
| with browser_to_create.Create() as b:
|
| if b.supports_extensions:
|
| - self.assertRaises(extension_dict_backend.ExtensionNotFoundException,
|
| - lambda: b.extensions[load_extension])
|
| + self.assertRaises(KeyError, lambda: b.extensions[load_extension])
|
|
|
| class MultipleExtensionTest(unittest.TestCase):
|
| def setUp(self):
|
|
|