More unit test updates

inventree · Jun 11, 2024 · 4f210ad · 4f210ad
1 parent b6f4f8a
commit 4f210ad
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/test/test_build.py b/test/test_build.py
@@ -70,7 +70,8 @@ def test_build_attachment(self):
 
         response = build.uploadAttachment(fn, comment='A self referencing upload!')
 
-        self.assertEqual(response['build'], build.pk)
+        self.assertEqual(response['model_type'], 'build')
+        self.assertEqual(response['model_id'], build.pk)
         self.assertEqual(response['comment'], 'A self referencing upload!')
 
         self.assertEqual(len(build.getAttachments()), n + 1)

diff --git a/test/test_order.py b/test/test_order.py
@@ -384,7 +384,7 @@ def test_po_attachment(self):
             po.uploadAttachment('not_found.txt')
 
         # Check that attachments uploaded OK
-        attachments = po.getAttachments(self.api)
+        attachments = po.getAttachments()
         self.assertEqual(len(attachments), 3)
 
 
@@ -519,7 +519,8 @@ def test_so_attachment(self):
 
         attachment = Attachment(self.api, pk=pk)
 
-        self.assertEqual(attachment.order, so.pk)
+        self.assertEqual(attachment.modeL_type, 'salesorder')
+        self.assertEqual(attachment.model_id, so.pk)
         self.assertEqual(attachment.comment, 'Sales order attachment')
 
         attachments = so.getAttachments()