67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
import logging
|
|
|
|
import pendulum
|
|
import pytest
|
|
|
|
from bs4 import BeautifulSoup
|
|
from tsstats.log import parse_logs
|
|
from tsstats.template import render_servers
|
|
from tsstats.utils import filter_threshold, seconds_to_text, sort_clients
|
|
|
|
servers = list(parse_logs('tsstats/tests/res/test.log', online_dc=False))
|
|
servers[0] = servers[0]._replace(sid=1) # add missing sid to server object
|
|
clients = servers[0].clients
|
|
|
|
logger = logging.getLogger('tsstats')
|
|
|
|
|
|
@pytest.fixture
|
|
def soup(output):
|
|
render_servers(servers, output)
|
|
return BeautifulSoup(open(output), 'html.parser')
|
|
|
|
|
|
def test_debug(output):
|
|
logger.setLevel(logging.DEBUG)
|
|
render_servers(servers, output)
|
|
logger.setLevel(logging.INFO)
|
|
soup = BeautifulSoup(open(output), 'html.parser')
|
|
# check debug-label presence
|
|
assert soup.find_all(style='color: red; padding-right: 10px;')
|
|
for client_item in soup.find('ul', id='1.onlinetime').find_all('li'):
|
|
nick = client_item.find('span').text
|
|
# check for right identifier
|
|
nick, encl_identifier = nick.split()
|
|
identifier = encl_identifier.replace('(', '').replace(')', '')
|
|
assert clients[identifier].nick == nick
|
|
|
|
|
|
def test_onlinetime(soup):
|
|
items = soup.find('ul', id='1.onlinetime').find_all('li')
|
|
assert len(items) == 2
|
|
for item in items:
|
|
nick, onlinetime = item.find_all('span')
|
|
nick = nick.text
|
|
onlinetime = onlinetime.text
|
|
# find corresponding client-object
|
|
client = list(filter(
|
|
lambda c: c.nick == nick and c.onlinetime > pendulum.Interval(),
|
|
clients
|
|
))
|
|
# assert existence
|
|
assert client
|
|
client = client[0]
|
|
# compare onlinetimes
|
|
client_onlinetime_text = seconds_to_text(
|
|
int(client.onlinetime.total_seconds())
|
|
)
|
|
assert onlinetime == client_onlinetime_text
|
|
|
|
|
|
def test_filter_threshold():
|
|
sorted_clients = sort_clients(
|
|
clients, lambda c: c.onlinetime.total_seconds())
|
|
assert len(filter_threshold(sorted_clients, -1)) == len(sorted_clients)
|
|
assert len(filter_threshold(sorted_clients, 20)) == 1
|
|
assert len(filter_threshold(sorted_clients, 500)) == 0
|