| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- import sys
- try:
- import unittest2 as unittest
- except ImportError:
- import unittest
- try:
- from unittest import mock
- except ImportError:
- import mock
- from qrcode.console_scripts import main
- def bad_read():
- raise UnicodeDecodeError('utf-8', b'0x80', 0, 1, 'invalid start byte')
- class ScriptTest(unittest.TestCase):
- @mock.patch('os.isatty', lambda *args: True)
- @mock.patch('qrcode.main.QRCode.print_ascii')
- def test_isatty(self, mock_print_ascii):
- main(['testtext'])
- mock_print_ascii.assert_called_with(tty=True)
- @mock.patch('os.isatty', lambda *args: False)
- @mock.patch('sys.stdout')
- def test_piped(self, mock_stdout):
- main(['testtext'])
- @mock.patch('os.isatty', lambda *args: True)
- @mock.patch('qrcode.main.QRCode.print_ascii')
- def test_stdin(self, mock_print_ascii):
- mock_stdin = mock.Mock(sys.stdin)
- stdin_buffer = getattr(mock_stdin, 'buffer', mock_stdin)
- stdin_buffer.read.return_value = 'testtext'
- with mock.patch('sys.stdin', mock_stdin):
- main([])
- self.assertTrue(stdin_buffer.read.called)
- mock_print_ascii.assert_called_with(tty=True)
- @unittest.skipIf(sys.version_info[0] < 3, 'Python 3')
- @mock.patch('os.isatty', lambda *args: True)
- @mock.patch('qrcode.main.QRCode.print_ascii')
- def test_stdin_py3_unicodedecodeerror(self, mock_print_ascii):
- mock_stdin = mock.Mock(sys.stdin)
- mock_stdin.buffer.read.return_value = 'testtext'
- mock_stdin.read.side_effect = bad_read
- with mock.patch('sys.stdin', mock_stdin):
- # sys.stdin.read() will raise an error...
- self.assertRaises(UnicodeDecodeError, sys.stdin.read)
- # ... but it won't be used now.
- main([])
- mock_print_ascii.assert_called_with(tty=True)
- @mock.patch('os.isatty', lambda *args: True)
- @mock.patch('qrcode.main.QRCode.print_ascii')
- def test_optimize(self, mock_print_ascii):
- main('testtext --optimize 0'.split())
- @mock.patch('sys.stdout')
- def test_factory(self, mock_stdout):
- main('testtext --factory svg'.split())
- @mock.patch('sys.stderr')
- def test_bad_factory(self, mock_stderr):
- self.assertRaises(SystemExit, main, 'testtext --factory fish'.split())
|
