cubefs

1
// Copyright 2022 The CubeFS Authors.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
12
// implied. See the License for the specific language governing
13
// permissions and limitations under the License.
14

15
package auditlog
16

17
import (
18
	"bytes"
19
	"net/http"
20
	"strings"
21
	"testing"
22

23
	"github.com/stretchr/testify/require"
24

25
	"github.com/cubefs/cubefs/blobstore/common/rpc"
26
)
27

28
func TestDefaultDecoder_DecodeReq(t *testing.T) {
29
	{
30
		request, err := http.NewRequest(http.MethodPost, "/test", strings.NewReader(`{"name":"test"}`))
31
		require.NoError(t, err)
32

33
		decoder := defaultDecoder{}
34

35
		request.Header.Set(rpc.HeaderContentType, rpc.MIMEJSON)
36
		request.Header.Set(rpc.HeaderContentMD5, "1")
37
		decodeReq := decoder.DecodeReq(request)
38
		require.Equal(t, "/test", decodeReq.Path)
39
		require.Equal(t, rpc.MIMEJSON, decodeReq.Header[rpc.HeaderContentType])
40
		require.Equal(t, "1", decodeReq.Header[rpc.HeaderContentMD5])
41
		require.True(t, len(decodeReq.Params) > 0)
42

43
		request.Header.Set(rpc.HeaderContentType, rpc.MIMEPOSTForm)
44
		decodeReq = decoder.DecodeReq(request)
45
		require.NotNil(t, decodeReq)
46
		require.Equal(t, "/test", decodeReq.Path)
47
		require.Equal(t, rpc.MIMEPOSTForm, decodeReq.Header[rpc.HeaderContentType])
48
		require.True(t, len(decodeReq.Params) > 0)
49
	}
50
	{
51
		body := bytes.NewReader(make([]byte, maxSeekableBodyLength+1))
52
		request, err := http.NewRequest(http.MethodPut, "/nobody", body)
53
		require.NoError(t, err)
54

55
		decoder := defaultDecoder{}
56
		request.Header.Set(rpc.HeaderContentType, rpc.MIMEJSON)
57
		request.Header.Set(rpc.HeaderCrcEncoded, "1")
58
		decodeReq := decoder.DecodeReq(request)
59
		require.Equal(t, "/nobody", decodeReq.Path)
60
		require.True(t, len(decodeReq.Params) == 0)
61
	}
62
}
63

64
func TestDefaultDecoder_RequestCompact(t *testing.T) {
65
	dst := []byte(`{"str":"aaa\nbbb","idx":10}`)
66
	for _, src := range [][]byte{
67
		[]byte(`{"str":"aaa\nbbb","idx":10}`),
68
		[]byte(`{"str": "aaa\nbbb",
69
                 "idx": 10}`),
70
		[]byte(`{
71
	"str": "aaa\nbbb",
72
"idx": 10
73
}`),
74
		[]byte(`    {
75
                "str": "aaa\nbbb",
76
                "idx": 10
77
            }`),
78
	} {
79
		require.True(t, bytes.Equal(compactNewline(src), dst))
80
	}
81
}
82

83
func Benchmark_Decode(b *testing.B) {
84
	request, err := http.NewRequest(http.MethodPost, "/test", strings.NewReader(`{"name":"test"}`))
85
	require.NoError(b, err)
86
	request.Header.Set(rpc.HeaderContentType, rpc.MIMEJSON)
87
	request.Header.Set(rpc.HeaderContentMD5, "1")
88
	request.Header.Set(rpc.HeaderContentMD5, "2")
89

90
	for ii := 0; ii < b.N; ii++ {
91
		decoder := defaultDecoder{}
92
		decoder.DecodeReq(request)
93
	}
94
}
95