Max/DragonMinded_bemaniutils
1
0
Fork 0
mirror of https://github.com/DragonMinded/bemaniutils.git synced 2026-10-11 08:48:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Bring tests under typing checks, fix existing type errors.

Browse Source
This commit is contained in:
Jennifer Taylor
2021-08-19 19:20:13 +00:00
parent 6efd07d02f
commit fa0b2bd6cd
9 changed files with 104 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bemani/tests/test_NetworkData.py
+12 -11
View File
@@ -3,6 +3,7 @@ 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
@@ -32,20 +33,20 @@ class TestNetworkData(unittest.TestCase):
with freeze_time('2016-01-01'):
# Check for should schedule if nothing in DB
network.execute = Mock(return_value=FakeCursor([]))
self.assertTrue(network.should_schedule('game', 1, 'work', 'daily'))
self.assertTrue(network.should_schedule('game', 1, 'work', 'weekly'))
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}]))
self.assertFalse(network.should_schedule('game', 1, 'work', 'daily'))
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}]))
self.assertFalse(network.should_schedule('game', 1, 'work', 'weekly'))
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}]))
self.assertTrue(network.should_schedule('game', 1, 'work', 'daily'))
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}]))
self.assertTrue(network.should_schedule('game', 1, 'work', 'weekly'))
network.execute = Mock(return_value=FakeCursor([{'year': None, 'day': 16790}])) # type: ignore
self.assertTrue(network.should_schedule(GameConstants.BISHI_BASHI, 1, 'work', 'weekly'))