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)
