def test_filter_subscriptions_by_user_data_all(database): database.sub.rule_logic = 'all' rule1 = Rule(name="department", value="a").put() rule2 = Rule(name="location", value="c").put() database.sub.user_rules = [rule1, rule2] database.sub.put() preference = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) preference.put() user = User(email='*****@*****.**', subscription_preferences=[preference.key]) user.metadata = {"department": "a", "location": "b"} user.put() merged_preferences = merge_subscriptions_with_preferences(user) subscriptions = filter_subscriptions_by_user_data(merged_preferences, user) assert subscriptions == [] user.metadata = {'department': 'a', 'location': 'c'} user.put() merged_preferences = merge_subscriptions_with_preferences(user) subscriptions = filter_subscriptions_by_user_data(merged_preferences, user) assert len(subscriptions) == 1 assert subscriptions[0]['id'] == database.sub.key.urlsafe()
def test_no_re_matches(minimal_database): pref_1 = SubscriptionDateTime( datetime=datetime.now() - timedelta(weeks=MEETING_COOLDOWN_WEEKS - 1)).put() subscription = MeetingSubscription(title='all engineering weekly', datetime=[pref_1]).put() user_pref = UserSubscriptionPreferences(preference=pref_1, subscription=subscription).put() meeting_spec = MeetingSpec(meeting_subscription=subscription, datetime=pref_1.get().datetime) meeting_spec.put() users = [] num_users = 20 for i in range(0, num_users): user = User(email='{}@yelp.com'.format(i), metadata={'department': 'dept{}'.format(i)}, subscription_preferences=[user_pref]) user.put() MeetingRequest(user=user.key, meeting_spec=meeting_spec.key).put() users.append(user) previous_meetings = { pair for pair in itertools.combinations([user.key.id() for user in users], 2) } previous_meetings = previous_meetings - { (users[0].key.id(), users[1].key.id()) } matches, unmatched = generate_meetings(users, meeting_spec, previous_meetings) assert len(unmatched) == num_users - 2 assert [(match[0].key.id(), match[1].key.id()) for match in matches] == [(users[0].key.id(), users[1].key.id())]
def test_get_meeting_participants(app, database): pref = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) pref.put() user1 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key]) user1.put() user2 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key]) user2.put() meeting1 = Meeting(meeting_spec=database.specs[0].key, cancelled=False).put() MeetingParticipant(meeting=meeting1, user=user1.key).put() MeetingParticipant(meeting=meeting1, user=user2.key).put() with app.test_request_context('/v1/metrics/meeting_participants'): participants = json.loads(meeting_participants()) assert participants == [{ 'date': '2017-10-27T23:00:00', 'meeting': 'agx0ZXN0YmVkLXRlc3RyDQsSB01lZXRpbmcYCgw', 'meeting_title': 'Yelp Weekly', 'participant': '*****@*****.**', 'time': '04:00PM' }, { 'date': '2017-10-27T23:00:00', 'meeting': 'agx0ZXN0YmVkLXRlc3RyDQsSB01lZXRpbmcYCgw', 'meeting_title': 'Yelp Weekly', 'participant': '*****@*****.**', 'time': '04:00PM' }]
def _fake_user(): user_list = [] subscription = MeetingSubscription.query().get() for user in FAKE_USER: preferences = UserSubscriptionPreferences( preference=subscription.datetime[0], subscription=subscription.key, ).put() user_entity = User( first_name=user['first_name'], last_name=user['last_name'], email=user['email'], photo_url=user['photo_url'], metadata={ 'department': user['department'], 'office': 'USA: CA SF New Montgomery Office', 'company_profile_url': 'https://www.yelp.com/user_details?userid=nkN_do3fJ9xekchVC-v68A', }, subscription_preferences=[preferences], ) user_entity.put() user_list.append(user_entity) return user_list[0]
def test_filter_subscriptions_by_user_data(database): preference = UserSubscriptionPreferences( subscription=database.sub.key, preference=database.prefs[0].key ) preference.put() user = User( email='*****@*****.**', subscription_preferences=[preference.key] ) user.metadata = {"department": "a"} user.put() rule = Rule(name="department", value="a").put() database.sub.rules = [rule] database.sub.put() merged_preferences = merge_subscriptions_with_preferences(user) subscriptions = filter_subscriptions_by_user_data(merged_preferences, user) assert len(subscriptions) == 1 assert subscriptions[0]['id'] == database.sub.key.urlsafe() user.metadata = {"department": "b"} user.put() merged_preferences = merge_subscriptions_with_preferences(user) subscriptions = filter_subscriptions_by_user_data(merged_preferences, user) assert subscriptions == []
def test_weekly_opt_in(minimal_database, subscription): preference = subscription.datetime[0] user_pref = UserSubscriptionPreferences( preference=preference, subscription=subscription.key, ).put() user1 = User(email="*****@*****.**", metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user1.put() response = weekly_opt_in() assert response == 'OK'
def test_add_preferences_adds_on_opt_in(minimal_database, subscription): preference = subscription.datetime[0] user = User(email='*****@*****.**', metadata={'department': 'dept'}) user.put() updated_preferences = {preference: True} assert len(user.subscription_preferences) == 0 added = add_preferences(user, updated_preferences, subscription.key) assert added.pop() == preference assert user.key.get().subscription_preferences[0].get( ).preference == preference
def test_get_meeting_requests(app, database): pref = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) pref.put() user = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key]) user.put() MeetingRequest(user=user.key, meeting_spec=database.specs[0].key).put() with app.test_request_context('/v1/metrics/meeting_requests'): requests = json.loads(meeting_requests()) assert requests == [{'title': 'Yelp Weekly', 'user': '******'}]
def test_generate_meetings_same_department(minimal_database, subscription): preference = subscription.datetime[0] user_pref = UserSubscriptionPreferences(preference=preference, subscription=subscription.key).put() user1 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user1.put() user2 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user2.put() user_list = [user1, user2] _, specs = get_specs_from_subscription(subscription) matches, unmatched = generate_meetings(user_list, specs[0]) assert len(unmatched) == 2 assert len(matches) == 0
def test_filter_subscriptions_by_user_data_without_rules(database): database.sub.rule_logic = 'all' database.sub.user_rules = [] database.sub.put() preference = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) preference.put() user = User(email='*****@*****.**', subscription_preferences=[preference.key]) user.metadata = {"department": "a", "location": "b"} user.put() merged_preferences = merge_subscriptions_with_preferences(user) with pytest.raises(AssertionError): filter_subscriptions_by_user_data(merged_preferences, user)
def test_filter_subscriptions_by_user_data_none(database): database.sub.put() preference = UserSubscriptionPreferences( subscription=database.sub.key, preference=database.prefs[0].key, ) preference.put() user = User(email='*****@*****.**', subscription_preferences=[preference.key]) user.put() merged_preferences = merge_subscriptions_with_preferences(user) subscriptions = filter_subscriptions_by_user_data(merged_preferences, user) assert len(subscriptions) == 1 assert subscriptions[0]['id'] == database.sub.key.urlsafe()
def test_user_preference(minimal_database, subscription): preference = subscription.datetime[0] user_pref = UserSubscriptionPreferences( preference=preference, subscription=subscription.key, ).put() user = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user.put() meeting_spec = MeetingSpec( meeting_subscription=subscription.key, datetime=subscription.datetime[0].get().datetime) meeting_spec.put() assert user_pref == user_preference(user, meeting_spec)
def test_add_preferences_adds_multiple_on_opt_in(minimal_database, subscription): preference_1 = subscription.datetime[0] preference_2 = subscription.datetime[1] user = User(email='*****@*****.**', metadata={'department': 'dept'}) user.put() updated_preferences = {preference_1: True, preference_2: True} assert len(user.subscription_preferences) == 0 added = add_preferences(user, updated_preferences, subscription.key) assert preference_1 in added assert preference_2 in added assert len(user.subscription_preferences) == 2 assert user.key.get().subscription_preferences[0].get().preference in ( preference_1, preference_2) assert user.key.get().subscription_preferences[1].get().preference in ( preference_1, preference_2)
def test_generate_group_meeting_invalid_number_of_users(minimal_database): pref_1 = SubscriptionDateTime(datetime=datetime.now() - timedelta(weeks=MEETING_COOLDOWN_WEEKS - 1)).put() subscription = MeetingSubscription(title='all engineering weekly', datetime=[pref_1]).put() user_pref = UserSubscriptionPreferences(preference=pref_1, subscription=subscription).put() meeting_spec = MeetingSpec(meeting_subscription=subscription, datetime=pref_1.get().datetime) meeting_spec.put() users = [] for i in range(0, 2): user = User(email='{}@yelp.com'.format(i), metadata={ 'department': 'dept{}'.format(i)}, subscription_preferences=[user_pref]) user.put() MeetingRequest(user=user.key, meeting_spec=meeting_spec.key).put() users.append(user) matches, unmatched = generate_meetings(users, meeting_spec, prev_meeting_tuples=None, group_size=3) assert(len(matches) == 0) assert (len(unmatched) == 2)
def test_remove_preferences_removes_on_opt_out(minimal_database, subscription): preference = subscription.datetime[0] user_pref = UserSubscriptionPreferences( preference=preference, subscription=subscription.key, ).put() user = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user.put() assert user.subscription_preferences == [user_pref] updated_preferences = {preference: False} removed = remove_preferences(user, updated_preferences, subscription.key) assert removed == {user_pref} user = user.key.get() assert user.subscription_preferences == [] assert UserSubscriptionPreferences.query().fetch() == []
def test_generate_meetings_with_history(minimal_database, subscription): rule = Rule(name='department', value='').put() subscription.dept_rules = [rule] preference = subscription.datetime[0] user_pref = UserSubscriptionPreferences( preference=preference, subscription=subscription.key).put() user1 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user1.put() user2 = User(email='*****@*****.**', metadata={'department': 'dept2'}, subscription_preferences=[user_pref]) user2.put() user3 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref]) user3.put() user4 = User(email='*****@*****.**', metadata={'department': 'dept2'}, subscription_preferences=[user_pref]) user4.put() user_list = [user1, user2, user3, user4] week_start, specs = get_specs_from_subscription(subscription) store_specs_from_subscription(subscription.key, week_start, specs) matches, unmatched = generate_meetings(user_list, specs[0]) assert len(matches) == 2 assert len(unmatched) == 0 meeting_history = set([ (user1.key.id(), user2.key.id()), (user3.key.id(), user4.key.id()), (user2.key.id(), user3.key.id()), (user1.key.id(), user4.key.id()), ]) matches, unmatched = generate_meetings(user_list, specs[0], meeting_history) assert len(matches) == 0 assert len(unmatched) == 4 meeting_history = set([ (user1.key.id(), user2.key.id()), (user3.key.id(), user4.key.id()), (user2.key.id(), user3.key.id()), ]) matches, unmatched = generate_meetings(user_list, specs[0], meeting_history) assert len(matches) == 1 assert len(unmatched) == 2
def test_previous_meeting_penalty(minimal_database): pref_1 = SubscriptionDateTime(datetime=datetime.now() - timedelta(weeks=MEETING_COOLDOWN_WEEKS - 1)).put() pref_2 = SubscriptionDateTime(datetime=datetime.now() - timedelta(weeks=MEETING_COOLDOWN_WEEKS - 2)).put() pref_3 = SubscriptionDateTime(datetime=datetime.now() - timedelta(weeks=MEETING_COOLDOWN_WEEKS - 3)).put() subscription = MeetingSubscription(title='all engineering weekly', datetime=[pref_1, pref_2, pref_3]).put() user_pref1 = UserSubscriptionPreferences(preference=pref_1, subscription=subscription).put() user_pref2 = UserSubscriptionPreferences(preference=pref_2, subscription=subscription).put() user_pref3 = UserSubscriptionPreferences(preference=pref_3, subscription=subscription).put() meeting_spec1 = MeetingSpec(meeting_subscription=subscription, datetime=pref_1.get().datetime) meeting_spec1.put() meeting_spec2 = MeetingSpec(meeting_subscription=subscription, datetime=pref_2.get().datetime) meeting_spec2.put() meeting_spec3 = MeetingSpec(meeting_subscription=subscription, datetime=pref_3.get().datetime) meeting_spec3.put() users = [] num_users = 20 for i in range(0, num_users): user = User(email='{}@yelp.com'.format(i), metadata={ 'department': 'dept{}'.format(i)}, subscription_preferences=[user_pref1, user_pref2, user_pref3]) user.put() MeetingRequest(user=user.key, meeting_spec=meeting_spec1.key).put() MeetingRequest(user=user.key, meeting_spec=meeting_spec2.key).put() MeetingRequest(user=user.key, meeting_spec=meeting_spec3.key).put() users.append(user) meeting1 = Meeting(meeting_spec=meeting_spec1.key, cancelled=False).put() MeetingParticipant(meeting=meeting1, user=users[1].key).put() MeetingParticipant(meeting=meeting1, user=users[0].key).put() meeting2 = Meeting(meeting_spec=meeting_spec2.key, cancelled=False).put() MeetingParticipant(meeting=meeting2, user=users[1].key).put() MeetingParticipant(meeting=meeting2, user=users[0].key).put() meeting3 = Meeting(meeting_spec=meeting_spec3.key, cancelled=False).put() MeetingParticipant(meeting=meeting3, user=users[1].key).put() MeetingParticipant(meeting=meeting3, user=users[0].key).put() for run in range(10): matches, unmatched = generate_meetings(users, meeting_spec1, prev_meeting_tuples=None, group_size=3) assert(len(matches) == 6) assert (len(unmatched) == 2) for matched_group in matches: assert(not (users[0] in matched_group and users[1] in matched_group))
def test_filter_subscriptions_by_user_data_none_when_rules_exist(database): rule = Rule(name="department", value="b").put() database.sub.user_rules = [rule] database.sub.rule_logic = 'none' database.sub.put() preference = UserSubscriptionPreferences( subscription=database.sub.key, preference=database.prefs[0].key, ) preference.put() user = User(email='*****@*****.**', subscription_preferences=[preference.key], metadata={'department': 'a'}) user.put() merged_preferences = merge_subscriptions_with_preferences(user) subscriptions = filter_subscriptions_by_user_data(merged_preferences, user) assert len(subscriptions) == 1 assert subscriptions[0]['id'] == database.sub.key.urlsafe()
def test_send_batch_meeting_confirmation_email(database): pref = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) pref.put() s3_url = 'https://s3-media2.fl.yelpcdn.com/assets/srv0/yelp_large_assets/' user_a = User(email='*****@*****.**', photo_url=s3_url + 'a315bcce34b3/assets/img/illustrations/mascots/hammy.png', first_name='Hammy', last_name='Yelp', metadata={'department': 'Engineering'}, subscription_preferences=[pref.key]) user_b = User(first_name='Darwin', last_name='Yelp', email='*****@*****.**', photo_url=s3_url + '36a31704362e/assets/img/illustrations/mascots/darwin.png', metadata={'department': 'Design'}, subscription_preferences=[pref.key]) user_c = User(first_name='Carmin', last_name='Yelp', email='*****@*****.**', photo_url=s3_url + 'd71947670be7/assets/img/illustrations/mascots/carmen.png', metadata={'department': 'Design'}, subscription_preferences=[pref.key]) user_a.put() user_b.put() user_c.put() matches = [tuple((user_a, user_b, user_c, pref))] send_batch_meeting_confirmation_email(matches, database.specs[0])
def test_get_meeting_participants(app, database): pref = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) pref.put() user1 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key]) user1.put() user2 = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key]) user2.put() meeting1 = Meeting(meeting_spec=database.specs[0].key, cancelled=False).put() MeetingParticipant(meeting=meeting1, user=user1.key).put() MeetingParticipant(meeting=meeting1, user=user2.key).put() with app.test_request_context('/v1/metrics/meeting_participants'): participants = meeting_participants().json assert len(participants) == 2 assert set(participants[0].keys()) == set( ['date', 'meeting', 'meeting_title', 'participant', 'time']) assert participants[0]['date'] == database.specs[0].datetime.isoformat( ) participant_lookup = { participant['participant']: participant for participant in participants } assert participant_lookup[user1.email] == { 'date': database.specs[0].datetime.isoformat(), 'meeting': meeting1.urlsafe(), 'meeting_title': database.specs[0].meeting_subscription.get().title, 'participant': user1.email, 'time': '04:00PM', }
def test_get_meeting_participants(app, database): pref = UserSubscriptionPreferences(subscription=database.sub.key, preference=database.prefs[0].key) pref.put() user1 = User( email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key] ) user1.put() user2 = User( email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[pref.key] ) user2.put() meeting1 = Meeting(meeting_spec=database.specs[0].key, cancelled=False).put() MeetingParticipant(meeting=meeting1, user=user1.key).put() MeetingParticipant(meeting=meeting1, user=user2.key).put() with app.test_request_context('/v1/metrics/meeting_participants'): participants = json.loads(meeting_participants()) assert len(participants) == 2 assert set(participants[0].keys()) == set(['date', 'meeting', 'meeting_title', 'participant', 'time']) assert participants[0]['date'] == database.specs[0].datetime.isoformat() assert participants[0]['time'] == database.specs[0].datetime.strftime('%I:%M%p')
def test_remove_preferences_multiple_remove_on_opt_in(minimal_database, subscription): preference_1 = subscription.datetime[0] preference_2 = subscription.datetime[1] user_pref_1 = UserSubscriptionPreferences( preference=preference_1, subscription=subscription.key, ).put() user_pref_2 = UserSubscriptionPreferences( preference=preference_2, subscription=subscription.key, ).put() user = User(email='*****@*****.**', metadata={'department': 'dept'}, subscription_preferences=[user_pref_1, user_pref_2]) user.put() assert user.subscription_preferences == [user_pref_1, user_pref_2] updated_preferences = {preference_1: False, preference_2: False} removed = remove_preferences(user, updated_preferences, subscription.key) assert removed == {user_pref_1, user_pref_2} user = user.key.get() assert user.subscription_preferences == [] assert UserSubscriptionPreferences.query().fetch() == []