def delete(tenant): """Delete tenant.""" admin_tnt = admin.Tenant(context.GLOBAL.ldap.conn) try: admin_tnt.delete(tenant) except ldap3.LDAPNoSuchObjectResult: click.echo('Tenant does not exist: %s' % tenant, err=True)
def configure(system, tenant): """Get or modify tenant configuration.""" admin_tnt = admin.Tenant(context.GLOBAL.ldap.conn) attrs = {} if system: attrs['systems'] = system if attrs: try: admin_tnt.create(tenant, attrs) except ldap3.LDAPEntryAlreadyExistsResult: admin_tnt.update(tenant, attrs) try: cli.out(formatter(admin_tnt.get(tenant))) except ldap3.LDAPNoSuchObjectResult: click.echo('Tenant does not exist: %s' % tenant, err=True)
def configure(system, tenant): """Create, get or modify tenant configuration""" admin_tnt = admin.Tenant(context.GLOBAL.ldap.conn) attrs = {} if system: attrs['systems'] = system if attrs: try: admin_tnt.create(tenant, attrs) except ldap_exceptions.LDAPEntryAlreadyExistsResult: admin_tnt.update(tenant, attrs) try: tenant_obj = admin_tnt.get(tenant, dirty=bool(attrs)) tenant_obj['allocations'] = admin_tnt.allocations(tenant) cli.out(formatter(tenant_obj)) except ldap_exceptions.LDAPNoSuchObjectResult: click.echo('Tenant does not exist: %s' % tenant, err=True)
def test_create(self): """Dummy test for treadmill.api.tenant.create()""" tnt_admin = admin.Tenant(None) self.tnt.create('some_tenant', {'systems': [1, 2, 3]}) tnt_admin.get.assert_called_with('some_tenant')
def test_get(self): """Dummy test for treadmill.api.tenant.get()""" tnt_admin = admin.Tenant(None) self.tnt.get('some_tenant') tnt_admin.get.assert_called_with('some_tenant')
def test_list(self): """Dummy test for treadmill.api.tenant._list()""" self.tnt.list() tnt_admin = admin.Tenant(None) self.assertTrue(tnt_admin.list.called)
def setUp(self): self.tnt = admin.Tenant(admin.Admin(None, 'dc=xx,dc=com'))
def _list(): """List configured tenants.""" admin_tnt = admin.Tenant(context.GLOBAL.ldap.conn) cli.out(formatter(admin_tnt.list({})))
def _admin_tnt(): """Lazily return admin object.""" return admin.Tenant(context.GLOBAL.ldap.conn)