64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
# vim: set fileencoding=utf-8
|
|
import unittest
|
|
from unittest.mock import Mock
|
|
from freezegun import freeze_time
|
|
|
|
from bemani.common import GameConstants
|
|
from bemani.data.mysql.network import NetworkData
|
|
from bemani.tests.helpers import FakeCursor
|
|
|
|
|
|
class TestNetworkData(unittest.TestCase):
|
|
def test_get_schedule_type(self) -> None:
|
|
network = NetworkData(Mock(), None)
|
|
|
|
with freeze_time("2016-01-01 12:00"):
|
|
# Check daily schedule
|
|
self.assertEqual(
|
|
# 2016-01-01 -> 2016-01-02
|
|
(1451606400, 1451692800),
|
|
network.get_schedule_duration("daily"),
|
|
)
|
|
|
|
# Check weekly schedule (weeks start on monday in python lol)
|
|
self.assertEqual(
|
|
# 2015-12-27 -> 2916-01-03
|
|
(1451260800, 1451865600),
|
|
network.get_schedule_duration("weekly"),
|
|
)
|
|
|
|
def test_should_schedule(self) -> None:
|
|
network = NetworkData(Mock(), None)
|
|
|
|
with freeze_time("2016-01-01"):
|
|
# Check for should schedule if nothing in DB
|
|
network.execute = Mock(return_value=FakeCursor([])) # type: ignore
|
|
self.assertTrue(
|
|
network.should_schedule(GameConstants.BISHI_BASHI, 1, "work", "daily")
|
|
)
|
|
self.assertTrue(
|
|
network.should_schedule(GameConstants.BISHI_BASHI, 1, "work", "weekly")
|
|
)
|
|
|
|
# Check for don't schedule if DB time is our current time
|
|
network.execute = Mock(return_value=FakeCursor([{"year": 2016, "day": 1}])) # type: ignore
|
|
self.assertFalse(
|
|
network.should_schedule(GameConstants.BISHI_BASHI, 1, "work", "daily")
|
|
)
|
|
|
|
network.execute = Mock(return_value=FakeCursor([{"year": None, "day": 16797}])) # type: ignore
|
|
self.assertFalse(
|
|
network.should_schedule(GameConstants.BISHI_BASHI, 1, "work", "weekly")
|
|
)
|
|
|
|
# Check for do schedule if DB time is older than our current time
|
|
network.execute = Mock(return_value=FakeCursor([{"year": 2015, "day": 365}])) # type: ignore
|
|
self.assertTrue(
|
|
network.should_schedule(GameConstants.BISHI_BASHI, 1, "work", "daily")
|
|
)
|
|
|
|
network.execute = Mock(return_value=FakeCursor([{"year": None, "day": 16790}])) # type: ignore
|
|
self.assertTrue(
|
|
network.should_schedule(GameConstants.BISHI_BASHI, 1, "work", "weekly")
|
|
)
|