def test_get_status_code_v2(self, mock_get): mock_get.return_value.status_code = 200 prog = MySuperProgram() status = prog.get_current_status_code() print('status code is ' + str(status.status_code)) self.assertEqual(status.status_code, 200)
def test_get_current_ip(self, mock_get): prog = MySuperProgram() ip = prog.get_current_ip() assert ip == '1.1.1.1' expected_url = 'https://api.ipify.org/?format=json' mock_get.assert_called_once_with(expected_url)
def test_read_all_lines_with_mock(self): """ Check read_and_sort_all_strings() function. """ prog = MySuperProgram() all_strings = prog.read_and_sort_all_strings() print(all_strings) assert all_strings == ['test1\n', 'test2\n', 'test3\n']
def test_read_one_line_with_mock(self): """ Check read_string_from_file() function. """ prog = MySuperProgram() str = prog.read_string_from_file() msg = 'Expected: "{0}" Actual: "{1}"'.format('test1\n', str) assert str == 'test1\n', msg
def test_title_all_words_with_mock(self): """ Check title_all_words_in_line() function. """ prog = MySuperProgram() all_strings = prog.title_all_words_in_line() print(all_strings) assert all_strings == 'Test1 Test2 Test3'
def test_get_current_country(self, mock_get): prog = MySuperProgram() country = prog.get_current_country() print(dir(mock_get)) assert country == 'Russia' expected_url = 'https://api.ipify.org/?format=json' mock_get.assert_called_once_with(expected_url)
def test_get_status_code(self, mock_get): prog = MySuperProgram() status = prog.get_current_status_code() print('status code is ' + str(status.status_code)) assert status.status_code == 300