forked from baron/baron-sso
Implement tenant import and RP auto login policies
This commit is contained in:
@@ -60,6 +60,49 @@ func TestTenantRepository(t *testing.T) {
|
||||
assert.Equal(t, "test-domain.com", found.Domains[0].Domain)
|
||||
})
|
||||
|
||||
t.Run("AddDomain allows same domain on multiple tenants", func(t *testing.T) {
|
||||
first := &domain.Tenant{
|
||||
Name: "Saman Existing",
|
||||
Slug: "saman-existing",
|
||||
Type: domain.TenantTypeCompany,
|
||||
}
|
||||
second := &domain.Tenant{
|
||||
Name: "Saman Current",
|
||||
Slug: "saman-current",
|
||||
Type: domain.TenantTypeCompany,
|
||||
}
|
||||
assert.NoError(t, repo.Create(ctx, first))
|
||||
assert.NoError(t, repo.Create(ctx, second))
|
||||
|
||||
assert.NoError(t, repo.AddDomain(ctx, first.ID, "samaneng.com", true))
|
||||
assert.NoError(t, repo.AddDomain(ctx, second.ID, "samaneng.com", true))
|
||||
|
||||
var rows []domain.TenantDomain
|
||||
err := testDB.Where("domain = ?", "samaneng.com").Find(&rows).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Len(t, rows, 2)
|
||||
})
|
||||
|
||||
t.Run("AddDomain restores deleted tenant domain", func(t *testing.T) {
|
||||
tenant := &domain.Tenant{
|
||||
Name: "Domain Restore",
|
||||
Slug: "domain-restore",
|
||||
Type: domain.TenantTypeCompany,
|
||||
}
|
||||
assert.NoError(t, repo.Create(ctx, tenant))
|
||||
assert.NoError(t, repo.AddDomain(ctx, tenant.ID, "restore.samaneng.com", true))
|
||||
assert.NoError(t, testDB.Where("tenant_id = ? AND domain = ?", tenant.ID, "restore.samaneng.com").Delete(&domain.TenantDomain{}).Error)
|
||||
|
||||
assert.NoError(t, repo.AddDomain(ctx, tenant.ID, "restore.samaneng.com", true))
|
||||
|
||||
var rows []domain.TenantDomain
|
||||
err := testDB.Where("tenant_id = ? AND domain = ?", tenant.ID, "restore.samaneng.com").Find(&rows).Error
|
||||
assert.NoError(t, err)
|
||||
if assert.Len(t, rows, 1) {
|
||||
assert.True(t, rows[0].Verified)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Update", func(t *testing.T) {
|
||||
tenant := &domain.Tenant{
|
||||
Name: "Before Update",
|
||||
|
||||
Reference in New Issue
Block a user