d7939bed70
- Implemented `useManualChallenge`, `useChallengePoll`, and `useManualChallengeMutations` hooks for managing manual DNS challenges. - Created tests for the `useManualChallenge` hooks to ensure correct fetching and mutation behavior. - Added `ManualDNSChallenge` component for displaying challenge details and actions. - Developed end-to-end tests for the Manual DNS Provider feature, covering provider selection, challenge UI, and accessibility compliance. - Included error handling tests for verification failures and network errors.
102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/Wikid82/charon/backend/internal/services"
|
|
)
|
|
|
|
func TestUpdateHandler_Check(t *testing.T) {
|
|
// Mock GitHub API
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/releases/latest" {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"tag_name":"v1.0.0","html_url":"https://github.com/example/repo/releases/tag/v1.0.0"}`))
|
|
}))
|
|
defer server.Close()
|
|
|
|
// Setup Service
|
|
svc := services.NewUpdateService()
|
|
err := svc.SetAPIURL(server.URL + "/releases/latest")
|
|
assert.NoError(t, err)
|
|
|
|
// Setup Handler
|
|
h := NewUpdateHandler(svc)
|
|
|
|
// Setup Router
|
|
gin.SetMode(gin.TestMode)
|
|
r := gin.New()
|
|
r.GET("/api/v1/update", h.Check)
|
|
|
|
// Test Request
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/update", http.NoBody)
|
|
resp := httptest.NewRecorder()
|
|
r.ServeHTTP(resp, req)
|
|
|
|
assert.Equal(t, http.StatusOK, resp.Code)
|
|
|
|
var info services.UpdateInfo
|
|
err = json.Unmarshal(resp.Body.Bytes(), &info)
|
|
assert.NoError(t, err)
|
|
assert.True(t, info.Available) // Assuming current version is not v1.0.0
|
|
assert.Equal(t, "v1.0.0", info.LatestVersion)
|
|
|
|
// Test Failure
|
|
serverError := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer serverError.Close()
|
|
|
|
svcError := services.NewUpdateService()
|
|
err = svcError.SetAPIURL(serverError.URL)
|
|
assert.NoError(t, err)
|
|
hError := NewUpdateHandler(svcError)
|
|
|
|
rError := gin.New()
|
|
rError.GET("/api/v1/update", hError.Check)
|
|
|
|
reqError := httptest.NewRequest(http.MethodGet, "/api/v1/update", http.NoBody)
|
|
respError := httptest.NewRecorder()
|
|
rError.ServeHTTP(respError, reqError)
|
|
|
|
assert.Equal(t, http.StatusOK, respError.Code)
|
|
var infoError services.UpdateInfo
|
|
err = json.Unmarshal(respError.Body.Bytes(), &infoError)
|
|
assert.NoError(t, err)
|
|
assert.False(t, infoError.Available)
|
|
|
|
// Test Client Error (Invalid URL)
|
|
// Note: This will now fail validation at SetAPIURL, which is expected
|
|
// The invalid URL won't pass our security checks
|
|
svcClientError := services.NewUpdateService()
|
|
err = svcClientError.SetAPIURL("http://localhost:1/invalid")
|
|
// Note: We can't test with truly invalid domains anymore due to validation
|
|
// This is actually a security improvement
|
|
if err != nil {
|
|
// Validation rejected the URL, which is expected for non-localhost/non-github URLs
|
|
t.Skip("Skipping invalid URL test - validation now prevents invalid URLs")
|
|
return
|
|
}
|
|
hClientError := NewUpdateHandler(svcClientError)
|
|
|
|
rClientError := gin.New()
|
|
rClientError.GET("/api/v1/update", hClientError.Check)
|
|
|
|
reqClientError := httptest.NewRequest(http.MethodGet, "/api/v1/update", http.NoBody)
|
|
respClientError := httptest.NewRecorder()
|
|
rClientError.ServeHTTP(respClientError, reqClientError)
|
|
|
|
// CheckForUpdates returns error on client failure
|
|
// Handler returns 500 on error
|
|
assert.Equal(t, http.StatusInternalServerError, respClientError.Code)
|
|
}
|