edan_f0/tests/unit/partner/price_tests.py

from django.test import TestCase
from decimal import Decimal as D

from oscar.core.prices import TaxNotKnown
from oscar.apps.partner import prices, models


class TestUnavailable(TestCase):

    def setUp(self):
        self.price = prices.Unavailable()

    def test_means_unknown_tax(self):
        self.assertFalse(self.price.is_tax_known)

    def test_means_prices_dont_exist(self):
        self.assertFalse(self.price.exists)

    def test_means_price_attributes_are_none(self):
        self.assertIsNone(self.price.incl_tax)
        self.assertIsNone(self.price.excl_tax)
        self.assertIsNone(self.price.tax)


class TestDelegateToStockRecord(TestCase):

    def setUp(self):
        self.record = models.StockRecord(
            price_excl_tax=D('12.99'))
        self.price = prices.DelegateToStockRecord(self.record)

    def test_assumes_tax_is_known(self):
        self.assertTrue(self.price.is_tax_known)

    def test_has_correct_price(self):
        self.assertEquals(self.record.price_excl_tax,
                          self.price.excl_tax)


class TestFixedPriceWithoutTax(TestCase):

    def setUp(self):
        self.price = prices.FixedPrice('GBP', D('9.15'))

    def test_means_unknown_tax(self):
        self.assertFalse(self.price.is_tax_known)

    def test_has_correct_price(self):
        self.assertEquals(D('9.15'), self.price.excl_tax)

    def test_raises_exception_when_asking_for_price_incl_tax(self):
        with self.assertRaises(TaxNotKnown):
            self.price.incl_tax