lancet/netutil/http_test.go

package netutil

import (
	"bytes"
	"encoding/json"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"net/url"
	"os"
	"testing"
	"time"

	"github.com/duke-git/lancet/v2/internal"
)

func TestHttpGet(t *testing.T) {
	url := "https://jsonplaceholder.typicode.com/todos/1"
	header := map[string]string{
		"Content-Type": "application/json",
	}

	resp, err := HttpGet(url, header)
	if err != nil {
		t.Log("net error: " + err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestHttpPost(t *testing.T) {
	url := "https://jsonplaceholder.typicode.com/todos"
	header := map[string]string{
		"Content-Type": "application/json",
	}
	type Todo struct {
		UserId int    `json:"userId"`
		Title  string `json:"title"`
	}
	todo := Todo{1, "TestAddToDo"}
	bodyParams, _ := json.Marshal(todo)

	resp, err := HttpPost(url, header, nil, bodyParams)
	if err != nil {
		t.Log("net error: " + err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestHttpPostFormData(t *testing.T) {
	apiUrl := "https://jsonplaceholder.typicode.com/todos"
	header := map[string]string{
		"Content-Type": "application/x-www-form-urlencoded",
	}

	postData := url.Values{}
	postData.Add("userId", "1")
	postData.Add("title", "TestToDo")

	resp, err := HttpPost(apiUrl, header, nil, postData)
	if err != nil {
		t.Log("net error: " + err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestHttpPut(t *testing.T) {
	url := "https://jsonplaceholder.typicode.com/todos/1"
	header := map[string]string{
		"Content-Type": "application/json",
	}
	type Todo struct {
		Id     int    `json:"id"`
		UserId int    `json:"userId"`
		Title  string `json:"title"`
	}
	todo := Todo{1, 1, "TestPutToDo"}
	bodyParams, _ := json.Marshal(todo)

	resp, err := HttpPut(url, header, nil, bodyParams)
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestHttpPatch(t *testing.T) {
	url := "https://jsonplaceholder.typicode.com/todos/1"
	header := map[string]string{
		"Content-Type": "application/json",
	}
	type Todo struct {
		Id     int    `json:"id"`
		UserId int    `json:"userId"`
		Title  string `json:"title"`
	}
	todo := Todo{1, 1, "TestPatchToDo"}
	bodyParams, _ := json.Marshal(todo)

	resp, err := HttpPatch(url, header, nil, bodyParams)
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestHttpDelete(t *testing.T) {
	url := "https://jsonplaceholder.typicode.com/todos/1"
	resp, err := HttpDelete(url)
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestConvertMapToQueryString(t *testing.T) {
	t.Parallel()

	assert := internal.NewAssert(t, "TestConvertMapToQueryString")

	var m = map[string]any{
		"c": 3,
		"a": 1,
		"b": 2,
	}

	assert.Equal("a=1&b=2&c=3", ConvertMapToQueryString(m))
}

func TestParseResponse(t *testing.T) {
	url := "https://jsonplaceholder.typicode.com/todos/1"
	header := map[string]string{
		"Content-Type": "application/json",
	}

	resp, err := HttpGet(url, header)
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	type Todo struct {
		Id        int    `json:"id"`
		UserId    int    `json:"userId"`
		Title     string `json:"title"`
		Completed bool   `json:"completed"`
	}

	toDoResp := &Todo{}
	err = ParseHttpResponse(resp, toDoResp)
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	t.Log("response: ", toDoResp)
}

func TestHttpClient_Get(t *testing.T) {
	t.Parallel()

	assert := internal.NewAssert(t, "TestHttpClient_Get")

	request := &HttpRequest{
		RawURL: "https://jsonplaceholder.typicode.com/todos/1",
		Method: "GET",
	}

	httpClient := NewHttpClient()
	resp, err := httpClient.SendRequest(request)
	if err != nil || resp.StatusCode != 200 {
		t.Log("net error: ", err.Error())
		return
	}

	type Todo struct {
		UserId    int    `json:"userId"`
		Id        int    `json:"id"`
		Title     string `json:"title"`
		Completed bool   `json:"completed"`
	}

	var todo Todo
	err = httpClient.DecodeResponse(resp, &todo)
	if err != nil {
		t.FailNow()
	}

	assert.Equal(1, todo.Id)
}

func TestHttpClent_Post(t *testing.T) {
	header := http.Header{}
	header.Add("Content-Type", "multipart/form-data")

	postData := url.Values{}
	postData.Add("userId", "1")
	postData.Add("title", "testItem")

	request := &HttpRequest{
		RawURL:   "https://jsonplaceholder.typicode.com/todos",
		Method:   "POST",
		Headers:  header,
		FormData: postData,
	}

	httpClient := NewHttpClient()
	resp, err := httpClient.SendRequest(request)
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	defer resp.Body.Close()
	t.Log("response status:", resp.StatusCode)
}

func TestStructToUrlValues(t *testing.T) {
	t.Parallel()

	assert := internal.NewAssert(t, "TestStructToUrlValues")

	type CommReq struct {
		Version string `json:"version"`
	}

	type TodoQuery struct {
		Id      int    `json:"id"`
		UserId  int    `json:"userId"`
		Name    string `json:"name,omitempty"`
		CommReq `json:",inline"`
	}
	item1 := TodoQuery{
		Id:     1,
		UserId: 123,
		Name:   "",
		CommReq: CommReq{
			Version: "1.0",
		},
	}

	todoValues, err := StructToUrlValues(item1)
	if err != nil {
		t.Errorf("params is invalid: %v", err)
	}

	assert.Equal("1", todoValues.Get("id"))
	assert.Equal("123", todoValues.Get("userId"))
	assert.Equal("", todoValues.Get("name"))
	assert.Equal("1.0", todoValues.Get("version"))
}

func handleFileRequest(t *testing.T, _ http.ResponseWriter, r *http.Request) {
	err := r.ParseMultipartForm(1024)
	if err != nil {
		t.Fatal(err)
	}

	key1 := r.FormValue("key1")
	expectedKey1 := "value1"
	if key1 != expectedKey1 {
		t.Fatalf("expected %s, got %s", expectedKey1, key1)
	}

	key2 := r.FormValue("key2")
	expectedKey2 := "value2"
	if key2 != expectedKey2 {
		t.Fatalf("expected %s, got %s", expectedKey2, key2)
	}

	file, header, err := r.FormFile("image")
	if err != nil {
		t.Fatal(err)
	}

	expectedFileName := "testImage.jpg"
	if header.Filename != expectedFileName {
		t.Fatalf("expected %s, got %s", expectedFileName, header.Filename)
	}

	defer file.Close()

	content, err := ioutil.ReadAll(file)
	if err != nil {
		t.Fatal(err)
	}

	expectedContent := []byte("file content")
	if !bytes.Equal(content, expectedContent) {
		t.Fatalf("expected %s, got %s", string(expectedContent), string(content))
	}
}

func TestSendRequestWithFileContent(t *testing.T) {
	handler := http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
		handleFileRequest(t, writer, request)
	})

	server := httptest.NewServer(handler)
	defer server.Close()

	client := NewHttpClient()
	request := &HttpRequest{
		RawURL:   server.URL,
		Method:   "POST",
		File:     &File{Content: []byte("file content"), FieldName: "image", FileName: "testImage.jpg"},
		FormData: url.Values{"key1": {"value1"}, "key2": {"value2"}},
	}

	resp, err := client.SendRequest(request)
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		t.Fatalf("expected %d, got %d", http.StatusOK, resp.StatusCode)
	}
}

func TestSendRequestWithFilePath(t *testing.T) {
	handler := http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
		handleFileRequest(t, writer, request)
	})

	server := httptest.NewServer(handler)
	defer server.Close()

	tmpFile, err := ioutil.TempFile("", "testImage.jpg")
	if err != nil {
		t.Fatal(err)
	}

	defer os.Remove(tmpFile.Name())

	tmpFile.Write([]byte("file content"))
	tmpFile.Close()

	client := NewHttpClient()
	request := &HttpRequest{
		RawURL:   server.URL,
		Method:   "POST",
		File:     &File{Path: tmpFile.Name(), FieldName: "image", FileName: "testImage.jpg"},
		FormData: url.Values{"key1": {"value1"}, "key2": {"value2"}},
	}

	resp, err := client.SendRequest(request)
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		t.Fatalf("expected %d, got %d", http.StatusOK, resp.StatusCode)
	}
}

func TestHttpClient_AsyncSendRequest(t *testing.T) {
	t.Parallel()

	assert := internal.NewAssert(t, "TestHttpClient_Get")

	request := &HttpRequest{
		RawURL: "https://jsonplaceholder.typicode.com/todos/1",
		Method: "GET",
	}

	httpClient := NewHttpClient()
	respCh := make(chan *http.Response, 20)
	errCh := make(chan error, 20)
	for i := 0; i < 50; i++ {
		httpClient.AsyncSendRequest(request, respCh, errCh)
	}
	for i := 0; i < 50; i++ {
		select {
		case resp := <-respCh:
			type Todo struct {
				UserId    int    `json:"userId"`
				Id        int    `json:"id"`
				Title     string `json:"title"`
				Completed bool   `json:"completed"`
			}
			var todo Todo
			err := httpClient.DecodeResponse(resp, &todo)
			if err != nil {
				t.FailNow()
			}
			assert.Equal(1, todo.Id)
		case err := <-errCh:
			if err != nil {
				t.Log("net error: ", err.Error())
			}
		}
	}
}

func TestProxy(t *testing.T) {
	config := &HttpClientConfig{
		HandshakeTimeout: 20 * time.Second,
		ResponseTimeout:  40 * time.Second,
		// Use the proxy ip to add it here
		//Proxy: &url.URL{
		//	Scheme: "http",
		//	Host:   "46.17.63.166:18888",
		//},
	}
	httpClient := NewHttpClientWithConfig(config)
	resp, err := httpClient.Get("https://www.ipplus360.com/getLocation")
	if err != nil {
		t.Log("net error: ", err.Error())
		return
	}

	if resp.StatusCode != http.StatusOK {
		t.Fatalf("expected %d, got %d", http.StatusOK, resp.StatusCode)
	}
}