1 files changed, 66 insertions, 0 deletions
diff --git a/worker/swift/handlers_test.go b/worker/swift/handlers_test.go
new file mode 100644
index 0000000..f9c2e0b
--- /dev/null
+++ b/worker/swift/handlers_test.go
@@ -0,0 +1,66 @@
+package main
+
+import (
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/labstack/echo/v4"
+)
+
+func TestNewBadRequestError(t *testing.T) {
+	httpErr := newBadRequestError(errors.New("test error"))
+	if httpErr.Code != http.StatusBadRequest {
+		t.Errorf("status code = %d, want %d", httpErr.Code, http.StatusBadRequest)
+	}
+	msg, ok := httpErr.Message.(string)
+	if !ok {
+		t.Fatalf("expected string message, got %T", httpErr.Message)
+	}
+	if !strings.Contains(msg, "test error") {
+		t.Errorf("message = %q, want it to contain %q", msg, "test error")
+	}
+}
+
+func TestHandleExec_InvalidJSON(t *testing.T) {
+	e := echo.New()
+	req := httptest.NewRequest(http.MethodPost, "/exec", strings.NewReader("not json"))
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
+	rec := httptest.NewRecorder()
+	c := e.NewContext(req, rec)
+
+	err := handleExec(c)
+	if err == nil {
+		t.Fatal("expected error, got nil")
+	}
+	httpErr, ok := err.(*echo.HTTPError)
+	if !ok {
+		t.Fatalf("expected *echo.HTTPError, got %T", err)
+	}
+	if httpErr.Code != http.StatusBadRequest {
+		t.Errorf("status code = %d, want %d", httpErr.Code, http.StatusBadRequest)
+	}
+}
+
+func TestHandleExec_ZeroMaxDuration(t *testing.T) {
+	e := echo.New()
+	body := `{"code":"print(1)","code_hash":"abc","stdin":"","max_duration_ms":0}`
+	req := httptest.NewRequest(http.MethodPost, "/exec", strings.NewReader(body))
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
+	rec := httptest.NewRecorder()
+	c := e.NewContext(req, rec)
+
+	err := handleExec(c)
+	if err == nil {
+		t.Fatal("expected error, got nil")
+	}
+	httpErr, ok := err.(*echo.HTTPError)
+	if !ok {
+		t.Fatalf("expected *echo.HTTPError, got %T", err)
+	}
+	if httpErr.Code != http.StatusBadRequest {
+		t.Errorf("status code = %d, want %d", httpErr.Code, http.StatusBadRequest)
+	}
+}