在使用Django和DRF为测试类中的特定模型创建fixture时,我遇到了一些问题。我使用的是model-mommy,但即使只使用Django ORM为该模型创建单个实例也不起作用:
from rest_framework.test import APIRequestFactory, APITestCase
from model_mommy import mommy
from api.v1 import views
from company.models.models import CompanyUserProfile, ManagerRelation, DynamicRelation
User = get_user_model()
class APIViewsTestCase(APITestCase):
@classmethod
def setUpTestData(cls):
supervisory_id = "1337"
emp_group_id = "119"
prd_scope_id = "1334"
user = mommy.make(User)
setattr(cls, 'user', user)
company_user_profile_1 = mommy.make(CompanyUserProfile)
requestor = mommy.make(
CompanyUserProfile, is_manager=True,
supervisory_id=supervisory_id, emp_group_id=emp_group_id,
prd_scope_id=prd_scope_id, user=user
)
cls.user = user
manager_relation = mommy.make(
ManagerRelation, manager_id=requestor.id, mgr_type="Direct Mgr", _quantity=5
)
dynamic_relations = mommy.make(
DynamicRelation,target_id=requestor.id, _quantity=5
)
def setUp(self) -> None:
super().setUp()
self.client.force_authenticate(user=self.user)
def test_employee_view(self):
CompanyUserProfile.objects.create(
guid=uuid.uuid4(),
csod_user_id='idcbijnccdocwocd', csod_username='djchbdcjnwdskjcn',
id=3333
)
print('cup count:', CompanyUserProfile.objects.count())
print('user count:', User.objects.count())
print('m_rels count:', ManagerRelation.objects.count())
print('d_rels count:', DynamicRelation.objects.count())
rsp = self.client.get('/api/v1/employees/')
self.assertEqual(rsp.status_code, HTTPStatus.OK)但是,无论使用mommy.make还是CompanyUserProfile.objects.create,都不会创建CompanyUserProfile记录
-- test run output
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
cup count: 0
user count: 1
m_rels count: 5
d_rels count: 5
Destroying test database for alias 'default'...CompanyUserProfile继承了AbstractCornerstoneUserProfile (抽象基础模型),这是它与其他模型的主要区别。
你知道这一切的根本原因是什么吗?
发布于 2019-05-13 18:44:13
更新:解决了-> 100%的错误,CompanyUserProfile的自定义ModelManager排除了在CompanyUserProfile.objects.count()上创建的对象。
https://stackoverflow.com/questions/56060846
复制相似问题