forked from baron/baron-sso
Merge origin/dev into dev
This commit is contained in:
151
backend/internal/handler/tenant_handler_get_test.go
Normal file
151
backend/internal/handler/tenant_handler_get_test.go
Normal file
@@ -0,0 +1,151 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"baron-sso-backend/internal/domain"
|
||||
"baron-sso-backend/internal/testsupport"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/mock"
|
||||
)
|
||||
|
||||
func TestTenantHandler_GetTenant_SuperAdmin(t *testing.T) {
|
||||
if !testsupport.DockerAvailable() {
|
||||
t.Skip("Docker provider is unavailable in this environment")
|
||||
}
|
||||
|
||||
db := newTenantHandlerSeedDeleteDB(t)
|
||||
if err := db.AutoMigrate(&domain.TenantDomain{}); err != nil {
|
||||
t.Fatalf("failed to migrate tenant domains: %v", err)
|
||||
}
|
||||
|
||||
// Create a test tenant in DB with a valid UUID
|
||||
tenant := domain.Tenant{
|
||||
ID: "00000000-0000-0000-0000-000000000010",
|
||||
Name: "Super Admin Test Tenant",
|
||||
Slug: "super-admin-test-tenant",
|
||||
Type: domain.TenantTypeCompany,
|
||||
Status: domain.TenantStatusActive,
|
||||
}
|
||||
if err := db.Create(&tenant).Error; err != nil {
|
||||
t.Fatalf("failed to create tenant: %v", err)
|
||||
}
|
||||
|
||||
app := fiber.New()
|
||||
mockSvc := new(MockTenantService)
|
||||
mockKeto := new(devMockKetoService)
|
||||
|
||||
h := &TenantHandler{
|
||||
DB: db,
|
||||
Service: mockSvc,
|
||||
Keto: mockKeto,
|
||||
}
|
||||
|
||||
// We'll simulate middleware setting "user_profile" for a Super Admin
|
||||
app.Get("/tenants/:id", func(c *fiber.Ctx) error {
|
||||
profile := &domain.UserProfileResponse{
|
||||
ID: "user-super-admin-id",
|
||||
Role: domain.RoleSuperAdmin,
|
||||
}
|
||||
c.Locals("user_profile", profile)
|
||||
return h.GetTenant(c)
|
||||
})
|
||||
|
||||
req := httptest.NewRequest("GET", "/tenants/00000000-0000-0000-0000-000000000010", nil)
|
||||
resp, err := app.Test(req)
|
||||
if err != nil {
|
||||
t.Fatalf("request failed: %v", err)
|
||||
}
|
||||
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
|
||||
var got tenantSummary
|
||||
err = json.NewDecoder(resp.Body).Decode(&got)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to decode response: %v", err)
|
||||
}
|
||||
|
||||
assert.Equal(t, "00000000-0000-0000-0000-000000000010", got.ID)
|
||||
assert.Equal(t, "Super Admin Test Tenant", got.Name)
|
||||
assert.NotNil(t, got.UserPermissions)
|
||||
assert.True(t, got.UserPermissions.View)
|
||||
assert.True(t, got.UserPermissions.Manage)
|
||||
assert.True(t, got.UserPermissions.ManageAdmins)
|
||||
}
|
||||
|
||||
func TestTenantHandler_GetTenant_NormalUser(t *testing.T) {
|
||||
if !testsupport.DockerAvailable() {
|
||||
t.Skip("Docker provider is unavailable in this environment")
|
||||
}
|
||||
|
||||
db := newTenantHandlerSeedDeleteDB(t)
|
||||
if err := db.AutoMigrate(&domain.TenantDomain{}); err != nil {
|
||||
t.Fatalf("failed to migrate tenant domains: %v", err)
|
||||
}
|
||||
|
||||
// Create a test tenant in DB with a valid UUID
|
||||
tenant := domain.Tenant{
|
||||
ID: "00000000-0000-0000-0000-000000000020",
|
||||
Name: "Normal User Test Tenant",
|
||||
Slug: "normal-user-test-tenant",
|
||||
Type: domain.TenantTypeCompany,
|
||||
Status: domain.TenantStatusActive,
|
||||
}
|
||||
if err := db.Create(&tenant).Error; err != nil {
|
||||
t.Fatalf("failed to create tenant: %v", err)
|
||||
}
|
||||
|
||||
app := fiber.New()
|
||||
mockSvc := new(MockTenantService)
|
||||
mockKeto := new(devMockKetoService)
|
||||
|
||||
h := &TenantHandler{
|
||||
DB: db,
|
||||
Service: mockSvc,
|
||||
Keto: mockKeto,
|
||||
}
|
||||
|
||||
// Mock Keto response: allowed view/manage but not manage_admins
|
||||
subject := "User:user-normal-id"
|
||||
mockKeto.On("CheckPermission", mock.Anything, subject, "Tenant", "00000000-0000-0000-0000-000000000020", "view").Return(true, nil)
|
||||
mockKeto.On("CheckPermission", mock.Anything, subject, "Tenant", "00000000-0000-0000-0000-000000000020", "manage").Return(true, nil)
|
||||
mockKeto.On("CheckPermission", mock.Anything, subject, "Tenant", "00000000-0000-0000-0000-000000000020", "manage_admins").Return(false, nil)
|
||||
mockKeto.On("CheckPermission", mock.Anything, subject, "Tenant", "00000000-0000-0000-0000-000000000020", mock.Anything).Return(false, nil).Maybe()
|
||||
|
||||
// We'll simulate middleware setting "user_profile" for a regular admin/user
|
||||
app.Get("/tenants/:id", func(c *fiber.Ctx) error {
|
||||
profile := &domain.UserProfileResponse{
|
||||
ID: "user-normal-id",
|
||||
Role: domain.RoleUser,
|
||||
}
|
||||
c.Locals("user_profile", profile)
|
||||
return h.GetTenant(c)
|
||||
})
|
||||
|
||||
req := httptest.NewRequest("GET", "/tenants/00000000-0000-0000-0000-000000000020", nil)
|
||||
resp, err := app.Test(req)
|
||||
if err != nil {
|
||||
t.Fatalf("request failed: %v", err)
|
||||
}
|
||||
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
|
||||
var got tenantSummary
|
||||
err = json.NewDecoder(resp.Body).Decode(&got)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to decode response: %v", err)
|
||||
}
|
||||
|
||||
assert.Equal(t, "00000000-0000-0000-0000-000000000020", got.ID)
|
||||
assert.Equal(t, "Normal User Test Tenant", got.Name)
|
||||
assert.NotNil(t, got.UserPermissions)
|
||||
assert.True(t, got.UserPermissions.View)
|
||||
assert.True(t, got.UserPermissions.Manage)
|
||||
assert.False(t, got.UserPermissions.ManageAdmins)
|
||||
|
||||
mockKeto.AssertExpectations(t)
|
||||
}
|
||||
Reference in New Issue
Block a user