From 3a380cc91efc464005a8f42dfc7d77c9ec0ebd93 Mon Sep 17 00:00:00 2001
From: Andrea Spacca <andrea.spacca@gmail.com>
Date: Tue, 22 Dec 2020 21:00:09 +0100
Subject: [PATCH] fix test

---
 pantalaimon/store.py | 8 ++++----
 tests/store_test.py  | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/pantalaimon/store.py b/pantalaimon/store.py
index 4e93f71..123f80f 100644
--- a/pantalaimon/store.py
+++ b/pantalaimon/store.py
@@ -182,12 +182,12 @@ class PanStore:
             return None
 
     @use_database
-    def save_upload(self, content_uri, media):
+    def save_upload(self, content_uri, upload):
         PanUploadInfo.insert(
             content_uri=content_uri,
-            key=media["key"],
-            iv=media["iv"],
-            hashes=media["hashes"],
+            key=upload["key"],
+            iv=upload["iv"],
+            hashes=upload["hashes"],
         ).on_conflict_ignore().execute()
 
     @use_database
diff --git a/tests/store_test.py b/tests/store_test.py
index 5ea5f96..218575a 100644
--- a/tests/store_test.py
+++ b/tests/store_test.py
@@ -185,7 +185,7 @@ class TestClass(object):
 
         upload = UploadInfo(event.url, event.key, event.iv, event.hashes)
 
-        panstore.save_upload(event.url)
+        panstore.save_upload(event.url, {"key": event.key, "iv": event.iv, "hashes": event.hashes})
 
         upload_cache = panstore.load_upload(event.url)