dela
0335d572de
ci / go (push) Waiting to run
ci / go-db (agent) (push) Waiting to run
ci / go-db (config) (push) Waiting to run
ci / go-db (db) (push) Waiting to run
ci / go-db (evidence) (push) Waiting to run
ci / go-db (llmrec) (push) Waiting to run
ci / go-db (server) (push) Waiting to run
web / web (push) Waiting to run
docs / links (push) Canceled after 0s
detections / detections (push) Canceled after 0s
89 lines
2.8 KiB
Go
89 lines
2.8 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestTaskArchiveAPIQueueListAndLimits(t *testing.T) {
|
|
m, err := NewManager(t.TempDir(), "")
|
|
if err != nil {
|
|
t.Skipf("postgres unavailable (%v) - skipping", err)
|
|
}
|
|
defer m.Close()
|
|
task, err := m.CreateTask("archive api", "verify 202", nil, 0, 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
taskID, err := strconv.ParseInt(task.ID, 10, 64)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
_, _ = m.pg.Exec(`DELETE FROM task_archives WHERE task_id=$1`, taskID)
|
|
_ = m.pg.DeleteTask(taskID)
|
|
}()
|
|
|
|
// Start with an already-cancelled context so the background archive worker
|
|
// exits before these route-contract assertions enqueue work.
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
dataDir := t.TempDir()
|
|
s := New(ctx, m, dataDir, dataDir, dataDir)
|
|
s.archiveWG.Wait()
|
|
token, err := signJWT(s.jwtKey)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
request := func(method, path, body string) *httptest.ResponseRecorder {
|
|
req := httptest.NewRequest(method, path, strings.NewReader(body))
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
recorder := httptest.NewRecorder()
|
|
s.Handler().ServeHTTP(recorder, req)
|
|
return recorder
|
|
}
|
|
|
|
recorder := request(http.MethodPost, "/api/tasks/"+task.ID+"/archive", "{}")
|
|
if recorder.Code != http.StatusConflict {
|
|
t.Fatalf("running archive status=%d body=%s", recorder.Code, recorder.Body.String())
|
|
}
|
|
if err := m.pg.SetPaused(taskID, true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
recorder = request(http.MethodPost, "/api/tasks/"+task.ID+"/archive", "{}")
|
|
if recorder.Code != http.StatusAccepted {
|
|
t.Fatalf("archive queue status=%d body=%s", recorder.Code, recorder.Body.String())
|
|
}
|
|
var queued struct {
|
|
ID int64 `json:"id"`
|
|
State string `json:"state"`
|
|
}
|
|
if err := json.Unmarshal(recorder.Body.Bytes(), &queued); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if queued.ID <= 0 || queued.State != "archive_queued" {
|
|
t.Fatalf("unexpected queue response: %+v", queued)
|
|
}
|
|
|
|
recorder = request(http.MethodGet, "/api/task-archives?page=1&size=10&q="+task.ID, "")
|
|
if recorder.Code != http.StatusOK || !strings.Contains(recorder.Body.String(), fmt.Sprintf(`"task_id":%d`, taskID)) {
|
|
t.Fatalf("archive list status=%d body=%s", recorder.Code, recorder.Body.String())
|
|
}
|
|
ids := make([]string, 101)
|
|
for index := range ids {
|
|
ids[index] = strconv.Itoa(index + 1)
|
|
}
|
|
body, _ := json.Marshal(map[string]any{"task_ids": ids})
|
|
recorder = request(http.MethodPost, "/api/tasks/archive/batch", string(body))
|
|
if recorder.Code != http.StatusBadRequest || !strings.Contains(recorder.Body.String(), "100") {
|
|
t.Fatalf("oversized batch status=%d body=%s", recorder.Code, recorder.Body.String())
|
|
}
|
|
}
|