Slight cleanup

This commit is contained in:
2022-01-08 17:24:28 +00:00
parent 1b32dc2db0
commit f15cfa8fce
4 changed files with 18 additions and 10 deletions

View File

@@ -2,10 +2,18 @@ import pytest
from training import models
from RIGS.models import Profile
@pytest.fixture
def trainee(db):
trainee = Profile.objects.create(username="trainee", first_name="Train", last_name="EE",
initials="TRN",
email="trainee@example.com", is_active=True, is_approved=True)
initials="TRN",
email="trainee@example.com", is_active=True, is_approved=True)
yield trainee
trainee.delete()
@pytest.fixture
def level(db):
level = models.TrainingLevel.objects.create(description="There is no description.", level=models.TrainingLevel.TECHNICIAN)
yield level
level.delete()