Fix KeyError in spark reader (#1369)

* Fix key name * Fix tests as well
metabrainz · Apr 2, 2021 · 792e3d6 · 792e3d6
1 parent 0f37c15
commit 792e3d6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/listenbrainz/spark/handlers.py b/listenbrainz/spark/handlers.py
@@ -348,7 +348,7 @@ def notify_artist_relation_import(data):
     if current_app.config['TESTING']:
         return
 
-    artist_relation_name = data['import_artist_relation']
+    artist_relation_name = data['imported_artist_relation']
     import_time = data['import_time']
     time_taken_to_import = data['time_taken_to_import']
 

diff --git a/listenbrainz/spark/test_handlers.py b/listenbrainz/spark/test_handlers.py
@@ -432,7 +432,7 @@ def test_notify_artist_relation_import(self, mock_send_mail):
             # testing, should not send a mail
             self.app.config['TESTING'] = True
             notify_artist_relation_import({
-                'import_artist_relation': artist_relation_name,
+                'imported_artist_relation': artist_relation_name,
                 'import_time': str(import_time),
                 'time_taken_to_import': str(time_taken_to_import),
             })
@@ -441,7 +441,7 @@ def test_notify_artist_relation_import(self, mock_send_mail):
             # in prod now, should send it
             self.app.config['TESTING'] = False
             notify_artist_relation_import({
-                'import_artist_relation': artist_relation_name,
+                'imported_artist_relation': artist_relation_name,
                 'import_time': str(import_time),
                 'time_taken_to_import': str(time_taken_to_import),
             })