# coding=UTF-8

import unittest2
from api.lib.prices import *
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))

    def test_canvas_frame_price(self):
        self.assertAlmostEqual(208.46, canvas_frame_price(
            50, 50, stock_price=1.04230824, additional_price=94.750000), places=2)
        self.assertAlmostEqual(177.19, canvas_frame_price(
            1, 1, stock_price=1.04230824, additional_price=94.750000), places=2)
        self.assertAlmostEqual(625.38, canvas_frame_price(
            500, 500, stock_price=1.04230824, additional_price=None), places=2)

    def test_old_canvas_diy_frame_price(self):
        self.assertEqual(
            403, old_canvas_diy_frame_price(50, 50, market=sweden))
        self.assertEqual(801, old_canvas_diy_frame_price(
            100, 110, market=denmark))
        self.assertEqual(463, old_canvas_diy_frame_price(
            50, 50, market=sweden, reseller=reseller))
        self.assertEqual(1327, old_canvas_diy_frame_price(
            500, 500, market=sweden))

    def test_old_canvas_price(self):
        self.assertEqual(250, old_canvas_price(
            width=50, height=50, material_price=799.2, market=sweden, framed=False))

        self.assertEqual(200, old_canvas_price(
            width=1, height=1, material_price=799.2, market=sweden, framed=False))

        self.assertEqual(7493, old_canvas_price(
            width=500, height=500, material_price=799.2, market=sweden, framed=False))

        self.assertEqual(5994, old_canvas_price(width=500, height=500,
                                                material_price=799.2, market=sweden, framed=False, inc_vat=False))
        self.assertEqual(7493, old_canvas_price(
            width=500, height=500, material_price=799.2, market=sweden, framed=False))
        self.assertEqual(236, old_canvas_price(
            width=50, height=50, material_price=799.2, market=denmark, framed=False))
        self.assertEqual(294, old_canvas_price(
            width=50, height=50, material_price=799.2, market=sweden, designer=designer, framed=False))
        self.assertEqual(510, old_canvas_price(
            width=50, height=50, material_price=799.2, market=sweden, framed=True))
        self.assertEqual(2972, old_canvas_price(
            width=150, height=140, material_price=799.2, market=sweden, framed=True))
        self.assertEqual(287, old_canvas_price(
            width=50, height=50, material_price=799.2, market=sweden, framed=False, reseller=reseller))