mirror of
https://github.com/alibaba/higress.git
synced 2026-02-27 06:00:51 +08:00
461 lines
13 KiB
Go
461 lines
13 KiB
Go
// Copyright (c) 2024 Alibaba Group Holding Ltd.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/higress-group/proxy-wasm-go-sdk/proxywasm/types"
|
|
"github.com/higress-group/wasm-go/pkg/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// 测试配置：基本配置（数字过期时间）
|
|
var basicConfig = func() json.RawMessage {
|
|
data, _ := json.Marshal(map[string]interface{}{
|
|
"suffix": "jpg|png|jpeg",
|
|
"expires": "3600",
|
|
})
|
|
return data
|
|
}()
|
|
|
|
// 测试配置：最大缓存时间配置
|
|
var maxExpiresConfig = func() json.RawMessage {
|
|
data, _ := json.Marshal(map[string]interface{}{
|
|
"suffix": "css|js",
|
|
"expires": "max",
|
|
})
|
|
return data
|
|
}()
|
|
|
|
// 测试配置：不缓存配置
|
|
var epochExpiresConfig = func() json.RawMessage {
|
|
data, _ := json.Marshal(map[string]interface{}{
|
|
"suffix": "html|htm",
|
|
"expires": "epoch",
|
|
})
|
|
return data
|
|
}()
|
|
|
|
// 测试配置：无后缀限制配置
|
|
var noSuffixConfig = func() json.RawMessage {
|
|
data, _ := json.Marshal(map[string]interface{}{
|
|
"expires": "7200",
|
|
})
|
|
return data
|
|
}()
|
|
|
|
// 测试配置：单后缀配置
|
|
var singleSuffixConfig = func() json.RawMessage {
|
|
data, _ := json.Marshal(map[string]interface{}{
|
|
"suffix": "pdf",
|
|
"expires": "1800",
|
|
})
|
|
return data
|
|
}()
|
|
|
|
// 测试配置：空后缀配置
|
|
var emptySuffixConfig = func() json.RawMessage {
|
|
data, _ := json.Marshal(map[string]string{
|
|
"suffix": "",
|
|
"expires": "3600",
|
|
})
|
|
return data
|
|
}()
|
|
|
|
func TestParseConfig(t *testing.T) {
|
|
test.RunGoTest(t, func(t *testing.T) {
|
|
// 测试基本配置解析
|
|
t.Run("basic config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
config, err := host.GetMatchConfig()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, config)
|
|
})
|
|
|
|
// 测试最大缓存时间配置解析
|
|
t.Run("max expires config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(maxExpiresConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
config, err := host.GetMatchConfig()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, config)
|
|
})
|
|
|
|
// 测试不缓存配置解析
|
|
t.Run("epoch expires config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(epochExpiresConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
config, err := host.GetMatchConfig()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, config)
|
|
})
|
|
|
|
// 测试无后缀限制配置解析
|
|
t.Run("no suffix config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(noSuffixConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
config, err := host.GetMatchConfig()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, config)
|
|
})
|
|
|
|
// 测试单后缀配置解析
|
|
t.Run("single suffix config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(singleSuffixConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
config, err := host.GetMatchConfig()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, config)
|
|
})
|
|
|
|
// 测试空后缀配置解析
|
|
t.Run("empty suffix config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(emptySuffixConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
config, err := host.GetMatchConfig()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, config)
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestOnHttpRequestHeaders(t *testing.T) {
|
|
test.RunTest(t, func(t *testing.T) {
|
|
// 测试基本请求头处理（带查询参数）
|
|
t.Run("request headers with query params", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 设置请求头，包含查询参数
|
|
action := host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/images/photo.jpg?size=large"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试请求头处理（无查询参数）
|
|
t.Run("request headers without query params", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 设置请求头，不包含查询参数
|
|
action := host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/images/photo.png"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试请求头处理（复杂路径）
|
|
t.Run("request headers with complex path", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 设置请求头，包含复杂路径
|
|
action := host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/static/css/main.css?v=1.0.0&theme=dark"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestOnHttpResponseHeaders(t *testing.T) {
|
|
test.RunTest(t, func(t *testing.T) {
|
|
// 测试匹配后缀的响应头处理（数字过期时间）
|
|
t.Run("matching suffix with numeric expires", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/images/photo.jpg"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "image/jpeg"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否添加了缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "maxAge=3600"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试匹配后缀的响应头处理（最大缓存时间）
|
|
t.Run("matching suffix with max expires", func(t *testing.T) {
|
|
host, status := test.NewTestHost(maxExpiresConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/static/main.css"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "text/css"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否添加了缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "maxAge=315360000"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试匹配后缀的响应头处理（不缓存）
|
|
t.Run("matching suffix with epoch expires", func(t *testing.T) {
|
|
host, status := test.NewTestHost(epochExpiresConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/page.html"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "text/html"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否添加了缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "no-cache"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试不匹配后缀的响应头处理
|
|
t.Run("non-matching suffix", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/api/data.json"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "application/json"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否没有添加缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.False(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.False(t, test.HasHeader(responseHeaders, "cache-control"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试无后缀限制的响应头处理
|
|
t.Run("no suffix restriction", func(t *testing.T) {
|
|
host, status := test.NewTestHost(noSuffixConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/any/file.txt"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "text/plain"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否添加了缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "maxAge=7200"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试单后缀匹配
|
|
t.Run("single suffix match", func(t *testing.T) {
|
|
host, status := test.NewTestHost(singleSuffixConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/documents/report.pdf"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "application/pdf"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否添加了缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "maxAge=1800"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
|
|
// 测试空后缀配置
|
|
t.Run("empty suffix config", func(t *testing.T) {
|
|
host, status := test.NewTestHost(emptySuffixConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 先处理请求头
|
|
host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/any/file.xyz"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 处理响应头
|
|
action := host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "application/octet-stream"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 验证是否添加了缓存控制头
|
|
responseHeaders := host.GetResponseHeaders()
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "maxAge=3600"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestCompleteFlow(t *testing.T) {
|
|
test.RunTest(t, func(t *testing.T) {
|
|
t.Run("complete cache control flow", func(t *testing.T) {
|
|
host, status := test.NewTestHost(basicConfig)
|
|
defer host.Reset()
|
|
require.Equal(t, types.OnPluginStartStatusOK, status)
|
|
|
|
// 1. 处理请求头
|
|
action := host.CallOnHttpRequestHeaders([][2]string{
|
|
{":authority", "example.com"},
|
|
{":path", "/images/logo.png"},
|
|
{":method", "GET"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 2. 处理响应头
|
|
action = host.CallOnHttpResponseHeaders([][2]string{
|
|
{":status", "200"},
|
|
{"content-type", "image/png"},
|
|
})
|
|
|
|
// 应该返回 ActionContinue
|
|
require.Equal(t, types.ActionContinue, action)
|
|
|
|
// 3. 验证完整的缓存控制流程
|
|
responseHeaders := host.GetResponseHeaders()
|
|
|
|
// 验证是否添加了必要的缓存控制响应头
|
|
require.True(t, test.HasHeader(responseHeaders, "expires"))
|
|
require.True(t, test.HasHeaderWithValue(responseHeaders, "cache-control", "maxAge=3600"))
|
|
|
|
host.CompleteHttp()
|
|
})
|
|
})
|
|
}
|