Skip to content

Commit 5db5990

Browse files
authored
Update discard_test.go
1 parent dbff7e6 commit 5db5990

File tree

1 file changed

+22
-9
lines changed

1 file changed

+22
-9
lines changed

discard_test.go

Lines changed: 22 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -33,16 +33,20 @@ func (m *mockRoundTripper) RoundTrip(req *http.Request) (*http.Response, error)
3333
}, nil
3434
}
3535

36-
func newMockClient() *Client {
37-
c, _ := NewClient(Config{APIKey: "test"})
36+
// updated to accept *testing.T for error handling
37+
func newMockClient(t *testing.T) *Client {
38+
c, err := NewClient(Config{APIKey: "test"})
39+
if err != nil {
40+
t.Fatalf("failed to create client: %v", err)
41+
}
3842
mock := &mockRoundTripper{}
3943
c.HTTPClient.Transport = mock
4044
c.SetFullResponse(true)
4145
return c
4246
}
4347

4448
func TestMakeRequestFlexible(t *testing.T) {
45-
client := newMockClient()
49+
client := newMockClient(t)
4650
resp, err := client.makeRequest("GET", "/api/test", nil, nil)
4751
if err != nil {
4852
t.Fatalf("makeRequest failed: %v", err)
@@ -57,7 +61,7 @@ func TestMakeRequestFlexible(t *testing.T) {
5761
}
5862

5963
func TestMakeFormDataRequest(t *testing.T) {
60-
client := newMockClient()
64+
client := newMockClient(t)
6165
buf := bytes.NewBufferString("dummy data")
6266
files := map[string]io.Reader{"file": buf}
6367
resp, err := client.MakeFormDataRequest("/api/upload", map[string]interface{}{"type": "x"}, files)
@@ -74,7 +78,7 @@ func TestMakeFormDataRequest(t *testing.T) {
7478
}
7579

7680
func TestReflectionCoversAllMethods(t *testing.T) {
77-
client := newMockClient()
81+
client := newMockClient(t)
7882
val := reflect.ValueOf(client)
7983
typ := reflect.TypeOf(client)
8084

@@ -100,28 +104,36 @@ func TestReflectionCoversAllMethods(t *testing.T) {
100104
}
101105

102106
func() {
103-
defer func() { recover() }()
107+
defer func() { _ = recover() }()
104108
m.Func.Call(args)
105109
}()
106110
}
107111
}
108112

109113
func TestTimeoutAndConfig(t *testing.T) {
110-
client, _ := NewClient(Config{APIKey: "x"})
114+
client, err := NewClient(Config{APIKey: "x"})
115+
if err != nil {
116+
t.Fatalf("failed to create client: %v", err)
117+
}
118+
111119
client.SetTimeout(10 * time.Second)
112120
if client.HTTPClient.Timeout != 10*time.Second {
113121
t.Errorf("timeout not applied properly")
114122
}
123+
115124
client.SetFullResponse(true)
116125
if !client.GetFullResponse() {
117126
t.Errorf("FullResponse not set")
118127
}
119128
}
120129

121130
func TestBuildURL(t *testing.T) {
122-
client, _ := NewClient(Config{APIKey: "x"})
123-
urlStr := client.buildURL("/api/test", map[string]interface{}{"q": "1", "x": 5})
131+
client, err := NewClient(Config{APIKey: "x"})
132+
if err != nil {
133+
t.Fatalf("failed to create client: %v", err)
134+
}
124135

136+
urlStr := client.buildURL("/api/test", map[string]interface{}{"q": "1", "x": 5})
125137
u, err := url.Parse(urlStr)
126138
if err != nil {
127139
t.Fatalf("failed to parse URL: %v", err)
@@ -139,6 +151,7 @@ func TestBuildURL(t *testing.T) {
139151
}
140152
}
141153

154+
142155
/*
143156
package discard
144157

0 commit comments

Comments
 (0)