tests: add check of basketcell get_badge method

This commit is contained in:
Frédéric Péters 2016-12-17 20:40:50 +01:00
parent 32a4aac3e1
commit 08738e4c53
1 changed files with 5 additions and 0 deletions

View File

@ -1,3 +1,5 @@
# -*- coding: utf-8 -*-
import pytest
from django.contrib.auth.models import User
@ -48,8 +50,10 @@ def test_basket_cell(regie, user):
context = Context({'request': RequestFactory().get('/')})
context['request'].user = None
assert cell.is_relevant(context) is False
assert cell.get_badge(context) is None
context['request'].user = user
assert cell.is_relevant(context) is False
assert cell.get_badge(context) is None
item = BasketItem()
item.user = user
item.regie = regie
@ -66,6 +70,7 @@ def test_basket_cell(regie, user):
item_anonymous.save()
assert cell.is_relevant(context) is True
assert cell.get_badge(context) == {'badge': u'12345€'}
item.cancellation_date = timezone.now()
item.save()