def test_remembers_multiple_tenants(self): expected_num_tenants = 100 tenants = ['id%s' % n for n in range(expected_num_tenants)] for tenant_id in tenants: db.remember_tenant(tenant_id) num_tenants_provisioned = db.num_provisioned_tenants() self.assertEqual(expected_num_tenants, num_tenants_provisioned, 'There should be %d tenants, not %d' % (expected_num_tenants, num_tenants_provisioned))
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), )