depot_tools/tests/fix_encoding_test.py

#!/usr/bin/env python3
# coding=utf-8
# Copyright (c) 2011 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Unit tests for fix_encoding.py."""

import os
import sys
import unittest

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

import fix_encoding


class FixEncodingTest(unittest.TestCase):
    # Nice mix of latin, hebrew, arabic and chinese. Doesn't mean anything.
    text = u'Héllô 偉大 سيد'

    def test_code_page(self):
        # Make sure printing garbage won't throw.
        print(self.text.encode() + b'\xff')
        print(self.text.encode() + b'\xff', file=sys.stderr)

    def test_utf8(self):
        # Make sure printing utf-8 works.
        print(self.text.encode('utf-8'))
        print(self.text.encode('utf-8'), file=sys.stderr)

    @unittest.skipIf(os.name == 'nt', 'Does not work on Windows')
    def test_unicode(self):
        # Make sure printing unicode works.
        print(self.text)
        print(self.text, file=sys.stderr)

    @unittest.skipIf(os.name == 'nt', 'Does not work on Windows')
    def test_default_encoding(self):
        self.assertEqual('utf-8', sys.getdefaultencoding())

    def test_win_console(self):
        if sys.platform != 'win32':
            return
        # This should fail if not redirected, e.g. run directly instead of
        # through the presubmit check. Can be checked with: python
        # tests\fix_encoding_test.py
        self.assertEqual(sys.stdout.__class__, fix_encoding.WinUnicodeOutput)
        self.assertEqual(sys.stderr.__class__, fix_encoding.WinUnicodeOutput)
        self.assertEqual(sys.stdout.encoding, sys.getdefaultencoding())
        self.assertEqual(sys.stderr.encoding, sys.getdefaultencoding())

    def test_multiple_calls(self):
        # Shouldn't do anything.
        self.assertEqual(False, fix_encoding.fix_encoding())


if __name__ == '__main__':
    fix_encoding.fix_encoding()
    unittest.main()