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)