edan_f0/tests/functional/stock_alert_tests.py

from decimal import Decimal as D

from django.test import TestCase

from oscar.test.factories import create_product, create_stockrecord
from oscar.apps.basket.models import Basket
from oscar.apps.partner.models import StockRecord, StockAlert
from oscar.apps.order.utils import OrderCreator
from oscar.test.basket import add_product
from oscar.test import factories


class TestPlacingAnOrder(TestCase):

    def setUp(self):
        self.product = create_product()
        self.stockrecord = create_stockrecord(self.product, D('12.00'),
                                              num_in_stock=5)
        self.basket = factories.create_basket(empty=True)
        add_product(self.basket, product=self.product)

    def set_threshold(self, threshold):
        self.stockrecord.low_stock_threshold = threshold
        self.stockrecord.save()

    def test_correctly_allocates_stock(self):
        OrderCreator().place_order(self.basket)

        record = StockRecord.objects.get(product=self.product)
        self.assertEqual(5, record.num_in_stock)
        self.assertEqual(1, record.num_allocated)
        self.assertEqual(4, record.net_stock_level)

    def test_does_not_raise_an_alert_if_threshold_not_broken(self):
        self.set_threshold(4)
        OrderCreator().place_order(self.basket)

        alerts = StockAlert.objects.all()
        self.assertEqual(0, len(alerts))

    def test_raises_alert_when_threshold_is_reached(self):
        self.set_threshold(5)
        OrderCreator().place_order(self.basket)

        alerts = StockAlert.objects.filter(
            stockrecord=self.stockrecord)
        self.assertEqual(1, len(alerts))

    def test_only_raises_an_alert_once(self):
        self.set_threshold(5)
        StockAlert.objects.create(stockrecord=self.stockrecord,
                                  threshold=10)
        OrderCreator().place_order(self.basket)

        alerts = StockAlert.objects.filter(
            stockrecord=self.stockrecord)
        self.assertEqual(1, len(alerts))


class TestRestockingProduct(TestCase):

    def setUp(self):
        self.product = create_product()
        self.stockrecord = create_stockrecord(self.product, D('12.00'),
                                              num_in_stock=5)
        self.basket = factories.create_basket(empty=True)
        add_product(self.basket, product=self.product)

    def set_threshold(self, threshold):
        self.stockrecord.low_stock_threshold = threshold
        self.stockrecord.save()

    def test_closes_open_alert(self):
        self.set_threshold(5)
        OrderCreator().place_order(self.basket)

        alert = StockAlert.objects.get(stockrecord=self.stockrecord)
        self.assertEqual(StockAlert.OPEN, alert.status)

        # Restock product
        self.stockrecord.num_in_stock = 15
        self.stockrecord.save()

        alert = StockAlert.objects.get(stockrecord=self.stockrecord)
        self.assertEqual(StockAlert.CLOSED, alert.status)