def test_users_subscribed_to_event_key(self):
# Make sure we event key
Subscription(parent=ndb.Key(Account, 'user_id_1'),
user_id='user_id_1',
model_key='2020miket',
model_type=ModelType.EVENT,
notification_types=[NotificationType.UPCOMING_MATCH
]).put()
Subscription(parent=ndb.Key(Account, 'user_id_2'),
user_id='user_id_2',
model_key='2020mike2',
model_type=ModelType.EVENT,
notification_types=[NotificationType.MATCH_SCORE]).put()
users = Subscription.users_subscribed_to_event(
MockEvent(key_name="2020miket", year=2020),
NotificationType.UPCOMING_MATCH)
self.assertEqual(users, ['user_id_1'])
def test_users_subscribed_to_event_year_key(self):
# Make sure we fetch both key and year together
Subscription(parent=ndb.Key(Account, 'user_id_1'),
user_id='user_id_1',
model_key='2020miket',
model_type=ModelType.EVENT,
notification_types=[NotificationType.UPCOMING_MATCH
]).put()
Subscription(parent=ndb.Key(Account, 'user_id_2'),
user_id='user_id_2',
model_key='2020*',
model_type=ModelType.EVENT,
notification_types=[NotificationType.UPCOMING_MATCH
]).put()
users = Subscription.users_subscribed_to_event(
MockEvent(key_name="2020miket", year=2020),
NotificationType.UPCOMING_MATCH)
self.assertItemsEqual(users, ['user_id_1', 'user_id_2'])
def test_week_str_not_2016(self):
event_week_0_year_2012 = MockEvent(week=0, year=2012)
self.assertEqual(event_week_0_year_2012.week_str, "Week 1")
event_week_1_year_2022 = MockEvent(week=1, year=2022)
self.assertEqual(event_week_1_year_2022.week_str, "Week 2")
def test_week_str_2016(self):
event_week_0_year_2016 = MockEvent(week=0, year=2016)
self.assertEqual(event_week_0_year_2016.week_str, "Week 0.5")
event_week_1_year_2016 = MockEvent(week=1, year=2016)
self.assertEqual(event_week_1_year_2016.week_str, "Week 1")
def test_week_str_none(self):
event_week_none = MockEvent(week=None, year=2012)
self.assertIsNone(event_week_none.week_str)