104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
|
package rest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NewMockRequest creates a new mock request.
|
||
|
|
func NewMockRequest(t *testing.T, method string, body io.ReadCloser) *http.Request {
|
||
|
|
req, err := http.NewRequest(method, "/vm/xxx?name=alice", body)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return req
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestDecodeReq tests the decodeReq function.
|
||
|
|
func TestDecodeReq(t *testing.T) {
|
||
|
|
// Define a test struct
|
||
|
|
type TestStruct struct {
|
||
|
|
Field string `json:"field"`
|
||
|
|
EnvID string `json:"env_id" in:"path=env_id" validate:"required"`
|
||
|
|
Name string `json:"name" in:"query=name"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Test case 1: GET request with no body
|
||
|
|
t.Run("GET request with no body", func(t *testing.T) {
|
||
|
|
req := NewMockRequest(t, http.MethodGet, nil)
|
||
|
|
result, err := decodeReq[TestStruct](req)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "alice", result.Name)
|
||
|
|
})
|
||
|
|
|
||
|
|
// Test case 2: POST request with valid JSON body
|
||
|
|
t.Run("POST request with valid JSON body", func(t *testing.T) {
|
||
|
|
jsonBody := `{"field": "value", "env_id": "test", "name": "alice1"}`
|
||
|
|
body := io.NopCloser(bytes.NewBufferString(jsonBody))
|
||
|
|
req := NewMockRequest(t, http.MethodPost, body)
|
||
|
|
result, err := decodeReq[TestStruct](req)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "value", result.Field)
|
||
|
|
assert.Equal(t, "test", result.EnvID)
|
||
|
|
assert.Equal(t, "alice1", result.Name)
|
||
|
|
})
|
||
|
|
|
||
|
|
// Test case 3: POST request with invalid JSON body
|
||
|
|
t.Run("POST request with invalid JSON body", func(t *testing.T) {
|
||
|
|
jsonBody := `{"field": "value"` // Invalid JSON
|
||
|
|
body := io.NopCloser(bytes.NewBufferString(jsonBody))
|
||
|
|
req := NewMockRequest(t, http.MethodPost, body)
|
||
|
|
result, err := decodeReq[TestStruct](req)
|
||
|
|
assert.Error(t, err)
|
||
|
|
assert.Nil(t, result)
|
||
|
|
})
|
||
|
|
|
||
|
|
// Test case 4: Invalid request header
|
||
|
|
t.Run("invalid request header", func(t *testing.T) {
|
||
|
|
jsonBody := `{"field": "value", "env_id": "test", "name": "alice1"}`
|
||
|
|
body := io.NopCloser(bytes.NewBufferString(jsonBody))
|
||
|
|
req := NewMockRequest(t, http.MethodPost, body)
|
||
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||
|
|
result, err := decodeReq[TestStruct](req)
|
||
|
|
assert.Error(t, err)
|
||
|
|
assert.Nil(t, result)
|
||
|
|
})
|
||
|
|
|
||
|
|
// Test case 5: EmptyReq type check
|
||
|
|
t.Run("EmptyReq type check", func(t *testing.T) {
|
||
|
|
req := NewMockRequest(t, http.MethodPost, nil)
|
||
|
|
result, err := decodeReq[EmptyReq](req)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NotNil(t, result)
|
||
|
|
})
|
||
|
|
|
||
|
|
// Test case 5: EmptyReq type check
|
||
|
|
t.Run("HttpRequest type check", func(t *testing.T) {
|
||
|
|
req := &http.Request{
|
||
|
|
Method: http.MethodPost,
|
||
|
|
}
|
||
|
|
result, err := decodeReq[http.Request](req)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, req.Method, result.Method)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkDecodeReq(b *testing.B) {
|
||
|
|
r := &http.Request{
|
||
|
|
Method: http.MethodPost,
|
||
|
|
}
|
||
|
|
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
result, err := decodeReq[http.Request](r)
|
||
|
|
if err != nil {
|
||
|
|
b.Error(err)
|
||
|
|
}
|
||
|
|
if result.Method != http.MethodPost {
|
||
|
|
b.Error("invalid result")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|