Function TestLoadURL_ErrorCases

query_test.go:194–246

(t *testing.T)

}

func TestLoadURL_ErrorCases(t *testing.T) {
	tests := []struct {
		name            string
		contentEncoding string
		responseBody    []byte
		expectedError   string
	}{
		{
			name:            "ZlibReaderError",
			contentEncoding: "deflate",
			responseBody:    []byte("invalid zlib data"),
			expectedError:   "zlib",
		},
		{
			name:            "EmptyContentEncodingNilBody",
			contentEncoding: "",
			responseBody:    nil,
			expectedError:   "EOF",
		},
		{
			name:            "GzipReaderError",
			contentEncoding: "gzip",
			responseBody:    []byte("invalid gzip data"),
			expectedError:   "gzip",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Create a local HTTP server
			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.Header().Set("Content-Encoding", tt.contentEncoding)
				w.WriteHeader(http.StatusOK)
				w.Write(tt.responseBody)
			}))
			defer server.Close()

			node, err := LoadURL(server.URL)

			// Assert that the returned node is nil
			if node != nil {
				t.Errorf("Expected node to be nil, but got %v", node)
			}

			// Assert that an error is returned
			if err == nil {
				t.Error("Expected an error, but got nil")
			} else if !strings.Contains(err.Error(), tt.expectedError) {
				t.Errorf("Expected error to contain '%s', but got: %v", tt.expectedError, err)
			}
		})
	}
}

func loadHTML(str string) *html.Node {
	node, err := Parse(strings.NewReader(str))