114 lines
4.6 KiB
Python
114 lines
4.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import os
|
||
|
|
import time
|
||
|
|
import uuid
|
||
|
|
from testfixtures import log_capture
|
||
|
|
from .utils import ANY, CustomResponse, NamedTemporaryFile, TestCase, mock
|
||
|
|
|
||
|
|
from wakatime.compat import u
|
||
|
|
from wakatime.constants import SUCCESS
|
||
|
|
from wakatime.main import execute
|
||
|
|
from wakatime.packages import requests
|
||
|
|
from wakatime.stats import number_lines_in_file
|
||
|
|
|
||
|
|
|
||
|
|
class StatsTestCase(TestCase):
|
||
|
|
patch_these = [
|
||
|
|
'wakatime.packages.requests.adapters.HTTPAdapter.send',
|
||
|
|
'wakatime.session_cache.SessionCache.save',
|
||
|
|
'wakatime.session_cache.SessionCache.delete',
|
||
|
|
['wakatime.session_cache.SessionCache.get', requests.session],
|
||
|
|
['wakatime.session_cache.SessionCache.connect', None],
|
||
|
|
]
|
||
|
|
|
||
|
|
@log_capture()
|
||
|
|
def test_guess_lexer_using_filename_analyse_text_exception(self, logs):
|
||
|
|
logging.disable(logging.NOTSET)
|
||
|
|
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = CustomResponse()
|
||
|
|
|
||
|
|
with NamedTemporaryFile() as fh:
|
||
|
|
with mock.patch('wakatime.offlinequeue.Queue._get_db_file') as mock_db_file:
|
||
|
|
mock_db_file.return_value = fh.name
|
||
|
|
|
||
|
|
mock_analyse = mock.MagicMock()
|
||
|
|
mock_analyse.analyse_text.side_effect = Exception('foobar')
|
||
|
|
with mock.patch('wakatime.stats.custom_pygments_guess_lexer_for_filename') as mock_lexer:
|
||
|
|
mock_lexer.return_value = mock_analyse
|
||
|
|
|
||
|
|
entity = os.path.realpath('tests/samples/codefiles/python.py')
|
||
|
|
args = ['--file', entity, '--key', str(uuid.uuid4()), '--config', 'tests/samples/configs/good_config.cfg', '--time', u(int(time.time()))]
|
||
|
|
retval = execute(args)
|
||
|
|
self.assertEquals(retval, SUCCESS)
|
||
|
|
|
||
|
|
self.assertNothingPrinted()
|
||
|
|
self.assertNothingLogged(logs)
|
||
|
|
|
||
|
|
heartbeat = {
|
||
|
|
'entity': entity,
|
||
|
|
'language': ANY,
|
||
|
|
'lines': ANY,
|
||
|
|
'project': ANY,
|
||
|
|
'branch': ANY,
|
||
|
|
'time': ANY,
|
||
|
|
'type': 'file',
|
||
|
|
'is_write': ANY,
|
||
|
|
'user_agent': ANY,
|
||
|
|
'dependencies': ANY,
|
||
|
|
}
|
||
|
|
self.assertHeartbeatSent(heartbeat)
|
||
|
|
|
||
|
|
@log_capture()
|
||
|
|
def test_guess_lexer_using_filename_analyse_text_exception_and_logs_error_when_debugging(self, logs):
|
||
|
|
logging.disable(logging.NOTSET)
|
||
|
|
self.patched['wakatime.packages.requests.adapters.HTTPAdapter.send'].return_value = CustomResponse()
|
||
|
|
|
||
|
|
with NamedTemporaryFile() as fh:
|
||
|
|
with mock.patch('wakatime.offlinequeue.Queue._get_db_file') as mock_db_file:
|
||
|
|
mock_db_file.return_value = fh.name
|
||
|
|
|
||
|
|
mock_analyse = mock.MagicMock()
|
||
|
|
mock_analyse.analyse_text.side_effect = Exception('foobar')
|
||
|
|
with mock.patch('wakatime.stats.custom_pygments_guess_lexer_for_filename') as mock_lexer:
|
||
|
|
mock_lexer.return_value = mock_analyse
|
||
|
|
|
||
|
|
entity = os.path.realpath('tests/samples/codefiles/python.py')
|
||
|
|
args = ['--file', entity, '--key', str(uuid.uuid4()), '--config', 'tests/samples/configs/good_config.cfg', '--time', u(int(time.time())), '--verbose']
|
||
|
|
retval = execute(args)
|
||
|
|
self.assertEquals(retval, SUCCESS)
|
||
|
|
|
||
|
|
self.assertNothingPrinted()
|
||
|
|
actual = self.getLogOutput(logs)
|
||
|
|
expected = 'Exception: foobar'
|
||
|
|
self.assertIn(expected, actual)
|
||
|
|
|
||
|
|
heartbeat = {
|
||
|
|
'entity': entity,
|
||
|
|
'language': ANY,
|
||
|
|
'lines': ANY,
|
||
|
|
'project': ANY,
|
||
|
|
'branch': ANY,
|
||
|
|
'time': ANY,
|
||
|
|
'type': 'file',
|
||
|
|
'is_write': ANY,
|
||
|
|
'user_agent': ANY,
|
||
|
|
'dependencies': ANY,
|
||
|
|
}
|
||
|
|
self.assertHeartbeatSent(heartbeat)
|
||
|
|
|
||
|
|
@log_capture()
|
||
|
|
def test_number_lines_in_file_getsize_os_error(self, logs):
|
||
|
|
logging.disable(logging.NOTSET)
|
||
|
|
|
||
|
|
with mock.patch('wakatime.stats.os.path.getsize') as mock_getsize:
|
||
|
|
mock_getsize.side_effect = os.error('')
|
||
|
|
|
||
|
|
entity = os.path.realpath('tests/samples/codefiles/java.java')
|
||
|
|
result = number_lines_in_file(entity)
|
||
|
|
self.assertEquals(result, 22)
|
||
|
|
|
||
|
|
self.assertNothingPrinted()
|
||
|
|
self.assertNothingLogged(logs)
|