async def test_something_to_send_multiple(self): self.dbmain.getAutoRepeats.return_value = AsyncIterator([ RepeatData('botgotsthis', '', 'FrankerZ', 10, 1, self.now - timedelta(minutes=1)), RepeatData('megotsthis', '', 'Kappa', 10, 1, self.now - timedelta(minutes=1)), ]) self.assertEqual([r async for r in self.data.getAutoRepeatToSend()], [ AutoRepeatMessage('botgotsthis', '', 'FrankerZ'), AutoRepeatMessage('megotsthis', '', 'Kappa') ])
async def test_multiple(self): self.data.getAutoRepeatToSend.return_value = AsyncIterator([ AutoRepeatMessage('botgotsthis', '', 'Kappa'), AutoRepeatMessage('botgotsthis', 'Kappa', 'Keepo'), ]) await tasks.autoRepeatMessage(self.now) self.data.sentAutoRepeat.assert_has_calls( [call('botgotsthis', ''), call('botgotsthis', 'Kappa')]) self.assertFalse(self.mock_timeout.called) self.channel.send.assert_has_calls([call('Kappa'), call('Keepo')])
async def test_something_to_send_multiple(self): dt = datetime.utcnow() dt -= timedelta(microseconds=dt.microsecond) await self.executemany( "INSERT INTO auto_repeat VALUES (?, ?, ?, ?, ?, ?, ?)", [('botgotsthis', '', 'Kappa', 10, 5, dt - timedelta(minutes=5), dt), ('botgotsthis', 'kappa', 'Kappa', None, 5, dt - timedelta(minutes=5), dt)]) self.assertEqual( [r async for r in self.database.getAutoRepeatToSend()], [ AutoRepeatMessage('botgotsthis', '', 'Kappa'), AutoRepeatMessage('botgotsthis', 'kappa', 'Kappa') ])
async def test(self): self.assertEqual( [r async for r in self.data.getAutoRepeatToSend(self.now)], [AutoRepeatMessage('megotsthis', '', 'Kappa')]) self.assertTrue(self.dbmain.getAutoRepeats.called) self.dbmain.getAutoRepeats.reset_mock() self.assertEqual( [r async for r in self.data.getAutoRepeatToSend(self.now)], [AutoRepeatMessage('megotsthis', '', 'Kappa')]) self.assertFalse(self.dbmain.getAutoRepeats.called) self.assertEqual( [r async for r in self.data.listAutoRepeat('botgotsthis')], [AutoRepeatList('', 'FrankerZ', None, 1, self.now)]) self.assertFalse(self.dbmain.getAutoRepeats.called) self.assertIsNotNone(await self.redis.get('autorepeat'))
async def test_no_channel(self): self.data.getAutoRepeatToSend.return_value = AsyncIterator( [AutoRepeatMessage('megotsthis', '', 'Kappa')]) await tasks.autoRepeatMessage(self.now) self.assertFalse(self.data.sentAutoRepeat.called) self.assertFalse(self.mock_timeout.called) self.assertFalse(self.channel.send.called)
async def test(self): self.data.getAutoRepeatToSend.return_value = AsyncIterator([ AutoRepeatMessage('botgotsthis', '', 'Kappa'), ]) await tasks.autoRepeatMessage(self.now) self.data.sentAutoRepeat.assert_called_once_with('botgotsthis', '') self.assertFalse(self.mock_timeout.called) self.channel.send.assert_called_once_with('Kappa')
async def test_something_to_send_decimal(self): self.dbmain.getAutoRepeats.return_value = AsyncIterator([ RepeatData('botgotsthis', '', 'FrankerZ', None, 1, self.now - timedelta(seconds=5)), RepeatData('megotsthis', '', 'Kappa', None, 0.1, self.now - timedelta(seconds=6)), ]) self.assertEqual([r async for r in self.data.getAutoRepeatToSend()], [AutoRepeatMessage('megotsthis', '', 'Kappa')])
async def test_something_to_send(self): await self.executemany( "INSERT INTO auto_repeat VALUES (?, ?, ?, ?, ?, ?, ?)", [('botgotsthis', '', 'Kappa', None, 5, datetime.utcnow() - timedelta(minutes=5), datetime.utcnow()), ('botgotsthis', 'kappa', 'Kappa', None, 5, datetime.utcnow(), datetime.utcnow())]) self.assertEqual( [r async for r in self.database.getAutoRepeatToSend()], [AutoRepeatMessage('botgotsthis', '', 'Kappa')])
async def test_mod(self): self.channel.isMod = True self.data.getAutoRepeatToSend.return_value = AsyncIterator([ AutoRepeatMessage('botgotsthis', '', 'Kappa'), ]) await tasks.autoRepeatMessage(self.now) self.data.sentAutoRepeat.assert_called_once_with('botgotsthis', '') self.mock_timeout.assert_called_once_with(self.channel, None, 'Kappa', None, 'autorepeat') self.channel.send.assert_called_once_with('Kappa')
async def test_something_to_send(self): self.assertEqual( [r async for r in self.data.getAutoRepeatToSend(self.now)], [AutoRepeatMessage('megotsthis', '', 'Kappa')])