add some tests

2023-11-02 19:00:21 +01:00
parent 796ffd1c87
commit 622143211c
5 changed files with 186 additions and 1 deletions
--- a/handlers/api.test.go
+++ b/handlers/api.test.go
@@ -0,0 +1,44 @@
+// BEGIN: 7d5e1f7c7d5e
+package handlers
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestApi(t *testing.T) {
+	app := fiber.New()
+	app.Get("/api/*", Api)
+
+	tests := []struct {
+		name           string
+		url            string
+		expectedStatus int
+	}{
+		{
+			name:           "valid url",
+			url:            "https://www.google.com",
+			expectedStatus: http.StatusOK,
+		},
+		{
+			name:           "invalid url",
+			url:            "invalid-url",
+			expectedStatus: http.StatusBadRequest,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			req := httptest.NewRequest(http.MethodGet, "/api/"+tt.url, nil)
+			resp, err := app.Test(req)
+			assert.NoError(t, err)
+			assert.Equal(t, tt.expectedStatus, resp.StatusCode)
+		})
+	}
+}
+
+// END: 7d5e1f7c7d5e
--- a/handlers/debug.test.go
+++ b/handlers/debug.test.go
@@ -0,0 +1,60 @@
+// BEGIN: 7f8d9e6d4b5c
+package handlers
+
+import (
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+func TestDebug(t *testing.T) {
+	app := fiber.New()
+	app.Get("/debug/*", Debug)
+
+	testCases := []struct {
+		name     string
+		url      string
+		expected string
+	}{
+		{
+			name:     "valid url",
+			url:      "https://www.google.com",
+			expected: "<!doctype html>",
+		},
+		{
+			name:     "invalid url",
+			url:      "invalid-url",
+			expected: "parse invalid-url: invalid URI for request",
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			req := httptest.NewRequest(http.MethodGet, "/debug/"+tc.url, nil)
+			resp, err := app.Test(req)
+			if err != nil {
+				t.Fatalf("unexpected error: %v", err)
+			}
+			defer resp.Body.Close()
+
+			if resp.StatusCode != http.StatusOK {
+				t.Errorf("expected status OK; got %v", resp.Status)
+			}
+
+			body, err := io.ReadAll(resp.Body)
+			if err != nil {
+				t.Fatalf("unexpected error: %v", err)
+			}
+
+			if !strings.Contains(string(body), tc.expected) {
+				t.Errorf("expected body to contain %q; got %q", tc.expected, string(body))
+			}
+		})
+	}
+}
+
+// END: 7f8d9e6d4b5c
--- a/handlers/proxy.test.go
+++ b/handlers/proxy.test.go
@@ -0,0 +1,58 @@
+// BEGIN: 6f8b3f5d5d5d
+package handlers
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"testing"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestProxySite(t *testing.T) {
+	app := fiber.New()
+	app.Get("/:url", ProxySite)
+
+	req := httptest.NewRequest("GET", "/https://example.com", nil)
+	resp, err := app.Test(req)
+	assert.NoError(t, err)
+	assert.Equal(t, http.StatusOK, resp.StatusCode)
+}
+
+func TestRewriteHtml(t *testing.T) {
+	bodyB := []byte(`
+		<html>
+			<head>
+				<title>Test Page</title>
+			</head>
+			<body>
+				<img src="/image.jpg">
+				<script src="/script.js"></script>
+				<a href="/about">About Us</a>
+				<div style="background-image: url('/background.jpg')"></div>
+			</body>
+		</html>
+	`)
+	u := &url.URL{Host: "example.com"}
+
+	expected := `
+		<html>
+			<head>
+				<title>Test Page</title>
+			</head>
+			<body>
+				<img src="/https://example.com/image.jpg">
+				<script script="/https://example.com/script.js"></script>
+				<a href="/https://example.com/about">About Us</a>
+				<div style="background-image: url('/https://example.com/background.jpg')"></div>
+			</body>
+		</html>
+	`
+
+	actual := rewriteHtml(bodyB, u)
+	assert.Equal(t, expected, actual)
+}
+
+// END: 6f8b3f5d5d5d