From 49bde10c0b4d1c296960fbe94e30672da42df0e5 Mon Sep 17 00:00:00 2001
From: Philipp Tanlak <philipp.tanlak@gmail.com>
Date: Mon, 18 Sep 2023 21:22:57 +0200
Subject: set default user-agent to flyscrape/0.1

---
 fetch_test.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'fetch_test.go')

diff --git a/fetch_test.go b/fetch_test.go
index 041ade2..b32ac0f 100644
--- a/fetch_test.go
+++ b/fetch_test.go
@@ -53,3 +53,16 @@ func TestFetchProxiedFetch(t *testing.T) {
 	require.NoError(t, err)
 	require.Equal(t, html, "foobar")
 }
+
+func TestFetchUserAgent(t *testing.T) {
+	var userAgent string
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		userAgent = r.Header.Get("User-Agent")
+	}))
+
+	fetch := flyscrape.Fetch()
+
+	_, err := fetch(srv.URL)
+	require.NoError(t, err)
+	require.Equal(t, "flyscrape/0.1", userAgent)
+}
-- 
cgit v1.2.3