forked from baron/baron-sso
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"baron-sso-backend/internal/domain"
|
|
"baron-sso-backend/internal/service"
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// MockTenantService is a mock for service.TenantService
|
|
type MockTenantService struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *MockTenantService) RegisterTenant(ctx context.Context, name, slug, description string, domains []string) (*domain.Tenant, error) {
|
|
args := m.Called(ctx, name, slug, description, domains)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*domain.Tenant), args.Error(1)
|
|
}
|
|
|
|
func (m *MockTenantService) GetTenantByDomain(ctx context.Context, domainName string) (*domain.Tenant, error) {
|
|
args := m.Called(ctx, domainName)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*domain.Tenant), args.Error(1)
|
|
}
|
|
|
|
func (m *MockTenantService) ApproveTenant(ctx context.Context, tenantID string) error {
|
|
args := m.Called(ctx, tenantID)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockTenantService) RequestRegistration(ctx context.Context, name, slug, description, domainName, adminEmail string) (*domain.Tenant, error) {
|
|
args := m.Called(ctx, name, slug, description, domainName, adminEmail)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*domain.Tenant), args.Error(1)
|
|
}
|
|
|
|
func (m *MockTenantService) GetTenantBySlug(ctx context.Context, slug string) (*domain.Tenant, error) {
|
|
args := m.Called(ctx, slug)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*domain.Tenant), args.Error(1)
|
|
}
|
|
|
|
func (m *MockTenantService) GetTenant(ctx context.Context, id string) (*domain.Tenant, error) {
|
|
args := m.Called(ctx, id)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*domain.Tenant), args.Error(1)
|
|
}
|
|
|
|
func (m *MockTenantService) SetKetoService(keto service.KetoService) {
|
|
m.Called(keto)
|
|
}
|
|
|
|
func TestTenantHandler_CreateTenant(t *testing.T) {
|
|
app := fiber.New()
|
|
mockSvc := new(MockTenantService)
|
|
// CreateTenant checks h.DB != nil
|
|
h := &TenantHandler{Service: mockSvc, DB: &gorm.DB{}}
|
|
|
|
app.Post("/tenants", h.CreateTenant)
|
|
|
|
input := map[string]interface{}{
|
|
"name": "Test Tenant",
|
|
"slug": "test-tenant",
|
|
"domains": []string{"test.com"},
|
|
}
|
|
body, _ := json.Marshal(input)
|
|
|
|
mockSvc.On("RegisterTenant", mock.Anything, "Test Tenant", "test-tenant", "", []string{"test.com"}).
|
|
Return(&domain.Tenant{ID: "t1", Name: "Test Tenant", Slug: "test-tenant"}, nil)
|
|
|
|
req := httptest.NewRequest("POST", "/tenants", bytes.NewReader(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
resp, _ := app.Test(req)
|
|
|
|
assert.Equal(t, http.StatusCreated, resp.StatusCode)
|
|
var got map[string]interface{}
|
|
json.NewDecoder(resp.Body).Decode(&got)
|
|
assert.Equal(t, "t1", got["id"])
|
|
}
|
|
|
|
func TestTenantHandler_ApproveTenant(t *testing.T) {
|
|
app := fiber.New()
|
|
mockSvc := new(MockTenantService)
|
|
h := &TenantHandler{Service: mockSvc}
|
|
|
|
app.Post("/tenants/:id/approve", h.ApproveTenant)
|
|
|
|
mockSvc.On("ApproveTenant", mock.Anything, "t1").Return(nil)
|
|
|
|
req := httptest.NewRequest("POST", "/tenants/t1/approve", nil)
|
|
resp, _ := app.Test(req)
|
|
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
}
|