def test_get_invite_by_user(self): with app.test_request_context(): notification = ObjectMother.get_notification() save_to_db(notification, 'Save') invite = DataGetter.get_invite_by_user_id(1) self.assertFalse(invite)
def test_get_user_notification(self): with app.test_request_context(): notification = ObjectMother.get_notification() save_to_db(notification, 'Save') self.assertTrue(DataGetter.get_user_notification(1))