mlogclub
eac039bba4
- Introduced OIDC configuration options in config.example.yaml. - Added OIDC client initialization and routes for OIDC login, callback, and exchange. - Implemented OIDC login service to handle user authentication via OIDC. - Created frontend components for OIDC login and callback handling. - Updated user creation logic to support OIDC users and their identities. - Enhanced error handling for OIDC login processes.
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"cs-agent/internal/pkg/config"
|
|
)
|
|
|
|
func TestNewServerRegistersGinRoutes(t *testing.T) {
|
|
config.SetCurrent(&config.Config{
|
|
Storage: config.StorageConfig{
|
|
Local: config.LocalStorageConfig{
|
|
Root: "storage",
|
|
BaseURL: "/storage",
|
|
},
|
|
},
|
|
})
|
|
|
|
app, err := NewServer()
|
|
if err != nil {
|
|
t.Fatalf("NewServer() error = %v", err)
|
|
}
|
|
|
|
routes := make(map[string]bool)
|
|
for _, route := range app.Routes() {
|
|
routes[route.Method+" "+route.Path] = true
|
|
}
|
|
|
|
expected := []string{
|
|
http.MethodPost + " /api/auth/login",
|
|
http.MethodGet + " /api/auth/oidc_login",
|
|
http.MethodGet + " /api/auth/oidc_callback",
|
|
http.MethodPost + " /api/auth/oidc_exchange",
|
|
http.MethodGet + " /api/auth/profile",
|
|
http.MethodGet + " /api/dashboard/user/list",
|
|
http.MethodGet + " /api/dashboard/user/:id",
|
|
http.MethodPost + " /api/dashboard/user/create",
|
|
http.MethodPost + " /api/dashboard/conversation/send_message",
|
|
http.MethodGet + " /api/ws/dashboard",
|
|
http.MethodGet + " /api/ws/open",
|
|
}
|
|
for _, route := range expected {
|
|
if !routes[route] {
|
|
t.Fatalf("expected route %s to be registered", route)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNewServerSeparatesAPIStaticAndSPA(t *testing.T) {
|
|
config.SetCurrent(&config.Config{
|
|
Storage: config.StorageConfig{
|
|
Local: config.LocalStorageConfig{
|
|
Root: "storage",
|
|
BaseURL: "/storage",
|
|
},
|
|
},
|
|
})
|
|
|
|
app, err := NewServer()
|
|
if err != nil {
|
|
t.Fatalf("NewServer() error = %v", err)
|
|
}
|
|
|
|
tests := []struct {
|
|
path string
|
|
wantStatus int
|
|
contentType string
|
|
}{
|
|
{path: "/api/not-exists", wantStatus: http.StatusNotFound, contentType: "application/json"},
|
|
{path: "/dashboard/not-exists", wantStatus: http.StatusOK, contentType: "text/html"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
rec := httptest.NewRecorder()
|
|
app.ServeHTTP(rec, httptest.NewRequest(http.MethodGet, tt.path, nil))
|
|
|
|
if rec.Code != tt.wantStatus {
|
|
t.Fatalf("%s status=%d want %d", tt.path, rec.Code, tt.wantStatus)
|
|
}
|
|
if !strings.Contains(rec.Header().Get("Content-Type"), tt.contentType) {
|
|
t.Fatalf("%s Content-Type=%q want %q", tt.path, rec.Header().Get("Content-Type"), tt.contentType)
|
|
}
|
|
}
|
|
}
|