network_inventory/licenses/tests/test_license.py

import pytest
from mixer.backend.django import mixer
from django.db import IntegrityError


pytestmark = pytest.mark.django_db


def test_license_two_licenses_per_user():
    mixer.blend('users.User')
    mixer.blend('licenses.UserLicense')
    with pytest.raises(IntegrityError):
        mixer.cycle(2).blend('licenses.LicenseWithUser',
                             user=mixer.SELECT,
                             license=mixer.SELECT)


def test_license_two_licenses_per_computer():
    mixer.blend('inventory.Computer')
    mixer.blend('licenses.ComputerLicense')
    with pytest.raises(IntegrityError):
        mixer.cycle(2).blend('licenses.LicenseWithComputer',
                             computer=mixer.SELECT,
                             license=mixer.SELECT)