1 files changed, 67 insertions, 0 deletions
diff --git a/api/api_test.go b/api/api_test.go
new file mode 100644
index 0000000..624b684
--- /dev/null
+++ b/api/api_test.go
@@ -0,0 +1,67 @@
+package api_test
+
+import (
+	"encoding/json"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"flyscrape/api"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestScrapeURL(t *testing.T) {
+	svc := &api.ServiceMock{
+		ScrapeURLFunc: func(url string, params map[string]any) (any, error) {
+			return map[string]any{"foo": "bar"}, nil
+		},
+	}
+	h := api.NewHandler(svc)
+
+	r := httptest.NewRequest("POST", "/scrape", strings.NewReader(`{"url": "https://example.com", "data": {"foo":".foo"}}`))
+	w := httptest.NewRecorder()
+	h.ServeHTTP(w, r)
+
+	require.Equal(t, w.Result().StatusCode, http.StatusOK)
+	require.Equal(t, w.Result().Header.Get("Content-Type"), "application/json")
+
+	result := map[string]any{}
+	require.NoError(t, json.NewDecoder(w.Result().Body).Decode(&result))
+	require.Equal(t, result["url"].(string), "https://example.com")
+	require.Equal(t, result["data"].(map[string]any)["foo"], "bar")
+}
+
+func TestScrapeURLInternalServerError(t *testing.T) {
+	svc := &api.ServiceMock{
+		ScrapeURLFunc: func(url string, params map[string]any) (any, error) {
+			return nil, errors.New("whoops")
+		},
+	}
+	h := api.NewHandler(svc)
+
+	r := httptest.NewRequest("POST", "/scrape", strings.NewReader(`{"url": "https://example.com", "data": {"foo":".foo"}}`))
+	w := httptest.NewRecorder()
+	h.ServeHTTP(w, r)
+
+	require.Equal(t, w.Result().StatusCode, http.StatusInternalServerError)
+	require.Equal(t, w.Result().Header.Get("Content-Type"), "application/json")
+}
+
+func TestScrapeURLBadRequest(t *testing.T) {
+	svc := &api.ServiceMock{
+		ScrapeURLFunc: func(url string, params map[string]any) (any, error) {
+			return nil, errors.New("whoops")
+		},
+	}
+	h := api.NewHandler(svc)
+
+	r := httptest.NewRequest("POST", "/scrape", strings.NewReader(`{"}`))
+	w := httptest.NewRecorder()
+	h.ServeHTTP(w, r)
+
+	require.Equal(t, w.Result().StatusCode, http.StatusBadRequest)
+	require.Equal(t, w.Result().Header.Get("Content-Type"), "application/json")
+}