feat: T1016 dns_config probe + server-side drift detection + Crucible DNS DRIFT badge

This commit is contained in:
AetherForge
2026-05-30 23:26:50 -07:00
parent 6704933568
commit d005d5d07c
48 changed files with 4621 additions and 22 deletions

View File

@@ -7,6 +7,7 @@ import (
"testing"
"crypto-miner-server/internal/db"
"crypto-miner-server/internal/models"
"github.com/go-chi/chi/v5"
)
@@ -89,3 +90,120 @@ func TestGetAgentNotFound(t *testing.T) {
t.Fatalf("expected 404, got %d", rec.Code)
}
}
const (
handlerAgentStatsDefaultLimit = 100
handlerAgentStatsMaxLimit = 1000
handlerSharesDefaultLimit = 50
handlerSharesMaxLimit = 1000
handlerListBuildsLimit = 50
)
func TestHandlerConstants(t *testing.T) {
if handlerAgentStatsDefaultLimit != 100 || handlerAgentStatsMaxLimit != 1000 {
t.Fatal("agent stats limit constants drifted")
}
if handlerSharesDefaultLimit != 50 || handlerSharesMaxLimit != 1000 {
t.Fatal("shares limit constants drifted")
}
if handlerListBuildsLimit != 50 {
t.Fatal("list builds limit constant drifted")
}
}
func TestGetDashboardStatsEmptyFleet(t *testing.T) {
h := newTestHandler(t)
req := httptest.NewRequest(http.MethodGet, "/api/v1/dashboard/stats", nil)
rec := httptest.NewRecorder()
h.GetDashboardStats(rec, req)
if rec.Code != http.StatusOK {
t.Fatalf("status %d body %s", rec.Code, rec.Body.String())
}
}
func TestListBuildsEmptyArray(t *testing.T) {
h := newTestHandler(t)
req := httptest.NewRequest(http.MethodGet, "/api/v1/builds", nil)
rec := httptest.NewRecorder()
h.ListBuilds(rec, req)
if rec.Code != http.StatusOK {
t.Fatalf("status %d", rec.Code)
}
var builds []json.RawMessage
if err := json.Unmarshal(rec.Body.Bytes(), &builds); err != nil {
t.Fatal(err)
}
if len(builds) != 0 {
t.Fatalf("expected empty builds, got %d", len(builds))
}
}
func TestPinBuildAndUnpinAll(t *testing.T) {
h := newTestHandler(t)
database := h.db
if err := database.InsertBuild(&models.BuildRecord{
ID: "pin-me", WorkerName: "w", ServerURL: "http://x", Wallet: "48x",
FilePath: "/tmp/x", FileName: "x.exe", Platform: "windows",
}); err != nil {
t.Fatal(err)
}
r := chi.NewRouter()
r.Put("/builds/{id}/pin", h.PinBuild)
r.Delete("/builds/pin", h.UnpinAll)
req := httptest.NewRequest(http.MethodPut, "/builds/pin-me/pin", nil)
rec := httptest.NewRecorder()
r.ServeHTTP(rec, req)
if rec.Code != http.StatusOK {
t.Fatalf("pin status %d body %s", rec.Code, rec.Body.String())
}
req = httptest.NewRequest(http.MethodDelete, "/builds/pin", nil)
rec = httptest.NewRecorder()
r.ServeHTTP(rec, req)
if rec.Code != http.StatusOK {
t.Fatalf("unpin status %d body %s", rec.Code, rec.Body.String())
}
}
func TestDeleteBuildSuccess(t *testing.T) {
h := newTestHandler(t)
if err := h.db.InsertBuild(&models.BuildRecord{
ID: "del-me", WorkerName: "w", ServerURL: "http://x", Wallet: "48x",
FilePath: "/tmp/x", FileName: "x.exe", Platform: "windows",
}); err != nil {
t.Fatal(err)
}
r := chi.NewRouter()
r.Delete("/builds/{id}", h.DeleteBuild)
req := httptest.NewRequest(http.MethodDelete, "/builds/del-me", nil)
rec := httptest.NewRecorder()
r.ServeHTTP(rec, req)
if rec.Code != http.StatusOK {
t.Fatalf("delete status %d body %s", rec.Code, rec.Body.String())
}
}
func TestGetRecentSharesDefaultLimit(t *testing.T) {
h := newTestHandler(t)
req := httptest.NewRequest(http.MethodGet, "/api/v1/shares", nil)
rec := httptest.NewRecorder()
h.GetRecentShares(rec, req)
if rec.Code != http.StatusOK {
t.Fatalf("status %d", rec.Code)
}
}
func TestGetAgentStatsInvalidLimitUsesDefault(t *testing.T) {
h := newTestHandler(t)
req := httptest.NewRequest(http.MethodGet, "/agents/x/stats?limit=abc", 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("invalid limit should not 500: %s", rec.Body.String())
}
}