| package admin |
|
|
| import ( |
| "context" |
| "net/http" |
| "testing" |
|
|
| "ds2api/internal/account" |
| "ds2api/internal/auth" |
| "ds2api/internal/config" |
| dsclient "ds2api/internal/deepseek/client" |
| adminaccounts "ds2api/internal/httpapi/admin/accounts" |
| adminconfig "ds2api/internal/httpapi/admin/configmgmt" |
| adminsettings "ds2api/internal/httpapi/admin/settings" |
| adminshared "ds2api/internal/httpapi/admin/shared" |
| ) |
|
|
| var intFrom = adminshared.IntFrom |
|
|
| func toAccount(m map[string]any) config.Account { return adminshared.ToAccount(m) } |
| func fieldString(m map[string]any, key string) string { |
| return adminshared.FieldString(m, key) |
| } |
| func maskSecretPreview(secret string) string { return adminshared.MaskSecretPreview(secret) } |
| func boolFrom(v any) bool { return adminsettings.BoolFrom(v) } |
|
|
| func newAdminTestHandler(t *testing.T, raw string) *Handler { |
| t.Helper() |
| t.Setenv("DS2API_CONFIG_JSON", raw) |
| store := config.LoadStore() |
| return &Handler{ |
| Store: store, |
| Pool: account.NewPool(store), |
| } |
| } |
|
|
| type testingDSMock struct { |
| loginToken string |
| deleteAllSessionsError error |
| deleteAllSessionsErrorOnce bool |
| sessionCount *dsclient.SessionStats |
| loginCalls int |
| deleteAllCalls int |
| } |
|
|
| func (m *testingDSMock) Login(_ context.Context, _ config.Account) (string, error) { |
| m.loginCalls++ |
| if m.loginToken == "" { |
| return "token", nil |
| } |
| return m.loginToken, nil |
| } |
|
|
| func (m *testingDSMock) CreateSession(_ context.Context, _ *auth.RequestAuth, _ int) (string, error) { |
| return "session-id", nil |
| } |
|
|
| func (m *testingDSMock) GetPow(_ context.Context, _ *auth.RequestAuth, _ int) (string, error) { |
| return "pow", nil |
| } |
|
|
| func (m *testingDSMock) CallCompletion(_ context.Context, _ *auth.RequestAuth, _ map[string]any, _ string, _ int) (*http.Response, error) { |
| return &http.Response{StatusCode: http.StatusOK, Body: http.NoBody}, nil |
| } |
|
|
| func (m *testingDSMock) DeleteAllSessionsForToken(_ context.Context, _ string) error { |
| m.deleteAllCalls++ |
| if m.deleteAllSessionsError != nil { |
| err := m.deleteAllSessionsError |
| if m.deleteAllSessionsErrorOnce { |
| m.deleteAllSessionsError = nil |
| } |
| return err |
| } |
| return nil |
| } |
|
|
| func (m *testingDSMock) GetSessionCountForToken(_ context.Context, _ string) (*dsclient.SessionStats, error) { |
| if m.sessionCount != nil { |
| return m.sessionCount, nil |
| } |
| return &dsclient.SessionStats{}, nil |
| } |
|
|
| func (m *testingDSMock) SetPlatform(_ string) {} |
|
|
| func (h *Handler) configHandler() *adminconfig.Handler { |
| return &adminconfig.Handler{Store: h.Store, Pool: h.Pool, DS: h.DS, OpenAI: h.OpenAI, ChatHistory: h.ChatHistory} |
| } |
|
|
| func (h *Handler) settingsHandler() *adminsettings.Handler { |
| return &adminsettings.Handler{Store: h.Store, Pool: h.Pool, DS: h.DS, OpenAI: h.OpenAI, ChatHistory: h.ChatHistory} |
| } |
|
|
| func (h *Handler) getConfig(w http.ResponseWriter, r *http.Request) { |
| h.configHandler().GetConfig(w, r) |
| } |
|
|
| func (h *Handler) updateConfig(w http.ResponseWriter, r *http.Request) { |
| h.configHandler().UpdateConfig(w, r) |
| } |
|
|
| func (h *Handler) configImport(w http.ResponseWriter, r *http.Request) { |
| h.configHandler().ConfigImport(w, r) |
| } |
|
|
| func (h *Handler) batchImport(w http.ResponseWriter, r *http.Request) { |
| h.configHandler().BatchImport(w, r) |
| } |
|
|
| func (h *Handler) getSettings(w http.ResponseWriter, r *http.Request) { |
| h.settingsHandler().GetSettings(w, r) |
| } |
|
|
| func (h *Handler) updateSettings(w http.ResponseWriter, r *http.Request) { |
| h.settingsHandler().UpdateSettings(w, r) |
| } |
|
|
| func (h *Handler) updateSettingsPassword(w http.ResponseWriter, r *http.Request) { |
| h.settingsHandler().UpdateSettingsPassword(w, r) |
| } |
|
|
| func runAccountTestsConcurrently(accounts []config.Account, maxConcurrency int, testFn func(int, config.Account) map[string]any) []map[string]any { |
| return adminaccounts.RunAccountTestsConcurrently(accounts, maxConcurrency, testFn) |
| } |
|
|
