From a9b61f84070cc7ca0d6e26f187c745619a91422a Mon Sep 17 00:00:00 2001
From: Philipp Tanlak <philipp.tanlak@gmail.com>
Date: Thu, 27 Jul 2023 19:03:41 +0200
Subject: init

---
 api/api_test.go | 67 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 api/api_test.go

(limited to 'api/api_test.go')

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")
+}
-- 
cgit v1.2.3