261 lines
11 KiB
Python
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)
|