From edbd0e863539bf09a30164cf189586e0d07242a1 Mon Sep 17 00:00:00 2001
From: Raimon Esteve <raimon@nan-tic.com>
Date: Fri, 26 May 2017 13:46:50 +0200
Subject: [PATCH] Add test

---
 tests/test_party_company.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/tests/test_party_company.py b/tests/test_party_company.py
index f6ef450..1142859 100644
--- a/tests/test_party_company.py
+++ b/tests/test_party_company.py
@@ -5,12 +5,40 @@ import unittest
 import trytond.tests.test_tryton
 from trytond.tests.test_tryton import ModuleTestCase, with_transaction
 from trytond.pool import Pool
+from trytond.modules.company.tests import create_company, set_company
 
 
 class PartyCompanyTestCase(ModuleTestCase):
     'Test Party Company module'
     module = 'party_company'
 
+    @with_transaction()
+    def test_party(self):
+        'Create party'
+        pool = Pool()
+        Party = pool.get('party.party')
+
+        party1, = Party.create([{
+                    'name': 'Party 1',
+                    }])
+        self.assert_(party1.id)
+        self.assertEqual(party1.company, None)
+
+    @with_transaction()
+    def test_party_company(self):
+        'Create party company'
+        pool = Pool()
+        Party = pool.get('party.party')
+
+        company = create_company()
+        with set_company(company):
+            party2, = Party.create([{
+                        'name': 'Party 2',
+                        }])
+            self.assert_(party2.id)
+            self.assertNotEqual(party2.company, None)
+
+
 def suite():
     suite = trytond.tests.test_tryton.suite()
     suite.addTests(unittest.TestLoader().loadTestsFromTestCase(