add referer override test

facundoolano · Jul 29, 2024 · 13c6c7d · 13c6c7d
1 parent 81a4bed
commit 13c6c7d
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/parser_test.go b/parser_test.go
@@ -25,3 +25,30 @@ func TestFormatRegex(t *testing.T) {
 	assertEqual(t, result["ua_type"], "desktop")
 	assertEqual(t, result["os"], "Windows")
 }
+
+func TestRefererOverride(t *testing.T) {
+	// when -, leave empty
+	line := `xx.xx.xx.xx - - [24/Jul/2024:00:00:28 +0000] "GET /feed HTTP/1.1" 301 169 "-" "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36"`
+	pattern := formatToRegex(DEFAULT_LOG_FORMAT)
+	result, err := parseLogLine(pattern, line)
+	assertEqual(t, err, nil)
+	assertEqual(t, result["referer"], "")
+
+	// get from field and strip protocol
+	line = `xx.xx.xx.xx - - [24/Jul/2024:00:00:28 +0000] "GET /feed HTTP/1.1" 301 169 "https://olano.dev/feed.xml" "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36"`
+	result, err = parseLogLine(pattern, line)
+	assertEqual(t, err, nil)
+	assertEqual(t, result["referer"], "olano.dev/feed.xml")
+
+	// pick from utm_source when present
+	line = `xx.xx.xx.xx - - [24/Jul/2024:00:00:28 +0000] "GET /feed?utm_source=example.com HTTP/1.1" 301 169 "-" "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36"`
+	result, err = parseLogLine(pattern, line)
+	assertEqual(t, err, nil)
+	assertEqual(t, result["referer"], "example.com")
+
+	// prefer field to utm_source
+	line = `xx.xx.xx.xx - - [24/Jul/2024:00:00:28 +0000] "GET /feed?utm_source=example.com HTTP/1.1" 301 169 "https://olano.dev/feed.xml" "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36"`
+	result, err = parseLogLine(pattern, line)
+	assertEqual(t, err, nil)
+	assertEqual(t, result["referer"], "olano.dev/feed.xml")
+}