diff --git a/zerver/tests/test_events.py b/zerver/tests/test_events.py index 6560e35eb5904..8d314a75d8e08 100644 --- a/zerver/tests/test_events.py +++ b/zerver/tests/test_events.py @@ -39,6 +39,7 @@ do_deactivate_stream, do_deactivate_user, do_reactivate_user, + do_refer_friend, do_regenerate_api_key, do_remove_alert_words, do_remove_realm_emoji, @@ -577,6 +578,19 @@ def test_pointer_events(self): error = schema_checker('events[0]', events[0]) self.assert_on_error(error) + def test_referral_events(self): + # type: () -> None + schema_checker = check_dict([ + ('type', equals('referral')), + ('referrals', check_dict([ + ('granted', check_int), + ('used', check_int), + ])), + ]) + events = self.do_test(lambda: do_refer_friend(self.user_profile, "friend@example.com")) + error = schema_checker('events[0]', events[0]) + self.assert_on_error(error) + def test_register_events(self): # type: () -> None realm_user_add_checker = check_dict([