500 lines
16 KiB
Python
500 lines
16 KiB
Python
# coding=UTF-8
|
|
|
|
import unittest2
|
|
import mock
|
|
from api.lib.prices import *
|
|
from tests.factories import (
|
|
DesignerFactory,
|
|
ContractCustomerFactory,
|
|
InquiryFactory,
|
|
MarketFactory,
|
|
)
|
|
from api.models import Inquiry, Product, Material
|
|
|
|
sweden = MarketFactory.build(name="SE", exchange_rate=1, price_adjustments=1, vat=1.25)
|
|
denmark = MarketFactory.build(
|
|
name="DK", price_adjustments=1.185, exchange_rate=0.799000, vat=1.25
|
|
)
|
|
us = MarketFactory.build(name="US", price_adjustments=1.06, exchange_rate=0.1198, vat=1)
|
|
nl = MarketFactory.build(name="NL", price_adjustments=1.15, exchange_rate=0.1, vat=1.21)
|
|
eu = MarketFactory.build(name="EU", price_adjustments=1.13, exchange_rate=0.1, vat=1.25)
|
|
|
|
designer = DesignerFactory.build(
|
|
pricepremium_wallpaper=17.647,
|
|
pricepremium_canvas=8,
|
|
pricepremium_poster=15.50,
|
|
pricepremium_framed_print=13,
|
|
)
|
|
|
|
|
|
reseller = ContractCustomerFactory.build(pricepremium=15)
|
|
|
|
designer_moomin = DesignerFactory.build(
|
|
pricepremium_wallpaper=2,
|
|
pricepremium_canvas=3,
|
|
pricepremium_poster=4,
|
|
pricepremium_framed_print=7,
|
|
)
|
|
|
|
standard_wallpaper = Material(name="standard-wallpaper", price=23600)
|
|
premium_wallpaper = Material(name="premium-wallpaper", price=26000)
|
|
standard_canvas = Material(name="standard-canvas", price=79920)
|
|
|
|
|
|
class TestPrice(unittest2.TestCase):
|
|
def test_m2_price(self):
|
|
self.assertEqual(
|
|
236,
|
|
wallpaper_m2_price(
|
|
material=standard_wallpaper, market=sweden, inc_vat=False
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
260,
|
|
wallpaper_m2_price(
|
|
material=premium_wallpaper, market=sweden, inc_vat=False
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
295,
|
|
wallpaper_m2_price(
|
|
material=standard_wallpaper, market=sweden, inc_vat=True
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
223,
|
|
wallpaper_m2_price(
|
|
material=standard_wallpaper, market=denmark, inc_vat=False
|
|
),
|
|
)
|
|
inquiry = Inquiry(pricepremium=36.029411)
|
|
|
|
self.assertEqual(
|
|
42,
|
|
wallpaper_m2_price(
|
|
material=standard_wallpaper,
|
|
market=eu,
|
|
reseller=reseller,
|
|
inquiry=inquiry,
|
|
inc_vat=False,
|
|
),
|
|
)
|
|
|
|
def test_wallpaper_price(self):
|
|
self.assertEqual(
|
|
295,
|
|
wallpaper_price(
|
|
width=1, height=1, material=standard_wallpaper, market=sweden
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
295,
|
|
wallpaper_price(
|
|
width=100, height=100, material=standard_wallpaper, market=sweden
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
1180,
|
|
wallpaper_price(
|
|
width=200, height=200, material=standard_wallpaper, market=sweden
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
325,
|
|
wallpaper_price(
|
|
width=100, height=100, material=premium_wallpaper, market=sweden
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
279,
|
|
wallpaper_price(
|
|
width=100, height=100, material=standard_wallpaper, market=denmark
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
236,
|
|
wallpaper_price(
|
|
width=100,
|
|
height=100,
|
|
material=standard_wallpaper,
|
|
market=sweden,
|
|
inc_vat=False,
|
|
),
|
|
)
|
|
|
|
# designer
|
|
self.assertEqual(
|
|
347,
|
|
wallpaper_price(
|
|
width=100,
|
|
height=100,
|
|
material=standard_wallpaper,
|
|
market=sweden,
|
|
designer=designer,
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
329,
|
|
wallpaper_price(
|
|
width=100,
|
|
height=100,
|
|
material=standard_wallpaper,
|
|
market=denmark,
|
|
designer=designer,
|
|
),
|
|
)
|
|
|
|
# reseller stores
|
|
self.assertEqual(
|
|
339,
|
|
wallpaper_price(
|
|
width=100,
|
|
height=100,
|
|
material=standard_wallpaper,
|
|
market=sweden,
|
|
reseller=reseller,
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
399,
|
|
wallpaper_price(
|
|
width=100,
|
|
height=100,
|
|
material=standard_wallpaper,
|
|
market=sweden,
|
|
designer=designer,
|
|
reseller=reseller,
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
154,
|
|
wallpaper_price(
|
|
width=100,
|
|
height=100,
|
|
material=standard_wallpaper,
|
|
market=sweden,
|
|
reseller=ContractCustomerFactory(pricepremium=-47.74),
|
|
),
|
|
)
|
|
|
|
def test_wallpaper_price_us(self):
|
|
self.assertEqual(
|
|
60,
|
|
wallpaper_price(
|
|
width=100, height=100, material=standard_wallpaper, market=us
|
|
),
|
|
"Minium price for US is 2 sqm",
|
|
)
|
|
|
|
def test_canvas_cm2_price(self):
|
|
self.assertEqual(0.24099500000000001, canvas_cm2_price(250))
|
|
self.assertEqual(0.24589580000000003, canvas_cm2_price(10))
|
|
self.assertEqual(0.22568000000000002, canvas_cm2_price(1000))
|
|
|
|
def test_canvas_price(self):
|
|
self.assertEqual(
|
|
488,
|
|
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(
|
|
341,
|
|
canvas_price(
|
|
width=10, height=10, market=sweden, framed=True, inc_vat=False
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
427,
|
|
canvas_price(width=40, height=40, market=sweden, framed=True, inc_vat=True),
|
|
)
|
|
|
|
self.assertEqual(
|
|
299,
|
|
canvas_price(
|
|
width=40, height=40, market=sweden, framed=False, inc_vat=True
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
731,
|
|
canvas_price(
|
|
width=80, height=80, market=sweden, framed=False, inc_vat=True
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
1044,
|
|
canvas_price(width=80, height=80, market=sweden, framed=True, inc_vat=True),
|
|
)
|
|
|
|
self.assertEqual(
|
|
1260,
|
|
canvas_price(
|
|
width=150, height=150, market=sweden, framed=False, inc_vat=True
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
1800,
|
|
canvas_price(
|
|
width=150, height=150, market=sweden, framed=True, inc_vat=True
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
201,
|
|
canvas_price(width=150, height=150, market=us, framed=True, inc_vat=True),
|
|
)
|
|
|
|
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(
|
|
462,
|
|
canvas_price(
|
|
width=50, height=50, market=denmark, framed=True, inc_vat=False
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
181,
|
|
canvas_price(
|
|
width=150,
|
|
height=100,
|
|
market=us,
|
|
designer=designer_moomin,
|
|
inc_vat=False,
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
181,
|
|
canvas_price(
|
|
width=150, height=100, market=us, designer=designer_moomin, inc_vat=True
|
|
),
|
|
)
|
|
|
|
def test_canvas_frame_price(self):
|
|
self.assertEqual(
|
|
624, canvas_frame_price(width=100, height=100, market=sweden, inc_vat=False)
|
|
)
|
|
self.assertEqual(
|
|
780, canvas_frame_price(width=100, height=100, market=sweden, inc_vat=True)
|
|
)
|
|
self.assertEqual(88, canvas_frame_price(width=100, height=100, market=eu))
|
|
self.assertEqual(
|
|
936, canvas_frame_price(500, 500, market=sweden, inc_vat=False)
|
|
)
|
|
self.assertEqual(222, canvas_frame_price(1, 1, market=sweden, inc_vat=False))
|
|
self.assertEqual(
|
|
277, canvas_frame_price(width=40, height=40, market=sweden, inc_vat=True)
|
|
)
|
|
self.assertEqual(71, canvas_frame_price(width=40, height=120, market=us))
|
|
self.assertEqual(66, canvas_frame_price(width=40, height=100, market=us))
|
|
self.assertEqual(578, canvas_frame_price(width=40, height=120, market=sweden))
|
|
|
|
def test_wallpaper_custom_inquiry_price(self):
|
|
with mock.patch.object(Inquiry, "market", nl):
|
|
inquiry = Inquiry(
|
|
product_group="photo-wallpaper",
|
|
pricepremium=0,
|
|
width=350,
|
|
height=270,
|
|
price_effect=0,
|
|
price_retouch=160.00,
|
|
material=standard_wallpaper,
|
|
)
|
|
|
|
self.assertEqual(333, inquiry_price(inquiry))
|
|
self.assertEqual(310, inquiry_price(inquiry, inc_price_retouch=False))
|
|
self.assertEqual(160, inquiry.price_retouch)
|
|
self.assertEqual(18.4, inquiry.local_price_retouch(rounded=False))
|
|
self.assertEqual(
|
|
22, inquiry.local_price_retouch(rounded=True, inc_vat=True)
|
|
)
|
|
|
|
# test with different material
|
|
self.assertEqual(
|
|
342,
|
|
inquiry_price(
|
|
inquiry, inc_price_retouch=False, material=premium_wallpaper
|
|
),
|
|
)
|
|
self.assertEqual(364, inquiry_price(inquiry, material=premium_wallpaper))
|
|
|
|
def test_wallpaper_product_inquiry_price(self):
|
|
designer = DesignerFactory()
|
|
designer.pricepremium_wallpaper = 4
|
|
product = Product(designer=designer)
|
|
with mock.patch.object(Inquiry, "market", sweden):
|
|
inquiry = Inquiry(
|
|
product_group="photo-wallpaper",
|
|
pricepremium=0,
|
|
width=350,
|
|
height=280,
|
|
price_retouch=0,
|
|
material=standard_wallpaper,
|
|
product=product,
|
|
)
|
|
|
|
self.assertEqual(3007, inquiry_price(inquiry))
|
|
self.assertEqual(3007, inquiry_price(inquiry, inc_price_retouch=False))
|
|
self.assertEqual(3312, inquiry_price(inquiry, material=premium_wallpaper))
|
|
|
|
def test_wallpaper_product2_inquiry_price(self):
|
|
designer = DesignerFactory()
|
|
designer.pricepremium_wallpaper = 4
|
|
product = Product(designer=designer)
|
|
with mock.patch.object(Inquiry, "market", sweden):
|
|
inquiry = Inquiry(
|
|
product_group="photo-wallpaper",
|
|
pricepremium=25,
|
|
width=250,
|
|
height=400,
|
|
price_retouch=200,
|
|
material=standard_wallpaper,
|
|
product=product,
|
|
)
|
|
|
|
self.assertEqual(3835, inquiry_price(inquiry, inc_price_retouch=False))
|
|
self.assertEqual(200, inquiry.price_retouch)
|
|
self.assertEqual(
|
|
250, inquiry.local_price_retouch(rounded=True, inc_vat=True)
|
|
)
|
|
|
|
def test_canvas_custom_inquiry_price(self):
|
|
with mock.patch.object(Inquiry, "market", sweden):
|
|
inquiry = Inquiry(
|
|
product_group="canvas",
|
|
width=100,
|
|
height=100,
|
|
framed=True,
|
|
price_retouch=200,
|
|
pricepremium=0,
|
|
material=standard_canvas,
|
|
)
|
|
|
|
self.assertEqual(1200, inquiry_price(inquiry, inc_price_retouch=False))
|
|
self.assertEqual(1450, inquiry_price(inquiry, inc_price_retouch=True))
|
|
self.assertEqual(
|
|
840, inquiry_price(inquiry, inc_price_retouch=False, framed=False)
|
|
)
|
|
|
|
def test_package_price(self):
|
|
self.assertEqual(150, package_price("canvas", 150, 40, market=us))
|
|
self.assertEqual(0, package_price("canvas", 150, 40, market=sweden))
|
|
self.assertEqual(150, package_price("poster", 150, 40, market=us))
|
|
self.assertEqual(100, package_price("poster", 40, 40, market=us))
|
|
self.assertEqual(0, package_price("poster", 150, 40, market=sweden))
|
|
|
|
def test_poster_cm2_price(self):
|
|
self.assertEqual(0.100789248, poster_cm2_price(1))
|
|
self.assertEqual(0.0997248, poster_cm2_price(100))
|
|
self.assertEqual(0.090048, poster_cm2_price(1000))
|
|
self.assertEqual(0.04032, poster_cm2_price(10000))
|
|
self.assertEqual(0.028, poster_cm2_price(100000))
|
|
|
|
def test_poster_price(self):
|
|
self.assertEqual(
|
|
354, poster_price(width=30, height=40, hanger=True, market=sweden)
|
|
)
|
|
self.assertEqual(
|
|
250, poster_price(width=30, height=40, hanger=False, market=sweden)
|
|
)
|
|
self.assertEqual(
|
|
48,
|
|
poster_price(width=30, height=40, hanger=True, market=us),
|
|
"extra cost for us market",
|
|
)
|
|
self.assertEqual(
|
|
76,
|
|
poster_price(width=120, height=40, market=us),
|
|
"high extra cost for us market wide posters",
|
|
)
|
|
self.assertEqual(
|
|
78, poster_price(width=120, height=40, market=us, designer=designer_moomin)
|
|
)
|
|
|
|
def test_poster_hanger_price(self):
|
|
self.assertEqual(166, poster_hanger_price(30, market=sweden))
|
|
self.assertEqual(390, poster_hanger_price(150, market=sweden))
|
|
self.assertEqual(29, poster_hanger_price(30, market=us))
|
|
self.assertEqual(58, poster_hanger_price(150, market=us))
|
|
|
|
def test_poster_custom_inquiry_price(self):
|
|
with mock.patch.object(Inquiry, "market", sweden):
|
|
inquiry = Inquiry(
|
|
product_group="poster",
|
|
width=30,
|
|
height=40,
|
|
hanger=True,
|
|
price_retouch=200,
|
|
pricepremium=0,
|
|
)
|
|
|
|
self.assertEqual(354, inquiry_price(inquiry, inc_price_retouch=False))
|
|
self.assertEqual(604, inquiry_price(inquiry, inc_price_retouch=True))
|
|
self.assertEqual(
|
|
250, inquiry_price(inquiry, inc_price_retouch=False, hanger=False)
|
|
)
|
|
|
|
def test_framed_print_price(self):
|
|
self.assertEqual(598, framed_print_price(sku="GLOBAL-CFP-12x12", market=sweden))
|
|
self.assertEqual(
|
|
639,
|
|
framed_print_price(
|
|
sku="GLOBAL-CFP-12x12", market=sweden, designer=designer_moomin
|
|
),
|
|
)
|
|
self.assertEqual(
|
|
528, framed_print_price(sku="GLOBAL-CFP-12x12", market=denmark)
|
|
)
|
|
self.assertEqual(159, framed_print_price(sku="GLOBAL-CFP-28x40", market=eu))
|
|
self.assertEqual(208, framed_print_price(sku="GLOBAL-CFP-28x40", market=us))
|
|
|
|
def test_invalid_framed_print_code(self):
|
|
with self.assertRaises(ValueError) as context:
|
|
framed_print_price(sku="invalid", market=sweden)
|
|
|
|
self.assertTrue("Invalid SKU number" in str(context.exception))
|
|
|
|
def test_wallpaper_kit_price(self):
|
|
self.assertEqual(143.2, wallpaper_kit_price(market=sweden, rounded=False, inc_vat=False))
|
|
self.assertEqual(179.0, wallpaper_kit_price(market=sweden, rounded=False, inc_vat=True))
|
|
self.assertEqual(18.0, wallpaper_kit_price(market=us))
|
|
self.assertEqual(169.0, wallpaper_kit_price(market=denmark))
|