network_inventory/licenses/tests/test_customer_license_table_view.py

import pytest

from django.test import Client
from mixer.backend.django import mixer

from core.tests import helper
from customers.models import Customer


pytestmark = pytest.mark.django_db


def test_customer_license_table_not_logged_in():
    response = Client().get('/customer/1/licenses/')
    assert response.status_code == 302 and 'login' in response.url


def test_customer_license_table(create_admin_user):
    fixture = create_admin_user()
    customer = fixture['customer']
    client = Client()
    client.login(username="pharma-admin", password="password")
    license = mixer.blend('licenses.UserLicense', customer=customer,
                          software=mixer.SELECT, key=mixer.RANDOM,
                          max_allowed_users=mixer.RANDOM)
    response = client.get('/customer/' + str(customer.id) + '/licenses/')
    assert (response.status_code == 200
            and helper.in_content(response, license.software)
            and helper.in_content(response, license.key)
            and helper.in_content(response, license.max_allowed_users))


def test_customer_license_table_no_license(create_admin_user):
    fixture = create_admin_user()
    customer = fixture['customer']
    client = Client()
    client.login(username="pharma-admin", password="password")
    response = client.get('/customer/' + str(customer.id) + '/licenses/')
    assert (response.status_code == 200
            and helper.not_in_content(response, customer))


def test_customer_license_table_no_permission(create_admin_user):
    create_admin_user()
    customer = Customer.objects.create(name='Water Corp.')
    client = Client()
    client.login(username="pharma-admin", password="password")
    mixer.blend('licenses.UserLicense', customer=customer)
    response = client.get('/customer/' + str(customer.id) + '/licenses/')
    assert response.status_code == 403


def test_customer_license_table_multiple_licenses(create_admin_user):
    fixture = create_admin_user()
    customer = fixture['customer']
    user1 = mixer.blend('users.User')
    user2 = mixer.blend('users.User')
    client = Client()
    client.login(username="pharma-admin", password="password")
    license1 = mixer.blend('licenses.UserLicense', customer=customer,
                           software=mixer.SELECT, key=mixer.RANDOM,
                           max_allowed_users=mixer.RANDOM)
    license2 = mixer.blend('licenses.UserLicense', customer=customer,
                           software=mixer.SELECT, key=mixer.RANDOM,
                           max_allowed_users=mixer.RANDOM)
    mixer.blend('licenses.LicenseWithUser', user=user1, license=license1)
    mixer.blend('licenses.LicenseWithUser', user=user2, license=license2)
    response = client.get('/customer/' + str(customer.id) + '/licenses/')
    assert (response.status_code == 200
            and helper.in_content(response, license1.software)
            and helper.in_content(response, license1.key)
            and helper.in_content(response, license1.max_allowed_users)
            and helper.in_content(response, license2.software)
            and helper.in_content(response, license2.key)
            and helper.in_content(response, license2.max_allowed_users))