edan_f0/tests/unit/basket_tests.py

import datetime
from decimal import Decimal as D

from django.test import TestCase
from django.test.client import RequestFactory

from oscar.apps.basket.models import Basket
from oscar.apps.basket.middleware import BasketMiddleware
from oscar.test import factories
from oscar.apps.basket.reports import (
    OpenBasketReportGenerator, SubmittedBasketReportGenerator)
from oscar.apps.catalogue.models import Option


class TestABasket(TestCase):

    def setUp(self):
        self.basket = Basket()
        self.product = factories.create_product()
        self.record = factories.create_stockrecord(
            self.product, price_excl_tax=D('10.00'))
        self.stockinfo = factories.create_stockinfo(self.record)

    def test_an_empty_basket_has_zero_lines(self):
        self.assertEqual(0, self.basket.num_lines)

    def test_new_baskets_are_empty(self):
        self.assertTrue(self.basket.is_empty)

    def test_adding_product_creates_line(self):
        self.basket.add_product(self.product, self.stockinfo)
        self.assertEqual(1, self.basket.num_lines)

    def test_adding_multiproduct_line_returns_correct_number_of_items(self):
        self.basket.add_product(self.product, self.stockinfo, 10)
        self.assertEqual(self.basket.num_items, 10)
        self.assertEqual(self.basket.num_lines, 1)

    def test_add_product_creates_line(self):
        self.basket.add_product(self.product, self.stockinfo)
        self.assertTrue(self.basket.num_lines == 1)

    def test_add_product_sets_line_prices(self):
        self.basket.add_product(self.product, self.stockinfo)
        basket_line = self.basket.lines.all()[0]
        self.assertEqual(basket_line.price_incl_tax, D('10.00'))
        self.assertEqual(basket_line.price_excl_tax, D('10.00'))

    def test_flushing_basket_removes_all_lines(self):
        self.basket.add_product(self.product, self.stockinfo, 10)
        self.assertEqual(self.basket.num_items, 10)
        self.basket.flush()
        self.assertEqual(self.basket.num_items, 0)

    def test_returns_correct_quantity_for_missing_product(self):
        self.assertEqual(0, self.basket.line_quantity(
            self.product, self.record))

    def test_returns_correct_quantity_for_existing_product(self):
        self.basket.add_product(self.product, self.stockinfo)
        self.assertEqual(1, self.basket.line_quantity(
            self.product, self.record))

    def test_returns_correct_quantity_for_existing_product_with_options(self):
        option = Option.objects.create(name="Message")
        options = [{"option": option, "value": "2"}]
        self.basket.add_product(self.product, self.stockinfo, options=options)
        self.assertEqual(0, self.basket.line_quantity(
            self.product, self.record))
        self.assertEqual(1, self.basket.line_quantity(
            self.product, self.record, options))
        self.assertEqual(1, self.basket.product_quantity(self.product))

    def test_has_method_to_test_if_submitted(self):
        self.basket.set_as_submitted()
        self.assertTrue(self.basket.is_submitted())


class TestBasketMiddleware(TestCase):

    def setUp(self):
        self.middleware = BasketMiddleware()

    def test_basket_is_attached_to_request(self):
        req = RequestFactory().get('/')
        self.middleware.process_request(req)
        self.assertTrue(hasattr(req, 'basket'))


class TestBasketReports(TestCase):

    def test_open_report_doesnt_error(self):
        data = {
            'start_date': datetime.date(2012, 5, 1),
            'end_date': datetime.date(2012, 5, 17),
            'formatter': 'CSV'
        }
        generator = OpenBasketReportGenerator(**data)
        generator.generate()

    def test_submitted_report_doesnt_error(self):
        data = {
            'start_date': datetime.date(2012, 5, 1),
            'end_date': datetime.date(2012, 5, 17),
            'formatter': 'CSV'
        }
        generator = SubmittedBasketReportGenerator(**data)
        generator.generate()