rana-cli/tests/test_heartbeat.py

111 lines
3 KiB
Python
Raw Normal View History

2017-11-23 03:29:04 +00:00
# -*- coding: utf-8 -*-
from wakatime.heartbeat import Heartbeat
import os
import logging
from testfixtures import log_capture
from .utils import TestCase
class HeartbeatTestCase(TestCase):
@log_capture()
def test_sanitize_removes_sensitive_data(self, logs):
logging.disable(logging.NOTSET)
class Args(object):
exclude = []
hidefilenames = ['.*']
include = []
plugin = None
data = {
'entity': os.path.realpath('tests/samples/codefiles/python.py'),
'type': 'file',
'project': 'aproject',
'branch': 'abranch',
}
heartbeat = Heartbeat(data, Args(), None)
sanitized = heartbeat.sanitize()
self.assertEquals('HIDDEN.py', sanitized.entity)
sensitive = [
'branch',
'dependencies',
'lines',
'lineno',
'cursorpos',
]
for item in sensitive:
self.assertIsNone(getattr(sanitized, item))
self.assertNothingPrinted()
self.assertNothingLogged(logs)
@log_capture()
def test_sanitize_does_nothing_when_hidefilenames_false(self, logs):
logging.disable(logging.NOTSET)
class Args(object):
exclude = []
hidefilenames = []
include = []
plugin = None
data = {
'entity': os.path.realpath('tests/samples/codefiles/python.py'),
'type': 'file',
'project': 'aproject',
'branch': 'abranch',
}
heartbeat = Heartbeat(data, Args(), None)
heartbeat.branch = data['branch']
sanitized = heartbeat.sanitize()
self.assertEquals(data['branch'], sanitized.branch)
self.assertNothingPrinted()
self.assertNothingLogged(logs)
@log_capture()
def test_sanitize_does_nothing_when_missing_entity(self, logs):
logging.disable(logging.NOTSET)
class Args(object):
hidefilenames = ['.*']
plugin = None
branch = 'abc123'
data = {
'entity': None,
'type': 'file',
'branch': branch,
}
heartbeat = Heartbeat(data, Args(), None, _clone=True)
sanitized = heartbeat.sanitize()
self.assertEquals(branch, sanitized.branch)
self.assertNothingPrinted()
self.assertNothingLogged(logs)
@log_capture()
def test_sanitize_does_nothing_when_type_not_file(self, logs):
logging.disable(logging.NOTSET)
class Args(object):
hidefilenames = ['.*']
plugin = None
branch = 'abc123'
data = {
'entity': 'not.a.file',
'type': 'app',
'branch': branch,
}
heartbeat = Heartbeat(data, Args(), None, _clone=True)
sanitized = heartbeat.sanitize()
self.assertEquals(branch, sanitized.branch)
self.assertNothingPrinted()
self.assertNothingLogged(logs)