183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehimport test.test_support, unittest
383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehimport os
483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehclass CodingTest(unittest.TestCase):
683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_bad_coding(self):
783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        module_name = 'bad_coding'
883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        self.verify_bad_module(module_name)
983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
1083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_bad_coding2(self):
1183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        module_name = 'bad_coding2'
1283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        self.verify_bad_module(module_name)
1383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
1483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def verify_bad_module(self, module_name):
1583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        self.assertRaises(SyntaxError, __import__, 'test.' + module_name)
1683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
1783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        path = os.path.dirname(__file__)
1883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        filename = os.path.join(path, module_name + '.py')
1983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        with open(filename) as fp:
2083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            text = fp.read()
2183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        self.assertRaises(SyntaxError, compile, text, filename, 'exec')
2283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
2383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_error_from_string(self):
2483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        # See http://bugs.python.org/issue6289
2583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        input = u"# coding: ascii\n\N{SNOWMAN}".encode('utf-8')
2683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        with self.assertRaises(SyntaxError) as c:
2783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            compile(input, "<string>", "exec")
2883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        expected = "'ascii' codec can't decode byte 0xe2 in position 16: " \
2983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                   "ordinal not in range(128)"
3083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        self.assertTrue(c.exception.args[0].startswith(expected))
3183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
3283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
3383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehdef test_main():
3483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    test.test_support.run_unittest(CodingTest)
3583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
3683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehif __name__ == "__main__":
3783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    test_main()
38