92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"crypto-miner-server/internal/db"
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
func newTestHandler(t *testing.T) *Handler {
|
|
t.Helper()
|
|
database, err := db.New(t.TempDir())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = database.Close() })
|
|
return NewHandler(database)
|
|
}
|
|
|
|
func TestHealthCheckReturnsOK(t *testing.T) {
|
|
h := newTestHandler(t)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/health", nil)
|
|
rec := httptest.NewRecorder()
|
|
h.HealthCheck(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status %d", rec.Code)
|
|
}
|
|
var body map[string]string
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if body["status"] != "ok" {
|
|
t.Fatalf("unexpected body: %v", body)
|
|
}
|
|
}
|
|
|
|
func TestListAgentsEmptyArray(t *testing.T) {
|
|
h := newTestHandler(t)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/agents", nil)
|
|
rec := httptest.NewRecorder()
|
|
h.ListAgents(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status %d body %s", rec.Code, rec.Body.String())
|
|
}
|
|
if rec.Body.String() != "[]\n" && rec.Body.String() != "[]" {
|
|
var agents []json.RawMessage
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &agents); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(agents) != 0 {
|
|
t.Fatalf("expected empty list, got %d", len(agents))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetAgentStatsLimitCap(t *testing.T) {
|
|
h := newTestHandler(t)
|
|
req := httptest.NewRequest(http.MethodGet, "/agents/missing-agent/stats?limit=5000", nil)
|
|
rec := httptest.NewRecorder()
|
|
r := chi.NewRouter()
|
|
r.Get("/agents/{id}/stats", h.GetAgentStats)
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code == http.StatusInternalServerError {
|
|
t.Fatalf("limit cap caused 500: %s", rec.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestGetRecentSharesLimitCap(t *testing.T) {
|
|
h := newTestHandler(t)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/shares?limit=999999", nil)
|
|
rec := httptest.NewRecorder()
|
|
h.GetRecentShares(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status %d body %s", rec.Code, rec.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestGetAgentNotFound(t *testing.T) {
|
|
h := newTestHandler(t)
|
|
req := httptest.NewRequest(http.MethodGet, "/agents/nope", nil)
|
|
rec := httptest.NewRecorder()
|
|
r := chi.NewRouter()
|
|
r.Get("/agents/{id}", h.GetAgent)
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusNotFound {
|
|
t.Fatalf("expected 404, got %d", rec.Code)
|
|
}
|
|
}
|