# 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

us = market.from_territory('US')
us.price_adjustments = 1.05
us.exchange_rate = 0.119800

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_wallpaper_price_us(self):
        self.assertEqual(59, wallpaper_price(width=100, height=100,
                                             material_price=236, market=us), "Minium price for US is 2 sqm")

    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))