30 lines
950 B
Python
30 lines
950 B
Python
import pytest
|
|
from django.contrib.auth import get_user_model
|
|
from apps.resumes.models import Resume
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
@pytest.fixture
|
|
def seeker():
|
|
return User.objects.create_user(username='seeker1', password='pass', role='seeker')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
class TestResumeModel:
|
|
def test_create_resume(self, seeker):
|
|
resume = Resume.objects.create(
|
|
user=seeker,
|
|
name='张三',
|
|
gender='male',
|
|
education=[{'school': '北京大学', 'degree': '本科', 'major': '计算机'}],
|
|
experience=[{'company': 'ABC公司', 'position': '工程师', 'duration': '2年'}],
|
|
)
|
|
assert resume.name == '张三'
|
|
assert len(resume.education) == 1
|
|
assert len(resume.experience) == 1
|
|
|
|
def test_seeker_has_one_resume(self, seeker):
|
|
Resume.objects.create(user=seeker, name='张三')
|
|
assert Resume.objects.filter(user=seeker).count() == 1
|