113 lines
3.0 KiB
Python
113 lines
3.0 KiB
Python
import pendulum
|
|
import pytest
|
|
|
|
from tsstats.exceptions import InvalidLog
|
|
from tsstats.log import TimedLog, _bundle_logs, _parse_details, parse_logs
|
|
from tsstats.template import render_servers
|
|
|
|
testlog_path = 'tsstats/tests/res/test.log'
|
|
|
|
|
|
@pytest.fixture
|
|
def clients():
|
|
return _parse_details(testlog_path, online_dc=False)
|
|
|
|
|
|
def test_log_client_count(clients):
|
|
assert len(clients) == 3
|
|
|
|
|
|
def test_log_onlinetime(clients):
|
|
assert clients['1'].onlinetime == pendulum.Interval(
|
|
seconds=402, microseconds=149208)
|
|
assert clients['2'].onlinetime == pendulum.Interval(
|
|
seconds=19, microseconds=759644)
|
|
|
|
|
|
def test_log_kicks(clients):
|
|
assert clients['UIDClient1'].kicks == 1
|
|
|
|
|
|
def test_log_pkicks(clients):
|
|
assert clients['2'].pkicks == 1
|
|
|
|
|
|
def test_log_bans(clients):
|
|
assert clients['UIDClient1'].bans == 1
|
|
|
|
|
|
def test_log_pbans(clients):
|
|
assert clients['2'].pbans == 1
|
|
|
|
|
|
@pytest.mark.parametrize("logs,bundled", [
|
|
(
|
|
['l1.log', 'l2.log'],
|
|
{'': [TimedLog('l1.log', None), TimedLog('l2.log', None)]}
|
|
),
|
|
(
|
|
[
|
|
'ts3server_2016-06-06__14_22_09.527229_1.log',
|
|
'ts3server_2017-07-07__15_23_10.638340_1.log'
|
|
],
|
|
{
|
|
'1': [
|
|
TimedLog(
|
|
'ts3server_2016-06-06__14_22_09.527229_1.log',
|
|
pendulum.create(
|
|
year=2016, month=6, day=6, hour=14, minute=22,
|
|
second=9, microsecond=527229
|
|
)
|
|
),
|
|
TimedLog(
|
|
'ts3server_2017-07-07__15_23_10.638340_1.log',
|
|
pendulum.create(
|
|
year=2017, month=7, day=7, hour=15, minute=23,
|
|
second=10, microsecond=638340
|
|
)
|
|
)
|
|
]
|
|
}
|
|
)
|
|
])
|
|
def test_log_bundle(logs, bundled):
|
|
assert _bundle_logs(logs) == bundled
|
|
|
|
|
|
def test_log_invalid():
|
|
with pytest.raises(InvalidLog):
|
|
_parse_details('tsstats/tests/res/test.log.broken')
|
|
|
|
|
|
def test_log_client_online():
|
|
current_time = pendulum.now()
|
|
|
|
pendulum.set_test_now(current_time)
|
|
clients = _parse_details(testlog_path)
|
|
old_onlinetime = int(clients['1'].onlinetime.total_seconds())
|
|
|
|
pendulum.set_test_now(current_time.add(seconds=2)) # add 2s to .now()
|
|
clients = _parse_details(testlog_path)
|
|
assert int(clients['1'].onlinetime.total_seconds()) == old_onlinetime + 2
|
|
|
|
|
|
def test_parse_logs():
|
|
assert len(_parse_details(testlog_path)) ==\
|
|
len(parse_logs(testlog_path)[0].clients)
|
|
|
|
|
|
def test_parse_groups():
|
|
clients = _parse_details('tsstats/tests/res/test.log.groups')
|
|
assert len(clients) == 0
|
|
|
|
|
|
def test_parse_utf8(output):
|
|
servers = parse_logs(testlog_path + '.utf8')
|
|
render_servers(servers, output)
|
|
|
|
|
|
def test_server_stop():
|
|
clients = _parse_details('tsstats/tests/res/test.log.stopped')
|
|
assert clients['1'].onlinetime.seconds / 60 == 20 # minutes
|
|
assert clients['2'].onlinetime.seconds / 60 == 10 # minutes
|