123 lines
3.5 KiB
Python
123 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import datetime
|
|
import mock
|
|
|
|
from q3stats.lib.charts import day
|
|
from q3stats.lib.scripts import utils
|
|
from q3stats.models import Game, Score
|
|
from q3stats.testing import BaseQ3StatsTestCase
|
|
|
|
|
|
class Test_LibChartsDay(BaseQ3StatsTestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(Test_LibChartsDay, cls).setUpClass()
|
|
|
|
with utils.db_session(cls._config) as session:
|
|
game1_dt = datetime.datetime(2017, 2, 14, 20, 21, 0)
|
|
|
|
game1 = Game(
|
|
uuid='game1',
|
|
map='Q3DM7',
|
|
date=game1_dt.date(),
|
|
time=game1_dt.time(),
|
|
fraglimit=20,
|
|
attrs={}
|
|
)
|
|
|
|
game1.scores.extend([
|
|
Score(
|
|
player='Player 1',
|
|
score=20,
|
|
kills=21,
|
|
deaths=10,
|
|
suicides=1,
|
|
net=11,
|
|
damage_taken=123,
|
|
damage_given=456,
|
|
total_health=123,
|
|
total_armor=456,
|
|
weapons={},
|
|
items={},
|
|
powerups={}
|
|
),
|
|
Score(
|
|
player='Player 2',
|
|
score=10,
|
|
kills=11,
|
|
deaths=20,
|
|
suicides=1,
|
|
net=10,
|
|
damage_taken=123,
|
|
damage_given=456,
|
|
total_health=123,
|
|
total_armor=456,
|
|
weapons={},
|
|
items={},
|
|
powerups={}
|
|
),
|
|
])
|
|
|
|
game2_dt = game1_dt + datetime.timedelta(seconds=60)
|
|
game2 = Game(
|
|
uuid='game2',
|
|
map='Q3DM17',
|
|
date=game2_dt.date(),
|
|
time=game2_dt.time(),
|
|
fraglimit=20,
|
|
attrs={}
|
|
)
|
|
|
|
game2.scores.extend([
|
|
Score(
|
|
player='Player 1',
|
|
score=10,
|
|
kills=11,
|
|
deaths=20,
|
|
suicides=1,
|
|
net=10,
|
|
damage_taken=123,
|
|
damage_given=456,
|
|
total_health=123,
|
|
total_armor=456,
|
|
weapons={},
|
|
items={},
|
|
powerups={}
|
|
),
|
|
Score(
|
|
player='Player 2',
|
|
score=20,
|
|
kills=21,
|
|
deaths=10,
|
|
suicides=1,
|
|
net=11,
|
|
damage_taken=123,
|
|
damage_given=456,
|
|
total_health=123,
|
|
total_armor=456,
|
|
weapons={},
|
|
items={},
|
|
powerups={}
|
|
),
|
|
])
|
|
|
|
session.add_all([game1, game2])
|
|
session.commit()
|
|
|
|
cls._day = game1_dt.date()
|
|
|
|
def test_get_day_chart(self):
|
|
with utils.db_session(self._config) as session:
|
|
maps, scores = day.get_day_chart(self._day, session)
|
|
|
|
assert maps == ['Q3DM7', 'Q3DM17']
|
|
|
|
pl1_scores, pl2_scores = scores
|
|
|
|
assert pl1_scores['name'] == 'Player 1'
|
|
assert pl1_scores['data'] == [20, 10]
|
|
|
|
assert pl2_scores['name'] == 'Player 2'
|
|
assert pl2_scores['data'] == [10, 20]
|