1
0
mirror of synced 2024-11-24 14:30:11 +01:00
bemaniutils/bemani/tests/test_PlayStats.py

261 lines
11 KiB
Python

# vim: set fileencoding=utf-8
import unittest
from freezegun import freeze_time
from typing import Dict, Any
from unittest.mock import Mock
from bemani.backend.base import Base
from bemani.common import GameConstants, Time, ValidatedDict
from bemani.data import UserID
# Make the normally-abstract base instantiable so we can test it.
class InstantiableBase(Base):
game = GameConstants.BISHI_BASHI
version = -1
name = "Test Class"
# Make an easier mock implementation of load/save stats.
def mock_stats(existing_value: Dict[str, Any]) -> Mock:
data = Mock()
data.local = Mock()
data.local.game = Mock()
data.local.game.get_settings = Mock(return_value=ValidatedDict(existing_value) if existing_value else None)
data.local.game.put_settings = Mock()
return data
def saved_stats(mock: Mock) -> ValidatedDict:
return ValidatedDict(mock.local.game.put_settings.call_args.args[2])
class TestPlayStats(unittest.TestCase):
def test_get_brand_new_profile(self) -> None:
with freeze_time('2021-08-24'):
stats = None
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
self.assertEqual(settings.game, GameConstants.BISHI_BASHI)
self.assertEqual(settings.total_plays, 1)
self.assertEqual(settings.today_plays, 1)
self.assertEqual(settings.total_days, 1)
self.assertEqual(settings.consecutive_days, 1)
self.assertEqual(settings.first_play_timestamp, Time.now())
self.assertEqual(settings.last_play_timestamp, Time.now())
def test_put_brand_new_profile(self) -> None:
with freeze_time('2021-08-24'):
stats = None
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
base.update_play_statistics(UserID(1337), settings)
new_settings = saved_stats(data)
self.assertEqual(new_settings.get_int('total_plays'), 1)
self.assertEqual(new_settings.get_int('today_plays'), 1)
self.assertEqual(new_settings.get_int('total_days'), 1)
self.assertEqual(new_settings.get_int('consecutive_days'), 1)
self.assertEqual(new_settings.get_int('first_play_timestamp'), Time.now())
self.assertEqual(new_settings.get_int('last_play_timestamp'), Time.now())
self.assertEqual(new_settings.get_int_array('last_play_date', 3), Time.todays_date())
def test_get_played_today(self) -> None:
with freeze_time('2021-08-24'):
play_date = Time.todays_date()
stats = {
'total_plays': 1234,
'today_plays': 420,
'total_days': 10,
'first_play_timestamp': 1234567890,
'last_play_timestamp': 9876543210,
'consecutive_days': 69,
'last_play_date': [play_date[0], play_date[1], play_date[2]],
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
self.assertEqual(settings.game, GameConstants.BISHI_BASHI)
self.assertEqual(settings.total_plays, 1235)
self.assertEqual(settings.today_plays, 421)
self.assertEqual(settings.total_days, 10)
self.assertEqual(settings.consecutive_days, 69)
self.assertEqual(settings.first_play_timestamp, 1234567890)
self.assertEqual(settings.last_play_timestamp, 9876543210)
def test_put_played_today(self) -> None:
with freeze_time('2021-08-24'):
play_date = Time.todays_date()
stats = {
'total_plays': 1234,
'today_plays': 420,
'total_days': 10,
'first_play_timestamp': 1234567890,
'last_play_timestamp': 1234567890,
'consecutive_days': 69,
'last_play_date': [play_date[0], play_date[1], play_date[2]],
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
base.update_play_statistics(UserID(1337), settings)
new_settings = saved_stats(data)
self.assertEqual(new_settings.get_int('total_plays'), 1235)
self.assertEqual(new_settings.get_int('today_plays'), 421)
self.assertEqual(new_settings.get_int('total_days'), 10)
self.assertEqual(new_settings.get_int('consecutive_days'), 69)
self.assertEqual(new_settings.get_int('first_play_timestamp'), 1234567890)
self.assertEqual(new_settings.get_int('last_play_timestamp'), Time.now())
self.assertEqual(new_settings.get_int_array('last_play_date', 3), Time.todays_date())
def test_get_played_yesterday(self) -> None:
with freeze_time('2021-08-24'):
play_date = Time.yesterdays_date()
stats = {
'total_plays': 1234,
'today_plays': 420,
'total_days': 10,
'first_play_timestamp': 1234567890,
'last_play_timestamp': 9876543210,
'consecutive_days': 69,
'last_play_date': [play_date[0], play_date[1], play_date[2]],
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
self.assertEqual(settings.game, GameConstants.BISHI_BASHI)
self.assertEqual(settings.total_plays, 1235)
self.assertEqual(settings.today_plays, 1)
self.assertEqual(settings.total_days, 11)
self.assertEqual(settings.consecutive_days, 70)
self.assertEqual(settings.first_play_timestamp, 1234567890)
self.assertEqual(settings.last_play_timestamp, 9876543210)
def test_put_played_yesterday(self) -> None:
with freeze_time('2021-08-24'):
play_date = Time.yesterdays_date()
stats = {
'total_plays': 1234,
'today_plays': 420,
'total_days': 10,
'first_play_timestamp': 1234567890,
'last_play_timestamp': 1234567890,
'consecutive_days': 69,
'last_play_date': [play_date[0], play_date[1], play_date[2]],
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
base.update_play_statistics(UserID(1337), settings)
new_settings = saved_stats(data)
self.assertEqual(new_settings.get_int('total_plays'), 1235)
self.assertEqual(new_settings.get_int('today_plays'), 1)
self.assertEqual(new_settings.get_int('total_days'), 11)
self.assertEqual(new_settings.get_int('consecutive_days'), 70)
self.assertEqual(new_settings.get_int('first_play_timestamp'), 1234567890)
self.assertEqual(new_settings.get_int('last_play_timestamp'), Time.now())
self.assertEqual(new_settings.get_int_array('last_play_date', 3), Time.todays_date())
def test_get_played_awhile_ago(self) -> None:
with freeze_time('2021-08-24'):
stats = {
'total_plays': 1234,
'today_plays': 420,
'total_days': 10,
'first_play_timestamp': 1234567890,
'last_play_timestamp': 9876543210,
'consecutive_days': 69,
'last_play_date': [2010, 4, 20],
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
self.assertEqual(settings.game, GameConstants.BISHI_BASHI)
self.assertEqual(settings.total_plays, 1235)
self.assertEqual(settings.today_plays, 1)
self.assertEqual(settings.total_days, 11)
self.assertEqual(settings.consecutive_days, 1)
self.assertEqual(settings.first_play_timestamp, 1234567890)
self.assertEqual(settings.last_play_timestamp, 9876543210)
def test_put_played_awhile_ago(self) -> None:
with freeze_time('2021-08-24'):
stats = {
'total_plays': 1234,
'today_plays': 420,
'total_days': 10,
'first_play_timestamp': 1234567890,
'last_play_timestamp': 1234567890,
'consecutive_days': 69,
'last_play_date': [2010, 4, 20],
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
base.update_play_statistics(UserID(1337), settings)
new_settings = saved_stats(data)
self.assertEqual(new_settings.get_int('total_plays'), 1235)
self.assertEqual(new_settings.get_int('today_plays'), 1)
self.assertEqual(new_settings.get_int('total_days'), 11)
self.assertEqual(new_settings.get_int('consecutive_days'), 1)
self.assertEqual(new_settings.get_int('first_play_timestamp'), 1234567890)
self.assertEqual(new_settings.get_int('last_play_timestamp'), Time.now())
self.assertEqual(new_settings.get_int_array('last_play_date', 3), Time.todays_date())
def test_get_extra_settings(self) -> None:
with freeze_time('2021-08-24'):
stats = {
'total_plays': 1234,
'key': 'value',
'int': 1337,
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
self.assertEqual(settings.get_int('int'), 1337)
self.assertEqual(settings.get_str('key'), 'value')
self.assertEqual(settings.get_int('total_plays'), 0)
def test_put_extra_settings(self) -> None:
with freeze_time('2021-08-24'):
stats = {
'total_plays': 1234,
'key': 'value',
'int': 1337,
}
data = mock_stats(stats)
base = InstantiableBase(data, Mock(), Mock())
settings = base.get_play_statistics(UserID(1337))
settings.replace_int('int', 420)
settings.replace_int('int2', 69)
settings.replace_int('total_plays', 37)
base.update_play_statistics(UserID(1337), settings)
new_settings = saved_stats(data)
self.assertEqual(new_settings.get_int('int'), 420)
self.assertEqual(new_settings.get_str('key'), 'value')
self.assertEqual(new_settings.get_int('int2'), 69)
self.assertEqual(new_settings.get_int('total_plays'), 1235)