mirror of
https://github.com/binwiederhier/ntfy.git
synced 2025-07-20 10:04:08 +00:00
116 lines
4.1 KiB
Go
116 lines
4.1 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"fmt"
|
|
"github.com/stretchr/testify/require"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestReadBoolParam(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "https://ntfy.sh/mytopic?up=1&firebase=no", nil)
|
|
up := readBoolParam(r, false, "x-up", "up")
|
|
firebase := readBoolParam(r, true, "x-firebase", "firebase")
|
|
require.Equal(t, true, up)
|
|
require.Equal(t, false, firebase)
|
|
|
|
r, _ = http.NewRequest("GET", "https://ntfy.sh/mytopic", nil)
|
|
r.Header.Set("X-Up", "yes")
|
|
r.Header.Set("X-Firebase", "0")
|
|
up = readBoolParam(r, false, "x-up", "up")
|
|
firebase = readBoolParam(r, true, "x-firebase", "firebase")
|
|
require.Equal(t, true, up)
|
|
require.Equal(t, false, firebase)
|
|
|
|
r, _ = http.NewRequest("GET", "https://ntfy.sh/mytopic", nil)
|
|
up = readBoolParam(r, false, "x-up", "up")
|
|
firebase = readBoolParam(r, true, "x-up", "up")
|
|
require.Equal(t, false, up)
|
|
require.Equal(t, true, firebase)
|
|
}
|
|
|
|
func TestRenderHTTPRequest_ValidShort(t *testing.T) {
|
|
r, _ := http.NewRequest("POST", "http://ntfy.sh/mytopic?p=2", strings.NewReader("some message"))
|
|
r.Header.Set("Title", "A title")
|
|
expected := `POST /mytopic?p=2 HTTP/1.1
|
|
Title: A title
|
|
|
|
some message`
|
|
require.Equal(t, expected, renderHTTPRequest(r))
|
|
}
|
|
|
|
func TestRenderHTTPRequest_ValidLong(t *testing.T) {
|
|
body := strings.Repeat("a", 5000)
|
|
r, _ := http.NewRequest("POST", "http://ntfy.sh/mytopic?p=2", strings.NewReader(body))
|
|
r.Header.Set("Accept", "*/*")
|
|
expected := `POST /mytopic?p=2 HTTP/1.1
|
|
Accept: */*
|
|
|
|
` + strings.Repeat("a", 4096) + " ... (peeked 4096 bytes)"
|
|
require.Equal(t, expected, renderHTTPRequest(r))
|
|
}
|
|
|
|
func TestRenderHTTPRequest_InvalidShort(t *testing.T) {
|
|
body := []byte{0xc3, 0x28}
|
|
r, _ := http.NewRequest("GET", "http://ntfy.sh/mytopic/json?since=all", bytes.NewReader(body))
|
|
r.Header.Set("Accept", "*/*")
|
|
expected := `GET /mytopic/json?since=all HTTP/1.1
|
|
Accept: */*
|
|
|
|
(peeked bytes not UTF-8, 2 bytes, hex: c328)`
|
|
require.Equal(t, expected, renderHTTPRequest(r))
|
|
}
|
|
|
|
func TestRenderHTTPRequest_InvalidLong(t *testing.T) {
|
|
body := make([]byte, 5000)
|
|
rand.Read(body)
|
|
r, _ := http.NewRequest("GET", "http://ntfy.sh/mytopic/json?since=all", bytes.NewReader(body))
|
|
r.Header.Set("Accept", "*/*")
|
|
expected := `GET /mytopic/json?since=all HTTP/1.1
|
|
Accept: */*
|
|
|
|
(peeked bytes not UTF-8, peek limit of 4096 bytes reached, hex: ` + fmt.Sprintf("%x", body[:4096]) + ` ...)`
|
|
require.Equal(t, expected, renderHTTPRequest(r))
|
|
}
|
|
|
|
func TestMaybeIgnoreSpecialHeader(t *testing.T) {
|
|
require.Empty(t, maybeIgnoreSpecialHeader("priority", "u=1"))
|
|
require.Empty(t, maybeIgnoreSpecialHeader("Priority", "u=1"))
|
|
require.Empty(t, maybeIgnoreSpecialHeader("Priority", "u=1, i"))
|
|
}
|
|
|
|
func TestMaybeDecodeHeaders(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "http://ntfy.sh/mytopic/json?since=all", nil)
|
|
r.Header.Set("Priority", "u=1") // Cloudflare priority header
|
|
r.Header.Set("X-Priority", "5") // ntfy priority header
|
|
require.Equal(t, "5", readHeaderParam(r, "x-priority", "priority", "p"))
|
|
}
|
|
|
|
func TestExtractIPAddress(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "http://ntfy.sh/mytopic/json?since=all", nil)
|
|
r.RemoteAddr = "10.0.0.1:1234"
|
|
r.Header.Set("X-Forwarded-For", " 1.2.3.4 , 5.6.7.8")
|
|
r.Header.Set("X-Client-IP", "9.10.11.12")
|
|
r.Header.Set("X-Real-IP", "13.14.15.16, 1.1.1.1")
|
|
|
|
trustedProxies := []string{"1.1.1.1"}
|
|
|
|
require.Equal(t, "5.6.7.8", extractIPAddress(r, true, "X-Forwarded-For", trustedProxies).String())
|
|
require.Equal(t, "9.10.11.12", extractIPAddress(r, true, "X-Client-IP", trustedProxies).String())
|
|
require.Equal(t, "13.14.15.16", extractIPAddress(r, true, "X-Real-IP", trustedProxies).String())
|
|
require.Equal(t, "10.0.0.1", extractIPAddress(r, false, "X-Forwarded-For", trustedProxies).String())
|
|
}
|
|
|
|
func TestExtractIPAddress_UnixSocket(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "http://ntfy.sh/mytopic/json?since=all", nil)
|
|
r.RemoteAddr = "@"
|
|
r.Header.Set("X-Forwarded-For", "1.2.3.4, 5.6.7.8, 1.1.1.1")
|
|
|
|
trustedProxies := []string{"1.1.1.1"}
|
|
|
|
require.Equal(t, "5.6.7.8", extractIPAddress(r, true, "X-Forwarded-For", trustedProxies).String())
|
|
require.Equal(t, "0.0.0.0", extractIPAddress(r, false, "X-Forwarded-For", trustedProxies).String())
|
|
}
|