Fix for unit test

inventree · Mar 29, 2023 · 01f64c4 · 01f64c4
1 parent 061a336
commit 01f64c4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/test_company.py b/test/test_company.py
@@ -23,10 +23,12 @@ class ContactTest(InvenTreeTestCase):
     def test_contact_create(self):
         """Test that we can create a new contact"""
 
+        cmp = company.Company.list(self.api, limit=1)[0]
         n = company.Contact.count(self.api)
 
         for idx in range(3):
             company.Contact.create(self.api, data={
+                'company': cmp.pk,
                 'name': f"Random Name {idx}",
                 'role': f"Random Role {idx}",
             })