Fix imports

OpenG2P · Nov 28, 2024 · 505cfea · 505cfea
1 parent e4c23ed
commit 505cfea
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/g2p_social_registry/tests/test_g2p_que_id_generation.py b/g2p_social_registry/tests/test_g2p_que_id_generation.py
@@ -1,5 +1,6 @@
 from odoo.tests import TransactionCase
 from odoo.exceptions import ValidationError
+from odoo import fields
 
 
 class TestG2PQueIDGeneration(TransactionCase):
@@ -25,7 +26,7 @@ def test_create_g2p_queue_record(self):
         """
         self.assertTrue(self.g2p_queue, "G2PQueIDGeneration record was not created.")
         self.assertEqual(
-            self.g2p_queue.registrant_id, self.partner1,
+            self.g2p_queue.registrant_id, self.partner1.id,
             "Registrant ID does not match the expected partner."
         )
         self.assertEqual(
@@ -95,7 +96,7 @@ def test_multiple_g2p_queue_records(self):
 
         self.assertTrue(record2, "Second G2PQueIDGeneration record was not created.")
         self.assertEqual(
-            record2.registrant_id, self.partner2,
+            record2.registrant_id, self.partner2.id,
             "Registrant ID for the second record does not match the expected partner."
         )
         self.assertEqual(