class TestAllianceNotification(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.notification = AwardsUpdatedNotification(self.event)
def tearDown(self):
self.testbed.deactivate()
def test_build(self):
expected = {}
expected['message_type'] = NotificationType.type_names[
NotificationType.AWARDS]
expected['message_data'] = {}
expected['message_data']['event_name'] = self.event.name
expected['message_data']['event_key'] = self.event.key_name
expected['message_data']['awards'] = [
ModelToDict.awardConverter(award)
for award in AwardHelper.organizeAwards(self.event.awards)
]
data = self.notification._build_dict()
self.assertEqual(expected, data)
class TestAwardsNotification(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.notification = AwardsUpdatedNotification(self.event)
def tearDown(self):
self.testbed.deactivate()
def test_build(self):
expected = {}
expected['notification_type'] = NotificationType.type_names[NotificationType.AWARDS]
expected['message_data'] = {}
expected['message_data']['event_name'] = self.event.name
expected['message_data']['event_key'] = self.event.key_name
expected['message_data']['awards'] = [ModelToDict.awardConverter(award) for award in AwardHelper.organizeAwards(self.event.awards)]
data = self.notification._build_dict()
self.assertEqual(expected, data)
