def test_mark_success(self): start_ts = timezone.now() gatekeeper = Gatekeeper.objects.by_ip(self.ip_address) self.assertTrue(gatekeeper.sync_ts == None) message = Messages.MARK_SUCCESS keymaster = Keymaster(gatekeeper) response = keymaster.process_message(message) self.assertEqual(response, Messages.SUCCESS_RESPONSE) self.assertFalse(gatekeeper.sync_ts == None) self.assertTrue(gatekeeper.sync_ts > start_ts)
def test_handshake(self): message = Messages.TEST_QUESTION gatekeeper = Gatekeeper.objects.by_ip(self.ip_address) keymaster = Keymaster(gatekeeper) response = keymaster.process_message(message) self.assertEqual(response, Messages.TEST_RESPONSE)