def test_tenant_is_removed(self): tenant_id = 'test' db.remember_tenant(tenant_id) db.forget_tenant(tenant_id) net_provisioned = db.is_tenant_provisioned(tenant_id) self.assertFalse(net_provisioned, 'The Tenant should be deleted')
def delete_tenant(self, tenant_id): """delete a tenant from DB. A tenant is deleted only if there is no network or VM configured configured for this tenant. """ objects_for_tenant = (db.num_nets_provisioned(tenant_id) + db.num_vms_provisioned(tenant_id)) if not objects_for_tenant: db.forget_tenant(tenant_id)
def test_removes_multiple_tenants(self): num_tenants = 100 tenants = ['id%s' % n for n in range(num_tenants)] for tenant_id in tenants: db.remember_tenant(tenant_id) for tenant_id in tenants: db.forget_tenant(tenant_id) num_tenants_provisioned = db.num_provisioned_tenants() expected = 0 self.assertEqual(expected, num_tenants_provisioned, 'There should be %d tenants, not %d' % (expected, num_tenants_provisioned))