def test_first_tick(self, sleep_mock, _):
with transaction_cursor(self.connection) as cursor:
cursor.execute('UPDATE scoring_gamecontrol SET start = datetime("now"), '
' end = datetime("now", "+1 day")')
controller.main_loop_step(self.connection, self.metrics, False)
sleep_mock.assert_called_once_with(0)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT current_tick FROM scoring_gamecontrol')
new_tick = cursor.fetchone()[0]
self.assertEqual(new_tick, 0)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag')
total_flag_count = cursor.fetchone()[0]
self.assertEqual(total_flag_count, 6)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag WHERE service_id=1')
service_flag_count = cursor.fetchone()[0]
self.assertEqual(service_flag_count, 3)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag WHERE protecting_team_id=4')
team_flag_count = cursor.fetchone()[0]
self.assertEqual(team_flag_count, 2)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag WHERE tick=0')
tick_flag_count = cursor.fetchone()[0]
self.assertEqual(tick_flag_count, 6)
def test_null(self, warning_mock, sleep_mock, _):
controller.main_loop_step(self.connection, False)
warning_mock.assert_called_with(
'Competition start and end time must be configured in the database'
)
sleep_mock.assert_called_once_with(60)
def test_before_game(self, sleep_mock, _):
with transaction_cursor(self.connection) as cursor:
cursor.execute('UPDATE scoring_gamecontrol SET start = datetime("now", "+1 hour"), '
' end = datetime("now", "+1 day")')
controller.main_loop_step(self.connection, self.metrics, False)
sleep_mock.assert_called_once_with(60)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT current_tick FROM scoring_gamecontrol')
new_tick = cursor.fetchone()[0]
self.assertEqual(new_tick, -1)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag')
total_flag_count = cursor.fetchone()[0]
self.assertEqual(total_flag_count, 0)
def test_after_game_nonstop(self, sleep_mock, _):
with transaction_cursor(self.connection) as cursor:
cursor.execute('UPDATE scoring_gamecontrol SET start = datetime("now", "-1 day"), '
' end = datetime("now"), '
' current_tick=479')
controller.main_loop_step(self.connection, self.metrics, True)
sleep_mock.assert_called_once_with(0)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT current_tick FROM scoring_gamecontrol')
new_tick = cursor.fetchone()[0]
self.assertEqual(new_tick, 480)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag WHERE tick=480')
tick_flag_count = cursor.fetchone()[0]
self.assertEqual(tick_flag_count, 6)
def test_next_tick_overdue(self, sleep_mock, _):
with transaction_cursor(self.connection) as cursor:
cursor.execute('UPDATE scoring_gamecontrol SET start = datetime("now", "-19 minutes"), '
' end = datetime("now", "+1421 minutes"), '
' current_tick=5')
controller.main_loop_step(self.connection, self.metrics, False)
sleep_mock.assert_called_once_with(0)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT current_tick FROM scoring_gamecontrol')
new_tick = cursor.fetchone()[0]
self.assertEqual(new_tick, 6)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag WHERE tick=6')
tick_flag_count = cursor.fetchone()[0]
self.assertEqual(tick_flag_count, 6)
def test_long_after_game(self, sleep_mock, _):
with transaction_cursor(self.connection) as cursor:
cursor.execute('UPDATE scoring_gamecontrol SET start = datetime("now", "-1465 minutes"), '
' end = datetime("now", "-25 minutes"), '
' current_tick=479')
controller.main_loop_step(self.connection, self.metrics, False)
self.assertEqual(sleep_mock.call_count, 2)
self.assertEqual(sleep_mock.call_args_list[0][0][0], 0)
self.assertEqual(sleep_mock.call_args_list[1][0][0], 60)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT current_tick FROM scoring_gamecontrol')
new_tick = cursor.fetchone()[0]
self.assertEqual(new_tick, 479)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag')
total_flag_count = cursor.fetchone()[0]
self.assertEqual(total_flag_count, 0)
def test_next_tick_undue(self, sleep_mock, _):
with transaction_cursor(self.connection) as cursor:
cursor.execute('UPDATE scoring_gamecontrol SET start = datetime("now", "-1030 seconds"), '
' end = datetime("now", "+85370 seconds"), '
' current_tick=5')
controller.main_loop_step(self.connection, self.metrics, False)
sleep_mock.assert_called_once()
sleep_arg = sleep_mock.call_args[0][0]
self.assertGreater(sleep_arg, 40)
self.assertLessEqual(sleep_arg, 50)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT current_tick FROM scoring_gamecontrol')
new_tick = cursor.fetchone()[0]
self.assertEqual(new_tick, 5)
with transaction_cursor(self.connection) as cursor:
cursor.execute('SELECT COUNT(*) FROM scoring_flag')
tick_flag_count = cursor.fetchone()[0]
self.assertEqual(tick_flag_count, 0)
