def _create_object_stub(self, with_relations=False, size=1, **kwargs):
"""
Create an object dict with relation dicts using factories.
"""
object_list = []
for iteration in range(0, size):
obj = self.factory_cls.stub(**kwargs).__dict__
del obj['tenant']
if with_relations:
# If relations are needed, override them, because a dict is needed instead of an instance.
obj['phone_numbers'] = [PhoneNumberFactory.stub().__dict__, ]
obj['social_media'] = [SocialMediaFactory.stub().__dict__, ]
obj['addresses'] = [AddressFactory.stub().__dict__, ]
obj['email_addresses'] = [EmailAddressFactory.stub().__dict__, ]
obj['accounts'] = [AccountFactory.stub().__dict__, ]
obj['accounts'][0]['status'] = {'id': AccountStatusFactory.create(tenant=self.user_obj.tenant).id}
obj['tags'] = [TagFactory.stub().__dict__, ]
del obj['accounts'][0]['tenant']
object_list.append(obj)
if size > 1:
return object_list
else:
# If required size is 1, just give the object instead of a list.
return object_list[0]
def _create_object_stub(self, with_relations=False, size=1, **kwargs):
"""
Create an object dict with relation dicts using factories.
"""
object_list = []
for iteration in range(0, size):
obj = self.factory_cls.stub(**kwargs).__dict__
del obj['tenant']
if with_relations:
# If relations are needed, override them, because a dict is needed instead of an instance.
obj['phone_numbers'] = [PhoneNumberFactory.stub().__dict__, ]
obj['social_media'] = [SocialMediaFactory.stub().__dict__, ]
obj['addresses'] = [AddressFactory.stub().__dict__, ]
obj['email_addresses'] = [EmailAddressFactory.stub().__dict__, ]
obj['accounts'] = [AccountFactory.stub().__dict__, ]
obj['accounts'][0]['status'] = {'id': AccountStatusFactory.create(tenant=self.user_obj.tenant).id}
obj['tags'] = [TagFactory.stub().__dict__, ]
del obj['accounts'][0]['tenant']
object_list.append(obj)
if size > 1:
return object_list
else:
# If required size is 1, just give the object instead of a list.
return object_list[0]
def _create_object_stub_with_relations(self):
"""
Create an object dict with relation dicts using factories.
"""
contact = self.factory_cls.stub().__dict__
contact['phone_numbers'] = [
PhoneNumberFactory.stub().__dict__,
]
contact['social_media'] = [
SocialMediaFactory.stub().__dict__,
]
contact['addresses'] = [
AddressFactory.stub().__dict__,
]
contact['email_addresses'] = [
EmailAddressFactory.stub().__dict__,
]
contact['accounts'] = [
AccountFactory.stub().__dict__,
]
contact['tags'] = [
TagFactory.stub().__dict__,
]
del contact['tenant']
del contact['accounts'][0]['tenant']
return contact
def _create_object_stub(self, with_relations=False, size=1, **kwargs):
"""
Create an object dict with relation dicts using factories.
"""
# Set a default tenant of the user.
kwargs['tenant'] = self.user_obj.tenant if not kwargs.get('tenant') else kwargs['tenant']
object_list = []
casetype = CaseTypeFactory(**kwargs)
casestatus = CaseStatusFactory(**kwargs)
for iteration in range(0, size):
obj = self.factory_cls.stub(**kwargs).__dict__
del obj['tenant']
# The minimum viable case instance needs these relations, so always make them.
obj['type'] = {
'id': casetype.pk
}
obj['status'] = {
'id': casestatus.pk
}
if with_relations:
# If relations are needed, override them, because a dict is needed instead of an instance.
obj['account'] = AccountFactory.stub().__dict__
obj['contact'] = ContactFactory.stub().__dict__
obj['assigned_to'] = LilyUserFactory.stub().__dict__
del obj['account']['tenant']
del obj['contact']['tenant']
del obj['assigned_to']['tenant']
del obj['created_by']
else:
# Delete the related objects, since they can't be serialized.
del obj['account']
del obj['assigned_to']
del obj['created_by']
object_list.append(obj)
if size > 1:
return object_list
else:
# If required size is 1, just give the object instead of a list.
return object_list[0]
def _create_object_stub(self, with_relations=False, size=1, **kwargs):
"""
Create an object dict with relation dicts using factories.
"""
# Set a default tenant of the user.
kwargs['tenant'] = self.user_obj.tenant if not kwargs.get(
'tenant') else kwargs['tenant']
object_list = []
casetype = CaseTypeFactory(**kwargs)
casestatus = CaseStatusFactory(**kwargs)
for iteration in range(0, size):
obj = self.factory_cls.stub(**kwargs).__dict__
del obj['tenant']
# The minimum viable case instance needs these relations, so always make them.
obj['type'] = {'id': casetype.pk}
obj['status'] = {'id': casestatus.pk}
if with_relations:
# If relations are needed, override them, because a dict is needed instead of an instance.
obj['account'] = AccountFactory.stub().__dict__
obj['contact'] = ContactFactory.stub().__dict__
obj['assigned_to'] = LilyUserFactory.stub().__dict__
del obj['account']['tenant']
del obj['contact']['tenant']
del obj['assigned_to']['tenant']
del obj['created_by']
else:
# Delete the related objects, since they can't be serialized.
del obj['account']
del obj['assigned_to']
del obj['created_by']
object_list.append(obj)
if size > 1:
return object_list
else:
# If required size is 1, just give the object instead of a list.
return object_list[0]
