feat: add test cases for Client.Post and Client.Head

presbrey · Aug 9, 2024 · 21c7f5e · 21c7f5e
1 parent 52a1a72
commit 21c7f5e
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/errors_test.go b/errors_test.go
@@ -52,6 +52,32 @@ func TestClientGetError(t *testing.T) {
 	assert.Contains(t, err.Error(), "invalid control character")
 }
 
+func TestClientPostError(t *testing.T) {
+	config := Config{
+		ProxyURLs:   []string{"http://10.255.255.1:8080"},
+		DialTimeout: 5 * time.Second,
+	}
+
+	client, err := NewClient(config)
+	require.NoError(t, err)
+	_, err = client.Post("\000", "text/plain", nil)
+	require.Error(t, err)
+	assert.Contains(t, err.Error(), "invalid control character")
+}
+
+func TestClientHeadError(t *testing.T) {
+	config := Config{
+		ProxyURLs:   []string{"http://10.255.255.1:8080"},
+		DialTimeout: 5 * time.Second,
+	}
+
+	client, err := NewClient(config)
+	require.NoError(t, err)
+	_, err = client.Head("\000")
+	require.Error(t, err)
+	assert.Contains(t, err.Error(), "invalid control character")
+}
+
 func TestClientNewError(t *testing.T) {
 	config := Config{
 		ProxyURLs:   []string{"\000"},