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)
