diff --git a/tests/master_test.py b/tests/master_test.py index 2bea6193..deb9e93a 100644 --- a/tests/master_test.py +++ b/tests/master_test.py @@ -7,6 +7,20 @@ import testlib import mitogen.master +class ScanCodeImportsTest(unittest.TestCase): + func = staticmethod(mitogen.master.scan_code_imports) + + def test_simple(self): + co = compile(open(__file__).read(), __file__, 'exec') + self.assertEquals(list(self.func(co)), [ + (-1, 'subprocess', ()), + (-1, 'time', ()), + (-1, 'unittest', ()), + (-1, 'testlib', ()), + (-1, 'mitogen.master', ()), + ]) + + class IterReadTest(unittest.TestCase): func = staticmethod(mitogen.master.iter_read) @@ -96,3 +110,7 @@ class WriteAllTest(unittest.TestCase): )) finally: proc.terminate() + + +if __name__ == '__main__': + unittest.main()