chore: Add logger, server handler, and middleware unit tests

Author: mirkobrombin

internal/logger/logger_test.go

@@ -0,0 +1,36 @@
+package logger
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"github.com/mirkobrombin/goup/internal/config"
+)
+
+func TestNewLogger(t *testing.T) {
+	identifier := "test_identifier"
+
+	tmpDir := t.TempDir()
+	config.SetCustomLogDir(tmpDir)
+	logger, err := NewLogger(identifier)
+	if err != nil {
+		t.Fatalf("Error creating new logger: %v", err)
+	}
+
+	if logger == nil {
+		t.Fatalf("Logger is nil")
+	}
+
+	year := time.Now().Year()
+	month := time.Now().Month()
+	day := time.Now().Day()
+
+	logDir := filepath.Join(tmpDir, identifier, fmt.Sprintf("%d", year), fmt.Sprintf("%02d", month))
+	logFile := filepath.Join(logDir, fmt.Sprintf("%02d.log", day))
+	if _, err := os.Stat(logFile); os.IsNotExist(err) {
+		t.Errorf("Expected log file %s to exist", logFile)
+	}
+}

internal/server/handler_test.go

@@ -0,0 +1,92 @@
+package server
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/mirkobrombin/goup/internal/config"
+	log "github.com/sirupsen/logrus"
+)
+
+func TestCreateHandler_Static(t *testing.T) {
+	tmpDir, err := os.MkdirTemp("", "test_static")
+	if err != nil {
+		t.Fatalf("Error creating temp dir: %v", err)
+	}
+	defer os.RemoveAll(tmpDir)
+
+	testFilePath := filepath.Join(tmpDir, "testfile.txt")
+	os.WriteFile(testFilePath, []byte("Test content"), 0644)
+
+	conf := config.SiteConfig{
+		Domain:        "example.com",
+		RootDirectory: tmpDir,
+		CustomHeaders: map[string]string{
+			"X-Test-Header": "TestValue",
+		},
+		RequestTimeout: 60,
+	}
+	logger := log.New()
+	identifier := "test"
+
+	handler, err := createHandler(conf, logger, identifier)
+	if err != nil {
+		t.Fatalf("Error creating handler: %v", err)
+	}
+
+	req := httptest.NewRequest("GET", "http://example.com/testfile.txt", nil)
+	w := httptest.NewRecorder()
+
+	handler.ServeHTTP(w, req)
+
+	if w.Code != 200 {
+		t.Errorf("Expected status code 200, got %d", w.Code)
+	}
+
+	if w.Header().Get("X-Test-Header") != "TestValue" {
+		t.Errorf("Expected custom header X-Test-Header to be 'TestValue', got %s", w.Header().Get("X-Test-Header"))
+	}
+
+	expectedBody := "Test content"
+	if w.Body.String() != expectedBody {
+		t.Errorf("Expected body %q, got %q", expectedBody, w.Body.String())
+	}
+}
+
+func TestCreateHandler_ProxyPass(t *testing.T) {
+	backend := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Write([]byte("Backend Response"))
+	}))
+	defer backend.Close()
+
+	conf := config.SiteConfig{
+		Domain:         "example.com",
+		ProxyPass:      backend.URL,
+		CustomHeaders:  map[string]string{},
+		RequestTimeout: 60,
+	}
+	logger := log.New()
+	identifier := "test"
+
+	handler, err := createHandler(conf, logger, identifier)
+	if err != nil {
+		t.Fatalf("Error creating handler: %v", err)
+	}
+
+	req := httptest.NewRequest("GET", "http://example.com/", nil)
+	w := httptest.NewRecorder()
+
+	handler.ServeHTTP(w, req)
+
+	if w.Code != 200 {
+		t.Errorf("Expected status code 200, got %d", w.Code)
+	}
+
+	expectedBody := "Backend Response"
+	if w.Body.String() != expectedBody {
+		t.Errorf("Expected body %q, got %q", expectedBody, w.Body.String())
+	}
+}

internal/server/middleware/middleware_test.go

@@ -0,0 +1,59 @@
+package middleware
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	log "github.com/sirupsen/logrus"
+)
+
+func TestLoggingMiddleware(t *testing.T) {
+	logger := log.New()
+	logger.Out = httptest.NewRecorder() // Discard output on purpose
+
+	domain := "example.com"
+	identifier := "test"
+
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(200)
+		w.Write([]byte("OK"))
+	})
+
+	middlewareHandler := LoggingMiddleware(logger, domain, identifier, handler)
+
+	req := httptest.NewRequest("GET", "http://example.com/test", nil)
+	w := httptest.NewRecorder()
+
+	middlewareHandler.ServeHTTP(w, req)
+
+	if w.Code != 200 {
+		t.Errorf("Expected status code 200, got %d", w.Code)
+	}
+}
+
+func TestTimeoutMiddleware(t *testing.T) {
+	timeout := 1 * time.Second
+
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		time.Sleep(2 * time.Second)
+		w.Write([]byte("This should timeout"))
+	})
+
+	middlewareHandler := TimeoutMiddleware(timeout, handler)
+
+	req := httptest.NewRequest("GET", "http://example.com/test", nil)
+	w := httptest.NewRecorder()
+
+	middlewareHandler.ServeHTTP(w, req)
+
+	if w.Code != http.StatusServiceUnavailable {
+		t.Errorf("Expected status code %d, got %d", http.StatusServiceUnavailable, w.Code)
+	}
+
+	expectedBody := "Request timed out"
+	if w.Body.String() != expectedBody {
+		t.Errorf("Expected body %q, got %q", expectedBody, w.Body.String())
+	}
+}