all tests for song

features/Songs/tests/test_song.py

@@ -1,5 +1,5 @@
 
-from features.Songs.song import resolve_song, resolve_songs, create_song
+from features.Songs.song import resolve_song, resolve_songs, create_song, update_song
 from aws_resources.mock_dynamo import setup_mocks, mock_song, another_mock_song, mock_create_song
 from moto import mock_dynamodb2
 
@@ -19,4 +19,12 @@ def test_resolve_song():
 def test_create_song():
  setup_mocks()
  response = create_song({}, {}, mock_create_song)
- assert response['message'] == 'success' and response['song']['title'] == "It don't mean a thing"
+ assert response['message'] == 'success' and response['song']['title'] == 'It don\'t mean a thing'
+
+@mock_dynamodb2
+def test_update_song():
+ setup_mocks()
+ new_attr = {'title': 'new song title'}
+ updated_song = {**mock_song, **new_attr}
+ response = update_song({}, {}, mock_song)
+ assert response['message'] == 'success'