diff --git a/tests/channel_test.py b/tests/channel_test.py index 0284449c..fbb30345 100644 --- a/tests/channel_test.py +++ b/tests/channel_test.py @@ -4,7 +4,7 @@ import mitogen.core import testlib -class ConstructorTest(testlib.RouterMixin, unittest.TestCase): +class ConstructorTest(testlib.RouterMixin, testlib.TestCase): def test_constructor(self): # issue 32 l1 = self.router.local() @@ -14,3 +14,7 @@ class ConstructorTest(testlib.RouterMixin, unittest.TestCase): assert chan.dst_handle == 123 assert chan.handle is not None assert chan.handle > 0 + + +if __name__ == '__main__': + unittest.main() diff --git a/tests/id_allocation_test.py b/tests/id_allocation_test.py index 78f66a26..a65f56a4 100644 --- a/tests/id_allocation_test.py +++ b/tests/id_allocation_test.py @@ -5,8 +5,12 @@ import testlib import id_allocation -class SlaveTest(testlib.RouterMixin, unittest.TestCase): +class SlaveTest(testlib.RouterMixin, testlib.TestCase): def test_slave_allocates_id(self): context = self.router.local() id_ = context.call(id_allocation.allocate_an_id) assert id_ == (self.router.id_allocator.next_id - 1) + + +if __name__ == '__main__': + unittest.main() diff --git a/tests/importer_test.py b/tests/importer_test.py index 8d940259..3a411c06 100644 --- a/tests/importer_test.py +++ b/tests/importer_test.py @@ -18,14 +18,14 @@ class ImporterMixin(testlib.RouterMixin): def setUp(self): super(ImporterMixin, self).setUp() self.context = mock.Mock() - self.importer = mitogen.core.Importer(self.router, self.context, '') + self.importer = mitogen.core.Importer(self.context, '') def tearDown(self): sys.modules.pop(self.modname, None) super(ImporterMixin, self).tearDown() -class LoadModuleTest(ImporterMixin, unittest.TestCase): +class LoadModuleTest(ImporterMixin, testlib.TestCase): data = zlib.compress("data = 1\n\n") path = 'fake_module.py' modname = 'fake_module' @@ -58,7 +58,7 @@ class LoadModuleTest(ImporterMixin, unittest.TestCase): self.assertTrue(mod.__package__ is None) -class LoadSubmoduleTest(ImporterMixin, unittest.TestCase): +class LoadSubmoduleTest(ImporterMixin, testlib.TestCase): data = zlib.compress("data = 1\n\n") path = 'fake_module.py' modname = 'mypkg.fake_module' @@ -70,7 +70,7 @@ class LoadSubmoduleTest(ImporterMixin, unittest.TestCase): self.assertEquals(mod.__package__, 'mypkg') -class LoadModulePackageTest(ImporterMixin, unittest.TestCase): +class LoadModulePackageTest(ImporterMixin, testlib.TestCase): data = zlib.compress("func = lambda: 1\n\n") path = 'fake_pkg/__init__.py' modname = 'fake_pkg' @@ -115,7 +115,7 @@ class LoadModulePackageTest(ImporterMixin, unittest.TestCase): self.assertEquals(mod.func.__module__, self.modname) -class EmailParseAddrSysTest(testlib.RouterMixin, unittest.TestCase): +class EmailParseAddrSysTest(testlib.RouterMixin, testlib.TestCase): @pytest.fixture(autouse=True) def initdir(self, caplog): self.caplog = caplog @@ -125,3 +125,7 @@ class EmailParseAddrSysTest(testlib.RouterMixin, unittest.TestCase): # while executing email.utils.parseaddr(). Ensure this needless # roundtrip has not reappeared. pass + + +if __name__ == '__main__': + unittest.main()