improve tests for --today argument

This commit is contained in:
Alan Hamlett 2019-05-07 18:11:02 -07:00
parent 28d736efe5
commit e673dc82f9
3 changed files with 131 additions and 4 deletions

View file

@ -323,7 +323,7 @@ class ArgumentsTestCase(TestCase):
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_missing_entity_argument_with_show_time_today_arg(self, logs):
def test_missing_entity_argument_with_today_arg(self, logs):
logging.disable(logging.NOTSET)
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = SummaryResponse()

View file

@ -232,6 +232,135 @@ class MainTestCase(utils.TestCase):
self.assertOfflineHeartbeatsNotSynced()
self.assertSessionCacheDeleted()
@log_capture()
def test_requests_exception_with_today_arg(self, logs):
logging.disable(logging.NOTSET)
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].side_effect = RequestException('requests exception')
key = str(uuid.uuid4())
args = ['--today', '--key', key]
retval = execute(args)
self.assertEquals(retval, API_ERROR)
self.assertNothingPrinted()
self.assertNothingLogged(logs)
self.assertHeartbeatNotSavedOffline()
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_requests_exception_with_today_arg_verbose(self, logs):
logging.disable(logging.NOTSET)
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].side_effect = RequestException('requests exception')
key = str(uuid.uuid4())
args = ['--today', '--key', key, '--verbose']
retval = execute(args)
self.assertEquals(retval, API_ERROR)
expected = 'RequestException: requests exception\n'
actual = self.getPrintedOutput()
self.assertEquals(actual, expected)
log_output = u("\n").join([u(' ').join(x) for x in logs.actual()])
expected = "'RequestException': u'requests exception'"
if is_py3:
expected = "'RequestException': 'requests exception'"
self.assertIn(expected, log_output)
self.assertHeartbeatNotSavedOffline()
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_generic_exception_with_today_arg(self, logs):
logging.disable(logging.NOTSET)
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].side_effect = Exception('generic exception')
key = str(uuid.uuid4())
args = ['--today', '--key', key]
retval = execute(args)
self.assertEquals(retval, API_ERROR)
self.assertNothingPrinted()
self.assertNothingLogged(logs)
self.assertHeartbeatNotSavedOffline()
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_generic_exception_with_today_arg_verbose(self, logs):
logging.disable(logging.NOTSET)
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].side_effect = Exception('generic exception')
key = str(uuid.uuid4())
args = ['--today', '--key', key, '--verbose']
retval = execute(args)
self.assertEquals(retval, API_ERROR)
expected = 'Exception: generic exception\n'
actual = self.getPrintedOutput()
self.assertEquals(actual, expected)
log_output = u("\n").join([u(' ').join(x) for x in logs.actual()])
expected = "'Exception': u'generic exception'"
if is_py3:
expected = "'Exception': 'generic exception'"
self.assertIn(expected, log_output)
self.assertHeartbeatNotSavedOffline()
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_error_response_with_today_arg(self, logs):
logging.disable(logging.NOTSET)
response = CustomResponse()
response.response_code = 400
response.response_text = 'error response text'
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = response
key = str(uuid.uuid4())
args = ['--today', '--key', key]
retval = execute(args)
self.assertEquals(retval, API_ERROR)
self.assertNothingPrinted()
self.assertNothingLogged(logs)
self.assertHeartbeatNotSavedOffline()
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_error_response_with_today_arg_verbose(self, logs):
logging.disable(logging.NOTSET)
response = CustomResponse()
response.response_code = 400
response.response_text = 'error response text'
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = response
key = str(uuid.uuid4())
args = ['--today', '--key', key, '--verbose']
retval = execute(args)
self.assertEquals(retval, API_ERROR)
expected = 'Error: 400\n'
actual = self.getPrintedOutput()
self.assertEquals(actual, expected)
log_output = u("\n").join([u(' ').join(x) for x in logs.actual()])
expected = "WakaTime DEBUG {u'response_code': 400, u'response_text': u'error response text'}"
if is_py3:
expected = "WakaTime DEBUG {'response_code': 400, 'response_text': 'error response text'}"
self.assertIn(expected, log_output)
self.assertHeartbeatNotSavedOffline()
self.assertOfflineHeartbeatsNotSynced()
@log_capture()
def test_requests_exception_without_offline_logging(self, logs):
logging.disable(logging.NOTSET)

View file

@ -37,7 +37,7 @@ class TestCase(unittest.TestCase):
# disable logging while testing
logging.disable(logging.CRITICAL)
self.maxDiff = 1000
self.maxDiff = None
patch_getproxies = mock.patch('wakatime.packages.requests.sessions.get_environ_proxies')
mocked_getproxies = patch_getproxies.start()
@ -293,8 +293,6 @@ class SummaryResponse(Response):
response_code = 200
response_text = '{"data": [{"grand_total": {"text": "4 hrs 23 mins"}}]}'
_count = 0
@property
def status_code(self):
return self.response_code