greq/option_test.go at main · gookit/greq · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package greq_test

import (
	"net/http"
	"testing"

	"github.com/gookit/goutil/testutil/assert"
	"github.com/gookit/greq"
)

func TestOptions_Struct(t *testing.T) {
	opt := greq.NewOpt()
	assert.NotNil(t, opt)
	assert.NotNil(t, opt.Header)
	assert.NotNil(t, opt.HeaderM)
	assert.NotNil(t, opt.Query)
}

func TestNewOpt(t *testing.T) {
	opt := greq.NewOpt()
	assert.NotNil(t, opt)
	assert.NotNil(t, opt.Header)
	assert.NotNil(t, opt.Query)
}

func TestNewOpt2(t *testing.T) {
	opt := greq.NewOpt2(nil, "POST")
	assert.NotNil(t, opt)
	assert.Equal(t, "POST", opt.Method)

	// Test with option functions
	opt = greq.NewOpt2([]greq.OptionFn{
		greq.WithMethod("GET"),
		greq.WithContentType("application/json"),
	}, "POST")

	assert.Equal(t, "GET", opt.Method) // Should be overridden by option function
	assert.Equal(t, "application/json", opt.ContentType)
}

func TestWithMethod(t *testing.T) {
	fn := greq.WithMethod("POST")
	opt := &greq.Options{}
	fn(opt)
	assert.Equal(t, "POST", opt.Method)

	// Test with empty method (should not change)
	fn = greq.WithMethod("")
	opt = &greq.Options{Method: "GET"}
	fn(opt)
	assert.Equal(t, "GET", opt.Method) // Should remain unchanged
}

func TestWithContentType(t *testing.T) {
	fn := greq.WithContentType("application/json")
	opt := &greq.Options{}
	fn(opt)
	assert.Equal(t, "application/json", opt.ContentType)
}

func TestWithUserAgent(t *testing.T) {
	fn := greq.WithUserAgent("test-agent/1.0")
	opt := &greq.Options{
		Header: make(http.Header),
	}
	fn(opt)
	assert.Equal(t, "test-agent/1.0", opt.Header.Get("User-Agent"))
}

func TestWithHeader(t *testing.T) {
	fn := greq.WithHeader("X-Custom", "value")
	opt := &greq.Options{
		Header: make(http.Header),
	}
	fn(opt)
	assert.Equal(t, "value", opt.Header.Get("X-Custom"))
}

func TestWithBody(t *testing.T) {
	fn := greq.WithBody("test body")
	opt := &greq.Options{}
	fn(opt)
	assert.Equal(t, "test body", opt.Body)
}

func TestWithData(t *testing.T) {
	fn := greq.WithData("test data")
	opt := &greq.Options{}
	fn(opt)
	assert.Equal(t, "test data", opt.Data)
}

func TestWithTimeout(t *testing.T) {
	// Test that timeout is set correctly
	fn := greq.WithTimeout(5000)
	opt := &greq.Options{}
	fn(opt)
	assert.Equal(t, 5000, opt.Timeout)
}