Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 1 | # Test the frozen module defined in frozen.c. |
| 2 | |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 3 | from test.support import captured_stdout, run_unittest |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 4 | import unittest |
Georg Brandl | 1b37e87 | 2010-03-14 10:45:50 +0000 | [diff] [blame] | 5 | import sys |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 6 | |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 7 | class FrozenTests(unittest.TestCase): |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 8 | |
| 9 | module_attrs = frozenset(['__builtins__', '__cached__', '__doc__', |
Eric V. Smith | 984b11f | 2012-05-24 20:21:04 -0400 | [diff] [blame] | 10 | '__loader__', '__name__', |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 11 | '__package__']) |
| 12 | package_attrs = frozenset(list(module_attrs) + ['__path__']) |
| 13 | |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 14 | def test_frozen(self): |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 15 | with captured_stdout() as stdout: |
| 16 | try: |
| 17 | import __hello__ |
| 18 | except ImportError as x: |
| 19 | self.fail("import __hello__ failed:" + str(x)) |
| 20 | self.assertEqual(__hello__.initialized, True) |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 21 | expect = set(self.module_attrs) |
| 22 | expect.add('initialized') |
| 23 | self.assertEqual(set(dir(__hello__)), expect) |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 24 | self.assertEqual(stdout.getvalue(), 'Hello world!\n') |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 25 | |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 26 | with captured_stdout() as stdout: |
| 27 | try: |
| 28 | import __phello__ |
| 29 | except ImportError as x: |
| 30 | self.fail("import __phello__ failed:" + str(x)) |
| 31 | self.assertEqual(__phello__.initialized, True) |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 32 | expect = set(self.package_attrs) |
| 33 | expect.add('initialized') |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 34 | if not "__phello__.spam" in sys.modules: |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 35 | self.assertEqual(set(dir(__phello__)), expect) |
Victor Stinner | 9cd6626 | 2011-05-16 16:32:33 +0200 | [diff] [blame] | 36 | else: |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 37 | expect.add('spam') |
| 38 | self.assertEqual(set(dir(__phello__)), expect) |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 39 | self.assertEqual(__phello__.__path__, [__phello__.__name__]) |
| 40 | self.assertEqual(stdout.getvalue(), 'Hello world!\n') |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 41 | |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 42 | with captured_stdout() as stdout: |
| 43 | try: |
| 44 | import __phello__.spam |
| 45 | except ImportError as x: |
| 46 | self.fail("import __phello__.spam failed:" + str(x)) |
| 47 | self.assertEqual(__phello__.spam.initialized, True) |
Brett Cannon | fd07415 | 2012-04-14 14:10:13 -0400 | [diff] [blame] | 48 | spam_expect = set(self.module_attrs) |
| 49 | spam_expect.add('initialized') |
| 50 | self.assertEqual(set(dir(__phello__.spam)), spam_expect) |
| 51 | phello_expect = set(self.package_attrs) |
| 52 | phello_expect.add('initialized') |
| 53 | phello_expect.add('spam') |
| 54 | self.assertEqual(set(dir(__phello__)), phello_expect) |
Victor Stinner | c77b931 | 2011-05-16 16:29:35 +0200 | [diff] [blame] | 55 | self.assertEqual(stdout.getvalue(), 'Hello world!\n') |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 56 | |
Christian Heimes | f19169f | 2007-11-12 19:19:07 +0000 | [diff] [blame] | 57 | try: |
| 58 | import __phello__.foo |
| 59 | except ImportError: |
| 60 | pass |
| 61 | else: |
| 62 | self.fail("import __phello__.foo should have failed") |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 63 | |
Ronald Oussoren | 94f2528 | 2010-05-05 19:11:21 +0000 | [diff] [blame] | 64 | try: |
| 65 | import __phello__.foo |
| 66 | except ImportError: |
| 67 | pass |
| 68 | else: |
| 69 | self.fail("import __phello__.foo should have failed") |
Christian Heimes | dae2a89 | 2008-04-19 00:55:37 +0000 | [diff] [blame] | 70 | |
| 71 | del sys.modules['__hello__'] |
| 72 | del sys.modules['__phello__'] |
| 73 | del sys.modules['__phello__.spam'] |
| 74 | |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 75 | def test_main(): |
| 76 | run_unittest(FrozenTests) |
Guido van Rossum | 3b7210d | 2007-10-15 00:25:56 +0000 | [diff] [blame] | 77 | |
| 78 | if __name__ == "__main__": |
| 79 | test_main() |