Unit test updates

inventree · Jun 14, 2024 · 60ea8b4 · 60ea8b4
1 parent 74df85b
commit 60ea8b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/inventree/company.py b/inventree/company.py
@@ -26,6 +26,7 @@ class Company(inventree.base.ImageMixin, inventree.base.MetadataMixin, inventree
     """ Class representing the Company database model """
 
     URL = 'company'
+    MODEL_TYPE = "company"
 
     def getContacts(self, **kwargs):
         """Return contacts associated with this Company"""
@@ -122,6 +123,7 @@ class ManufacturerPart(
     """
 
     URL = 'company/part/manufacturer'
+    MODEL_TYPE = "manufacturerpart"
 
     def getParameters(self, **kwargs):
         """

diff --git a/test/test_order.py b/test/test_order.py
@@ -519,7 +519,7 @@ def test_so_attachment(self):
 
         attachment = Attachment(self.api, pk=pk)
 
-        self.assertEqual(attachment.modeL_type, 'salesorder')
+        self.assertEqual(attachment.model_type, 'salesorder')
         self.assertEqual(attachment.model_id, so.pk)
         self.assertEqual(attachment.comment, 'Sales order attachment')