mirror of
https://github.com/GAM-team/GAM.git
synced 2026-06-29 18:31:38 +00:00
a0c410be0e
* Begin breaking apart gam.py into logical pieces Start with one of the deepest parts of the stack, Google API request execution calls and associated errors. Critical information printing functions and application control logic are also broken out into their own components. This change also adds unit tests for migrated content and makes code more PEP8 compliant. This commit starts work on jay0lee/GAM#147 * Add unit tests to Travis config * Swap assert_called_once() with assertEqual() and Mock.call_count Makes tests compatible with Python 3.5. assert_called_once() is only available in Python 3.6+
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
"""Tests for display."""
|
|
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
import display
|
|
from var import ERROR_PREFIX
|
|
from var import WARNING_PREFIX
|
|
|
|
|
|
class DisplayTest(unittest.TestCase):
|
|
|
|
@patch.object(display.sys.stderr, 'write')
|
|
def test_print_error_prints_to_stderr(self, mock_write):
|
|
message = 'test error'
|
|
display.print_error(message)
|
|
printed_message = mock_write.call_args[0][0]
|
|
self.assertIn(message, printed_message)
|
|
|
|
@patch.object(display.sys.stderr, 'write')
|
|
def test_print_error_prints_error_prefix(self, mock_write):
|
|
message = 'test error'
|
|
display.print_error(message)
|
|
printed_message = mock_write.call_args[0][0]
|
|
self.assertLess(
|
|
printed_message.find(ERROR_PREFIX), printed_message.find(message),
|
|
'The error prefix does not appear before the error message')
|
|
|
|
@patch.object(display.sys.stderr, 'write')
|
|
def test_print_error_ends_message_with_newline(self, mock_write):
|
|
message = 'test error'
|
|
display.print_error(message)
|
|
printed_message = mock_write.call_args[0][0]
|
|
self.assertRegex(printed_message, '\n$',
|
|
'The error message does not end in a newline.')
|
|
|
|
@patch.object(display.sys.stderr, 'write')
|
|
def test_print_warning_prints_to_stderr(self, mock_write):
|
|
message = 'test warning'
|
|
display.print_warning(message)
|
|
printed_message = mock_write.call_args[0][0]
|
|
self.assertIn(message, printed_message)
|
|
|
|
@patch.object(display.sys.stderr, 'write')
|
|
def test_print_warning_prints_error_prefix(self, mock_write):
|
|
message = 'test warning'
|
|
display.print_error(message)
|
|
printed_message = mock_write.call_args[0][0]
|
|
self.assertLess(
|
|
printed_message.find(WARNING_PREFIX), printed_message.find(message),
|
|
'The warning prefix does not appear before the error message')
|
|
|
|
@patch.object(display.sys.stderr, 'write')
|
|
def test_print_warning_ends_message_with_newline(self, mock_write):
|
|
message = 'test warning'
|
|
display.print_error(message)
|
|
printed_message = mock_write.call_args[0][0]
|
|
self.assertRegex(printed_message, '\n$',
|
|
'The warning message does not end in a newline.')
|