add tests for the warranties table
This commit is contained in:
parent
b2bc1f5d88
commit
8f69665a95
|
@ -0,0 +1,55 @@
|
|||
from datetime import datetime
|
||||
from datetime import timedelta
|
||||
import pytest
|
||||
from mixer.backend.django import mixer
|
||||
|
||||
from django.test import Client
|
||||
|
||||
from core.tests import helper
|
||||
|
||||
pytestmark = pytest.mark.django_db
|
||||
|
||||
|
||||
def test_warranties_view_not_logged_in():
|
||||
response = Client().get('/warranties/')
|
||||
assert response.status_code == 302 and 'login' in response.url
|
||||
|
||||
|
||||
def test_warranties_no_warranties_found(create_admin_user):
|
||||
create_admin_user()
|
||||
client = Client()
|
||||
customer = mixer.blend('customers.Customer')
|
||||
device = mixer.blend('devices.Device', customer=customer)
|
||||
warranty = mixer.blend('devices.Warranty', device=device)
|
||||
client.login(username="novartis-admin", password="password")
|
||||
response = client.get('/warranties/')
|
||||
assert (response.status_code == 200
|
||||
and helper.not_in_content(response, warranty.customer))
|
||||
|
||||
|
||||
def test_warranties_view_warranty_expired(create_admin_user):
|
||||
fixture = create_admin_user()
|
||||
device = mixer.blend('devices.Device', customer=fixture['customer'])
|
||||
mixer.blend('devices.Warranty', device=device, valid_until="1970-01-01")
|
||||
client = Client()
|
||||
client.login(username="novartis-admin", password="password")
|
||||
response = client.get('/warranties/')
|
||||
assert (response.status_code == 200
|
||||
and ('bgcolor="red"' in response.content.decode('utf8')))
|
||||
|
||||
|
||||
def test_warranties_view_warranty_one_year_till_expiration(create_admin_user):
|
||||
fixture = create_admin_user()
|
||||
user = fixture['admin']
|
||||
user.is_superuser = True
|
||||
user.save()
|
||||
device = mixer.blend('devices.Device', customer=fixture['customer'])
|
||||
not_one_year_more = (datetime.date(datetime.today() + timedelta(200)))
|
||||
mixer.blend('devices.Warranty',
|
||||
device=device,
|
||||
valid_until=not_one_year_more)
|
||||
client = Client()
|
||||
client.login(username="novartis-admin", password="password")
|
||||
response = client.get('/warranties/')
|
||||
assert (response.status_code == 200
|
||||
and ('bgcolor="orange"' in response.content.decode('utf8')))
|
Loading…
Reference in New Issue