mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-02 23:06:36 +08:00
ryan
71ebfa555f
Remove the /openflare prefix from management APIs (/api/v1/d/*) and move Agent, Relay, and Tunnel protocol endpoints under /api/v1. Update Wavelet frontend services and node client binaries to match.
139 lines
3.0 KiB
Go
139 lines
3.0 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package integration
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/openflare/compat"
|
|
"github.com/Rain-kl/Wavelet/internal/common/response"
|
|
v1 "github.com/Rain-kl/Wavelet/internal/router/v1"
|
|
ofrouter "github.com/Rain-kl/Wavelet/internal/router/v1/openflare"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func decodeAPIResponse(t *testing.T, rec *httptest.ResponseRecorder) response.Any {
|
|
t.Helper()
|
|
|
|
var resp response.Any
|
|
require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &resp))
|
|
return resp
|
|
}
|
|
|
|
func requireAPIOK(t *testing.T, rec *httptest.ResponseRecorder) response.Any {
|
|
t.Helper()
|
|
|
|
resp := decodeAPIResponse(t, rec)
|
|
require.Empty(t, resp.ErrorMsg, "unexpected API error: %s", resp.ErrorMsg)
|
|
return resp
|
|
}
|
|
|
|
func decodeEnvelope(t *testing.T, rec *httptest.ResponseRecorder) compat.Envelope {
|
|
t.Helper()
|
|
|
|
var envelope compat.Envelope
|
|
require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &envelope))
|
|
return envelope
|
|
}
|
|
|
|
func unmarshalAPIData(t *testing.T, data any, target any) {
|
|
t.Helper()
|
|
|
|
payload, err := json.Marshal(data)
|
|
require.NoError(t, err)
|
|
require.NoError(t, json.Unmarshal(payload, target))
|
|
}
|
|
|
|
func unmarshalEnvelopeData(t *testing.T, data any, target any) {
|
|
t.Helper()
|
|
unmarshalAPIData(t, data, target)
|
|
}
|
|
|
|
func unmarshalAPIMap(t *testing.T, data any) map[string]any {
|
|
t.Helper()
|
|
|
|
var result map[string]any
|
|
unmarshalAPIData(t, data, &result)
|
|
return result
|
|
}
|
|
|
|
func unmarshalEnvelopeMap(t *testing.T, data any) map[string]any {
|
|
t.Helper()
|
|
return unmarshalAPIMap(t, data)
|
|
}
|
|
|
|
func unmarshalAPISlice(t *testing.T, data any) []any {
|
|
t.Helper()
|
|
|
|
var result []any
|
|
unmarshalAPIData(t, data, &result)
|
|
return result
|
|
}
|
|
|
|
func unmarshalEnvelopeSlice(t *testing.T, data any) []any {
|
|
t.Helper()
|
|
return unmarshalAPISlice(t, data)
|
|
}
|
|
|
|
func mountOpenFlareTestRoutes(engine *gin.Engine) {
|
|
api := engine.Group("/api")
|
|
apiV1 := api.Group("/v1")
|
|
v1.RegisterV1Routes(apiV1, api)
|
|
}
|
|
|
|
func apiPath(subpath string) string {
|
|
return ofrouter.V1BasePath + subpath
|
|
}
|
|
|
|
func performJSONRequest(
|
|
t *testing.T,
|
|
engine http.Handler,
|
|
method, path string,
|
|
body any,
|
|
headers map[string]string,
|
|
) *httptest.ResponseRecorder {
|
|
t.Helper()
|
|
|
|
var payload []byte
|
|
if body != nil {
|
|
var err error
|
|
payload, err = json.Marshal(body)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
req := httptest.NewRequest(method, path, bytes.NewReader(payload))
|
|
if body != nil {
|
|
req.Header.Set("Content-Type", "application/json")
|
|
}
|
|
for key, value := range headers {
|
|
req.Header.Set(key, value)
|
|
}
|
|
|
|
rec := httptest.NewRecorder()
|
|
engine.ServeHTTP(rec, req)
|
|
return rec
|
|
}
|
|
|
|
func performLegacyRequest(
|
|
t *testing.T,
|
|
engine http.Handler,
|
|
method, path string,
|
|
body any,
|
|
headers map[string]string,
|
|
) *httptest.ResponseRecorder {
|
|
t.Helper()
|
|
return performJSONRequest(t, engine, method, path, body, headers)
|
|
}
|
|
|
|
func adminAuthHeaders(token string) map[string]string {
|
|
return map[string]string{
|
|
"X-Access-Token": token,
|
|
}
|
|
}
|