def render(self, context):
obj = self.obj_var.resolve(context)
self.item = Item(object_id=obj.pk, content_type_id=content_type(obj))
self.item.quantity = self.qty
self.form = ItemForm(instance=self.item)
context['form'] = self.form
return template.Template(
"""<form action={% url order:add-item %} method="post">
{% csrf_token %}
{{ form }}
<input type="submit" value="Add to Order">
</form>""").render(template.Context(context))
def render(self, context):
obj = self.obj_var.resolve(context)
self.item = Item(object_id=obj.pk, content_type_id=content_type(obj))
self.item.quantity = self.qty
self.form = ItemForm(instance=self.item)
context["form"] = self.form
return template.Template(
"""<form action={% url order:add-item %} method="post">
{% csrf_token %}
{{ form }}
<input type="submit" value="Add to Order">
</form>"""
).render(template.Context(context))
def test_HTTP_remove_from_order(self):
"""
Assert Correct messages level is returned to user and item is
removed from order
"""
# Add item to order
response = self.add_item_to_order()
ct = content_type(self.product)
response = self.client.get(reverse('order:remove-item',
kwargs={'content_type': ct.pk, 'object_id': self.product.pk}), follow=True)
self.assertTrue('messages' in response.context)
assert 'info' in list(response.context['messages'])[0].tags.split()
# Test order has been populated
self.assertTrue(response.context['order'].is_empty())
def __item_kwargs(self, product):
"Default arguments to get or create an item"
return {'content_type': content_type(product), 'object_id': product.pk}
def __item_kwargs(self, product):
"Default arguments to get or create an item"
return {'content_type': content_type(product), 'object_id': product.pk}
def add_item_to_order(self):
ct = content_type(self.product)
return self.client.post(reverse('order:add-item'),
{'content_type': ct.pk, 'object_id': self.product.pk, 'quantity': 1}, follow=True)
