class TestUpcomingMatchNotification(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.match.predicted_time = self.match.time self.notification = UpcomingMatchNotification(self.match, self.event) def tearDown(self): self.testbed.deactivate() def test_build(self): expected = {} self.maxDiff = None expected['notification_type'] = NotificationType.type_names[ NotificationType.UPCOMING_MATCH] expected['message_data'] = {} expected['message_data']['event_key'] = self.event.key_name expected['message_data']['event_name'] = self.event.name expected['message_data']['match_key'] = self.match.key_name expected['message_data']['team_keys'] = self.match.team_key_names if self.match.time: expected['message_data']['scheduled_time'] = calendar.timegm( self.match.time.utctimetuple()) else: expected['message_data']['scheduled_time'] = None expected['message_data']['predicted_time'] = expected['message_data'][ 'scheduled_time'] expected['message_data']['webcast'] = { 'channel': '6540154', 'status': 'unknown', 'stream_title': None, 'type': 'ustream', 'viewer_count': None, } data = self.notification._build_dict() self.assertEqual(expected, data)
class TestUpcomingMatchNotification(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.match.predicted_time = self.match.time self.notification = UpcomingMatchNotification(self.match, self.event) def tearDown(self): self.testbed.deactivate() def test_build(self): expected = {} self.maxDiff = None expected['message_type'] = NotificationType.type_names[NotificationType.UPCOMING_MATCH] expected['message_data'] = {} expected['message_data']['event_key'] = self.event.key_name expected['message_data']['event_name'] = self.event.name expected['message_data']['match_key'] = self.match.key_name expected['message_data']['team_keys'] = self.match.team_key_names if self.match.time: expected['message_data']['scheduled_time'] = calendar.timegm(self.match.time.utctimetuple()) else: expected['message_data']['scheduled_time'] = None expected['message_data']['predicted_time'] = expected['message_data']['scheduled_time'] expected['message_data']['webcast'] = { 'channel': '6540154', 'status': 'unknown', 'stream_title': None, 'type': 'ustream' } data = self.notification._build_dict() self.assertEqual(expected, data)
class TestUpcomingMatchNotification(unittest2.TestCase): def setUp(self): self.testbed = testbed.Testbed() self.testbed.activate() self.testbed.init_datastore_v3_stub() self.testbed.init_memcache_stub() 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 = UpcomingMatchNotification(self.match, self.event) def tearDown(self): self.testbed.deactivate() def test_build(self): expected = {} expected['message_type'] = NotificationType.type_names[ NotificationType.UPCOMING_MATCH] expected['message_data'] = {} expected['message_data']['event_name'] = self.event.name expected['message_data']['match_key'] = self.match.key_name expected['message_data']['team_keys'] = self.match.team_key_names if self.match.time: expected['message_data']['scheduled_time'] = calendar.timegm( self.match.time.utctimetuple()) expected['message_data']['predicted_time'] = calendar.timegm( self.match.time.utctimetuple()) else: expected['message_data']['scheduled_time'] = None expected['message_data']['predicted_time'] = None data = self.notification._build_dict() self.assertEqual(expected, data)
class TestUpcomingMatchNotification(unittest2.TestCase): def setUp(self): self.testbed = testbed.Testbed() self.testbed.activate() self.testbed.init_datastore_v3_stub() self.testbed.init_memcache_stub() 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 = UpcomingMatchNotification(self.match, self.event) def tearDown(self): self.testbed.deactivate() def test_build(self): expected = {} expected['message_type'] = NotificationType.type_names[NotificationType.UPCOMING_MATCH] expected['message_data'] = {} expected['message_data']['event_key'] = self.event.key_name expected['message_data']['event_name'] = self.event.name expected['message_data']['match_key'] = self.match.key_name expected['message_data']['team_keys'] = self.match.team_key_names if self.match.time: expected['message_data']['scheduled_time'] = calendar.timegm(self.match.time.utctimetuple()) expected['message_data']['predicted_time'] = calendar.timegm(self.match.time.utctimetuple()) else: expected['message_data']['scheduled_time'] = None expected['message_data']['predicted_time'] = None data = self.notification._build_dict() self.assertEqual(expected, data)