CodeMirror/ntfy
1
0
Fork 0
mirror of https://github.com/binwiederhier/ntfy.git synced 2024-06-01 18:21:00 +12:00
Code Issues Projects Releases Wiki Activity

Added tests for the user-header authentication

Browse source
This commit is contained in:
Sven 2023-08-01 13:18:54 +02:00
parent 7126af6d7c
commit 5f0c30a151
1 changed files with 55 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
server/server_test.go
View file

@ -6,8 +6,6 @@ import (
"encoding/base64"
"encoding/json"
"fmt"
"golang.org/x/crypto/bcrypt"
"heckel.io/ntfy/user"
"io"
"math/rand"
"net/http"
@ -22,6 +20,9 @@ import (
"testing"
"time"
"golang.org/x/crypto/bcrypt"
"heckel.io/ntfy/user"
"github.com/SherClockHolmes/webpush-go"
"github.com/stretchr/testify/require"
"heckel.io/ntfy/log"
@ -754,6 +755,58 @@ func TestServer_SubscribeWithQueryFilters(t *testing.T) {
require.Equal(t, keepaliveEvent, messages[2].Event)
}
func TestServer_User_Auth_Success_Admin(t *testing.T) {
c := newTestConfigWithAuthFile(t)
header := "X-User-Header"
c.AuthUserHeader = header
c.BehindProxy = true
s := newTestServer(t, c)
require.Nil(t, s.userManager.AddUser("phil", "phil", user.RoleAdmin))
response := request(t, s, "GET", "/mytopic/auth", "", map[string]string{
header: "phil",
})
require.Equal(t, 200, response.Code)
require.Equal(t, `{"success":true}`+"\n", response.Body.String())
}
func TestServer_User_Auth_Unknown_Admin(t *testing.T) {
c := newTestConfigWithAuthFile(t)
header := "X-User-Header"
c.AuthUserHeader = header
c.BehindProxy = true
s := newTestServer(t, c)
response := request(t, s, "GET", "/mytopic/auth", "", map[string]string{
header: "unknown",
})
require.Equal(t, 401, response.Code)
require.Equal(t, 40101, toHTTPError(t, response.Body.String()).Code)
}
func TestServer_User_Auth_Fail_Rate_Limit(t *testing.T) {
c := newTestConfigWithAuthFile(t)
header := "X-User-Header"
c.AuthUserHeader = header
c.BehindProxy = true
c.VisitorAuthFailureLimitBurst = 10
s := newTestServer(t, c)
for i := 0; i < 10; i++ {
response := request(t, s, "PUT", "/announcements", "test", map[string]string{
header: "phil",
})
require.Equal(t, 401, response.Code)
}
response := request(t, s, "PUT", "/announcements", "test", map[string]string{
header: "phil",
})
require.Equal(t, 429, response.Code)
require.Equal(t, 42909, toHTTPError(t, response.Body.String()).Code)
}
func TestServer_Auth_Success_Admin(t *testing.T) {
c := newTestConfigWithAuthFile(t)
s := newTestServer(t, c)