Fix missing method error (#425)

ascoderu · May 3, 2020 · d4ef260 · d4ef260
1 parent 950cc46
commit d4ef260
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/opwen_email_server/actions.py b/opwen_email_server/actions.py
@@ -277,7 +277,7 @@ def _action(self, resource_id):  # type: ignore
             formatted_email_id = new_email_id(formatted_email)
             formatted_email['_uid'] = formatted_email_id
 
-            self._email_storage.store_email(formatted_email, formatted_email_id)
+            self._email_storage.store_object(formatted_email_id, formatted_email)
 
             self._next_task(formatted_email_id)
 

diff --git a/tests/opwen_email_server/test_actions.py b/tests/opwen_email_server/test_actions.py
@@ -344,7 +344,7 @@ def test_200(self):
 
         _, status = self._execute_action(resource_id)
         self.assertEqual(status, 200)
-        self.email_storage.store_email.assert_called_once()
+        self.email_storage.store_object.assert_called_once()
         self.next_task.assert_called_once()
 
     def _execute_action(self, *args, **kwargs):