Commit 38ba856d by Diana Huang

Start building tests

parent 88f54fff
...@@ -4,24 +4,32 @@ Tests for the Shopping Cart ...@@ -4,24 +4,32 @@ Tests for the Shopping Cart
from factory import DjangoModelFactory from factory import DjangoModelFactory
from django.test import TestCase from django.test import TestCase
from shoppingcart import models from shoppingcart.models import Order, VerifiedCertificate
from student.tests.factories import UserFactory from student.tests.factories import UserFactory
class OrderFactory(DjangoModelFactory): class OrderFactory(DjangoModelFactory):
FACTORY_FOR = models.Order FACTORY_FOR = Order
class OrderItem(DjangoModelFactory): class VerifiedCertificateFactory(DjangoModelFactory):
FACTORY_FOR = models.OrderItem FACTORY_FOR = VerifiedCertificate
class OrderTest(TestCase): class OrderTest(TestCase):
def setUp(self): def setUp(self):
self.user = UserFactory.create() self.user = UserFactory.create()
self.cart = OrderFactory.create(user=self.user, status='cart') self.cart = OrderFactory.create(user=self.user, status='cart')
self.course_id = "test/course"
def test_add_item_to_cart(self):
pass
def test_total_cost(self): def test_total_cost(self):
# add items to the order # add items to the order
for _ in xrange(5): cost = 30
pass iterations = 5
for _ in xrange(iterations):
VerifiedCertificate.add_to_order(self.cart, self.user, self.course_id, cost)
self.assertEquals(self.cart.total_cost, cost * iterations)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment