package rest import ( "net/http" "net/http/httptest" "testing" ) func TestHandleGetAddressRequiresDB(t *testing.T) { server := NewServer(nil, 138) req := httptest.NewRequest(http.MethodGet, "/api/v1/addresses/138/0xAbCdEf1234567890ABCdef1234567890abCDef12?address=0xAbCdEf1234567890ABCdef1234567890abCDef12", nil) w := httptest.NewRecorder() server.handleGetAddress(w, req) if w.Code != http.StatusServiceUnavailable { t.Fatalf("expected 503 when db is unavailable, got %d", w.Code) } }