merge P5-567

tests/factories.py

@@ -0,0 +1,30 @@
+import factory
+from api import models
+
+
+class DesignerFactory(factory.Factory):
+
+    class Meta:
+        model = models.Designer
+
+    name = 'Spiderman'
+    username = 'spiderman'
+    territory = 'US'
+    currency = 'USD'
+
+
+class ContractCustomerFactory(factory.Factory):
+
+    class Meta:
+        model = models.ContractCustomer
+
+    pricepremium = 0
+
+
+class InquiryFactory(factory.Factory):
+
+    class Meta:
+        model = models.Inquiry
+
+    price_retouch = None
+    pricepremium = 0

tests/test_price.py

@@ -0,0 +1,66 @@
+# coding=UTF-8
+
+import unittest2
+from api.lib.prices import wallpaper_price, canvas_price, canvas_cm2_price
+from tests.factories import DesignerFactory, ContractCustomerFactory, InquiryFactory
+from api.models import market
+
+
+sweden = market.from_territory('SE')
+sweden.price_adjustments = 1
+sweden.exchange_rate = 1
+
+denmark = market.from_territory('DK')
+denmark.price_adjustments = 1.185
+denmark.exchange_rate = 0.799000
+
+designer = DesignerFactory()
+designer.pricepremium = 17.647
+reseller = ContractCustomerFactory(pricepremium=15)
+
+
+class TestPrice(unittest2.TestCase):
+
+    def test_wallpaper_price(self):
+        self.assertEqual(295, wallpaper_price(
+            width=1, height=1, material_price=236, market=sweden))
+        self.assertEqual(295, wallpaper_price(
+            width=100, height=100, material_price=236, market=sweden))
+        self.assertEqual(1180, wallpaper_price(
+            width=200, height=200, material_price=236, market=sweden))
+        self.assertEqual(325, wallpaper_price(
+            width=100, height=100, material_price=260, market=sweden))
+        self.assertEqual(279, wallpaper_price(
+            width=100, height=100, material_price=236, market=denmark))
+        self.assertEqual(236, wallpaper_price(
+            width=100, height=100, material_price=236, market=sweden, inc_vat=False))
+
+        # designer
+        self.assertEqual(347, wallpaper_price(
+            width=100, height=100, material_price=236, market=sweden, designer=designer))
+        self.assertEqual(329, wallpaper_price(
+            width=100, height=100, material_price=236, market=denmark, designer=designer))
+
+        # reseller stores
+        self.assertEqual(339, wallpaper_price(
+            width=100, height=100, material_price=236, market=sweden, reseller=reseller))
+        self.assertEqual(399, wallpaper_price(width=100, height=100,
+                                              material_price=236, market=sweden, designer=designer, reseller=reseller))
+        self.assertEqual(154, wallpaper_price(width=100, height=100, material_price=236,
+                                              market=sweden, reseller=ContractCustomerFactory(pricepremium=-47.74)))
+
+    def test_canvas_cm2_price(self):
+        self.assertEqual(0.22, canvas_cm2_price(250))
+        self.assertEqual(0.22384, canvas_cm2_price(10))
+        self.assertEqual(0.208, canvas_cm2_price(1000))
+
+    def test_canvas_price(self):
+        self.assertEqual(460, canvas_price(width=50, height=50, market=sweden, framed=True, inc_vat=False))
+        self.assertEqual(960, canvas_price(width=100, height=100, market=sweden, framed=True, inc_vat=False))
+        self.assertEqual(1359, canvas_price(width=150, height=120, market=sweden, framed=True, inc_vat=False))
+        self.assertEqual(256000, canvas_price(width=2000, height=2000, market=sweden, framed=True, inc_vat=False))
+        self.assertEqual(397, canvas_price(width=10, height=10, market=sweden, framed=True, inc_vat=False))
+        self.assertEqual(672, canvas_price(width=100, height=100, market=sweden, framed=False, inc_vat=False))
+        self.assertEqual(840, canvas_price(width=100, height=100, market=sweden, framed=False, inc_vat=True))
+
+        self.assertEqual(436, canvas_price(width=50, height=50, market=denmark, framed=True, inc_vat=False))