class TestMatchScoreNotification(unittest2.TestCase): def setUp(self): self.testbed = testbed.Testbed() self.testbed.activate() self.testbed.init_datastore_v3_stub() self.testbed.init_memcache_stub() ndb.get_context().clear_cache() # Prevent data from leaking between tests self.testbed.init_taskqueue_stub(root_path=".") for team_number in range(7): Team(id="frc%s" % team_number, team_number=team_number).put() self.event = EventTestCreator.createPresentEvent() self.match = self.event.matches[0] self.notification = CompLevelStartingNotification(self.match, self.event) def tearDown(self): self.testbed.deactivate() def test_build(self): expected = {} expected['message_type'] = NotificationType.type_names[NotificationType.LEVEL_STARTING] expected['message_data'] = {} expected['message_data']['event_name'] = self.event.name expected['message_data']['event_key'] = self.event.key_name expected['message_data']['comp_level'] = self.match.comp_level expected['message_data']['scheduled_time'] = calendar.timegm(self.match.time.utctimetuple()) data = self.notification._build_dict() self.assertEqual(expected, data)
class TestMatchScoreNotification(unittest2.TestCase): def setUp(self): self.testbed = testbed.Testbed() self.testbed.activate() self.testbed.init_datastore_v3_stub() self.testbed.init_memcache_stub() ndb.get_context().clear_cache() # Prevent data from leaking between tests self.testbed.init_taskqueue_stub(root_path=".") for team_number in range(7): Team(id="frc%s" % team_number, team_number=team_number).put() self.event = EventTestCreator.createPresentEvent() self.match = self.event.matches[0] self.notification = CompLevelStartingNotification(self.match, self.event) def tearDown(self): self.testbed.deactivate() def test_build(self): expected = {} expected['message_type'] = NotificationType.type_names[NotificationType.LEVEL_STARTING] expected['message_data'] = {} expected['message_data']['event_name'] = self.event.name expected['message_data']['event_key'] = self.event.key_name expected['message_data']['comp_level'] = self.match.comp_level expected['message_data']['scheduled_time'] = calendar.timegm(self.match.time.utctimetuple()) data = self.notification._build_dict() self.assertEqual(expected, data)