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')
