network_inventory/inventory/tests/test_models/test_license.py

25 lines
779 B
Python

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('inventory.User')
mixer.blend('inventory.UserLicense')
with pytest.raises(IntegrityError):
mixer.cycle(2).blend('inventory.LicenseWithUser',
user=mixer.SELECT,
license=mixer.SELECT)
def test_license_two_licenses_per_computer():
mixer.blend('inventory.Computer')
mixer.blend('inventory.ComputerLicense')
with pytest.raises(IntegrityError):
mixer.cycle(2).blend('inventory.LicenseWithComputer',
computer=mixer.SELECT,
license=mixer.SELECT)