def setUpClass(cls):
super(TenantTests, cls).setUpClass()
cls.user_obj.tenant.name = "User Tenant"
cls.user_obj.tenant.save()
ExternalAppLinkFactory.create(tenant_id=cls.user_obj.tenant_id)
cls.other_tenant_user_obj.tenant.name = "Other Tenant User Tenant"
cls.other_tenant_user_obj.tenant.save()
ExternalAppLinkFactory.create(tenant_id=cls.other_tenant_user_obj.tenant_id)
def setUpTestData(cls):
super(TenantTests, cls).setUpTestData()
cls.user_obj.tenant.name = "User Tenant"
cls.user_obj.tenant.save()
ExternalAppLinkFactory.create(tenant_id=cls.user_obj.tenant_id)
cls.other_tenant_user_obj.tenant.name = "Other Tenant User Tenant"
cls.other_tenant_user_obj.tenant.save()
ExternalAppLinkFactory.create(
tenant_id=cls.other_tenant_user_obj.tenant_id)
def _create_object_stub(self, size=1, **kwargs):
"""
Create an object dict with relation dicts using factories.
"""
object_list = []
app_link_kwargs = kwargs
app_link_kwargs['tenant_id'] = self.user_obj.tenant.pk
for iteration in range(0, size):
obj = self.factory_cls.stub(**kwargs).__dict__
obj['external_app_links'] = ExternalAppLinkFactory.stub(**app_link_kwargs).__dict__
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, size=1, **kwargs):
"""
Create an object dict with relation dicts using factories.
"""
object_list = []
app_link_kwargs = kwargs
app_link_kwargs['tenant_id'] = self.user_obj.tenant.pk
for iteration in range(0, size):
obj = self.factory_cls.stub(**kwargs).__dict__
obj['external_app_links'] = ExternalAppLinkFactory.stub(
**app_link_kwargs).__dict__
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 external_app_links(self, create, extracted, **kwargs):
if not create:
return
size = extracted.get('size', 1) if extracted else 1
ExternalAppLinkFactory.create_batch(tenant_id=self.pk, size=size)
