075613c4ac
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
fleetai "crypto-miner-server/internal/ai"
|
|
"crypto-miner-server/internal/db"
|
|
)
|
|
|
|
type stubFleetAIConfig struct {
|
|
view FleetAIConfigView
|
|
}
|
|
|
|
func (s *stubFleetAIConfig) GetFleetAIConfig() FleetAIConfigView { return s.view }
|
|
func (s *stubFleetAIConfig) UpdateFleetAIConfig(v FleetAIConfigView) error {
|
|
s.view = v
|
|
return nil
|
|
}
|
|
|
|
func TestFleetAIHandlerGetPutConfig(t *testing.T) {
|
|
cfg := &stubFleetAIConfig{view: FleetAIConfigView{
|
|
AIControlEnabled: false, AIEndpoint: "http://127.0.0.1:11434/v1", AIDecisionIntervalSec: 60,
|
|
}}
|
|
h := NewFleetAIHandler(cfg, nil)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/ai/config", nil)
|
|
rec := httptest.NewRecorder()
|
|
h.GetConfig(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status %d", rec.Code)
|
|
}
|
|
|
|
body := `{"ai_control_enabled":true,"ai_endpoint":"http://127.0.0.1:11434/v1","ai_model":"llama3.2","ai_no_context":true,"ai_decision_interval_sec":45}`
|
|
req = httptest.NewRequest(http.MethodPut, "/api/v1/ai/config", strings.NewReader(body))
|
|
rec = httptest.NewRecorder()
|
|
h.PutConfig(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("put status %d body %s", rec.Code, rec.Body.String())
|
|
}
|
|
if !cfg.view.AIControlEnabled || cfg.view.AIModel != "llama3.2" {
|
|
t.Fatalf("config not updated: %+v", cfg.view)
|
|
}
|
|
}
|
|
|
|
func TestFleetAIHandlerGetDecisions(t *testing.T) {
|
|
database, err := db.New(t.TempDir())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { database.Close() })
|
|
_ = database.InsertAIDecision("agent-x", "abc", `{"commands":[]}`, "noop:ok")
|
|
|
|
h := NewFleetAIHandler(nil, database)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/ai/decisions?agent_id=agent-x", nil)
|
|
rec := httptest.NewRecorder()
|
|
h.GetDecisions(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status %d", rec.Code)
|
|
}
|
|
var rows []map[string]interface{}
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &rows); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(rows) != 1 {
|
|
t.Fatalf("rows: %v", rows)
|
|
}
|
|
}
|
|
|
|
func TestFleetAIExecutorRestartMining(t *testing.T) {
|
|
hub := NewWSHub(nil)
|
|
exec := &FleetAIExecutor{Hub: hub}
|
|
_, err := exec.Execute("missing", fleetai.Command{Type: fleetai.CmdRestartMining})
|
|
if err == nil {
|
|
t.Fatal("expected error for disconnected agent")
|
|
}
|
|
}
|