def test_add_favorite_304(self): favorite = Favorite( parent=ndb.Key(Account, 'user_id'), user_id='user_id', model_key='frc7332', model_type=ModelType.TEAM) favorite.put() # Favorite already exists - return a 304 with patch.object(NotificationHelper, 'send_favorite_update') as mock_send_favorite_update: self.assertEqual(MyTBAHelper.add_favorite(favorite), 304) mock_send_favorite_update.assert_not_called()
def test_remove_favorite_200_sending_device_key(self): favorite = Favorite( parent=ndb.Key(Account, 'user_id'), user_id='user_id', model_key='frc7332', model_type=ModelType.TEAM) favorite_key = favorite.put() # Favorite does not exist - 404 with patch.object(NotificationHelper, 'send_favorite_update') as mock_send_favorite_update, \ patch.object(ndb, 'delete_multi') as mock_delete: self.assertEqual(MyTBAHelper.remove_favorite(favorite.user_id, favorite.model_key, favorite.model_type, 'sending_device_key'), 200) mock_delete.assert_called_once_with([favorite_key]) mock_send_favorite_update.assert_called_once_with('user_id', 'sending_device_key')