[IMP] estate: adding tests

Author: jucop-odoo

estate/tests/__init__.py

@@ -0,0 +1,3 @@
+from . import common
+from . import test_estate_property
+from . import test_estate_property_offer

estate/tests/common.py

@@ -0,0 +1,64 @@
+import datetime
+
+from odoo.addons.base.tests.common import BaseCommon
+
+
+class EstateTestCommon(BaseCommon):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+
+        cls.partner_a = cls.env['res.partner'].create({
+            'name': 'Test Partner A',
+        })
+
+        cls.property_type_house = cls.env['estate.property.type'].create({
+            'name': 'House',
+        })
+        cls.property_type_apartment = cls.env['estate.property.type'].create({
+            'name': 'Apartment',
+        })
+
+    def create_property(self, state, **kwargs):
+        vals = {
+            'name': 'Property ' + str(datetime.datetime.now()),
+            'postcode': '12345',
+            'date_availability': datetime.datetime.now() + datetime.timedelta(days=7),
+            'expected_price': 100000,
+            'bedrooms': 2,
+            'living_area': 120,
+            'facades': 2,
+            'property_type_id': self.property_type_house.id,
+            'state': 'new'
+        }
+
+        vals.update(kwargs)
+
+        estate_property = self.env['estate.property'].create(vals)
+
+        if state in {'offer_received', 'offer_accepted', 'sold', 'canceled'}:
+            offer_a = self.create_offer(estate_property, 5000, partner_id=self.partner.id, validity=7)
+            offer_b = self.create_offer(estate_property, 10000, partner_id=self.partner_a.id, validity=14)
+
+            if state in {'offer_accepted', 'sold'}:
+                offer_a.action_mark_as_refused()
+                offer_b.action_mark_as_accepted()
+
+                if state == 'sold':
+                    estate_property.action_mark_as_sold()
+            elif state == 'canceled':
+                estate_property.action_mark_as_canceled()
+
+        return estate_property
+
+    def create_offer(self, estate_property, price_increase=0, **kwargs):
+        vals = {
+            'price': estate_property.expected_price + price_increase,
+            'partner_id': self.partner.id,
+            'property_id': estate_property.id,
+            'validity': 7,
+        }
+
+        vals.update(kwargs)
+
+        return self.env['estate.property.offer'].create(vals)

estate/tests/test_estate_property.py

@@ -0,0 +1,39 @@
+from odoo.addons.estate.tests.common import EstateTestCommon
+from odoo.exceptions import UserError
+from odoo.tests import tagged, Form
+
+
+@tagged('post_install', '-at_install')
+class EstatePropertyTestCase(EstateTestCommon):
+    def test_sell_property_with_no_offers(self):
+        estate_property = self.create_property('offer_received')
+
+        with self.assertRaises(UserError):
+            estate_property.action_mark_as_sold()
+
+    def test_sell_property_on_state_new(self):
+        estate_property = self.create_property('new')
+
+        with self.assertRaises(UserError):
+            estate_property.action_mark_as_sold()
+
+    def test_garden_reactivity(self):
+        estate_property = self.create_property('new')
+
+        with Form(estate_property) as form_estate_property:
+            form_estate_property.garden = True
+            form_estate_property.save()
+            self.assertEqual(estate_property.garden_area, 10)
+            self.assertEqual(estate_property.garden_orientation, 'north')
+
+            form_estate_property.garden = False
+            form_estate_property.save()
+            self.assertEqual(estate_property.garden_area, 0)
+            self.assertEqual(estate_property.garden_orientation, False)
+
+            form_estate_property.garden = True
+            form_estate_property.garden_area = 100
+            form_estate_property.garden_orientation = 'south'
+            form_estate_property.save()
+            self.assertEqual(estate_property.garden_area, 100)
+            self.assertEqual(estate_property.garden_orientation, 'south')

estate/tests/test_estate_property_offer.py

@@ -0,0 +1,26 @@
+from odoo.addons.estate.tests.common import EstateTestCommon
+from odoo.exceptions import UserError
+from odoo.tests import tagged
+
+
+@tagged('post_install', '-at_install')
+class EstatePropertyOfferTestCase(EstateTestCommon):
+    def test_stop_offer_creation_on_sold_property(self):
+        estate_property = self.create_property('sold')
+
+        with self.assertRaises(UserError):
+            self.create_offer(estate_property, 50000)
+
+    def test_offer_price_too_low_compared_to_other_offers(self):
+        estate_property = self.create_property('offer_received')
+
+        with self.assertRaises(UserError):
+            self.create_offer(estate_property, 0)
+
+    def test_offer_price_too_low_compared_to_expected_price(self):
+        estate_property = self.create_property('new')
+        offer = self.create_offer(estate_property, -estate_property.expected_price / 2)
+
+        with self.assertRaises(UserError):
+            offer.action_mark_as_accepted()
+            estate_property.action_mark_as_sold()