| Index: testing_support/auto_stub.py
|
| diff --git a/testing_support/auto_stub.py b/testing_support/auto_stub.py
|
| index e1155a64ae71f4ec5e0607ad95dd2eb352dd14c8..bcede8773a416835d1d8b5373201ac302af3a82e 100644
|
| --- a/testing_support/auto_stub.py
|
| +++ b/testing_support/auto_stub.py
|
| @@ -4,7 +4,8 @@
|
|
|
| import collections
|
| import inspect
|
| -import unittest
|
| +
|
| +from testing_support import thread_watcher
|
|
|
|
|
| class AutoStubMixIn(object):
|
| @@ -63,8 +64,8 @@ class SimpleMock(object):
|
| caller_name, ', '.join(filter(None, [str_args, str_kwargs]))))
|
|
|
|
|
| -class TestCase(unittest.TestCase, AutoStubMixIn):
|
| +class TestCase(thread_watcher.TestCase, AutoStubMixIn):
|
| """Adds self.mock() and self.has_failed() to a TestCase."""
|
| def tearDown(self):
|
| AutoStubMixIn.tearDown(self)
|
| - unittest.TestCase.tearDown(self)
|
| + super(TestCase, self).tearDown()
|
|
|