mirror of
https://github.com/Forceu/Gokapi.git
synced 2026-02-08 01:18:36 -06:00
f36d39e728
* Added Manage_Logs API permission, added API endpoint to delete logs, added more logging, added filtering and deletion of logs in UI, fixed bug that prevented setting Manage_Users API permission on new API key
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
package logging
|
|
|
|
import (
|
|
"github.com/forceu/gokapi/internal/models"
|
|
"github.com/forceu/gokapi/internal/test"
|
|
"github.com/forceu/gokapi/internal/test/testconfiguration"
|
|
"net/http/httptest"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
testconfiguration.Create(false)
|
|
exitVal := m.Run()
|
|
testconfiguration.Delete()
|
|
os.Exit(exitVal)
|
|
}
|
|
|
|
func TestGetIpAddress(t *testing.T) {
|
|
r := httptest.NewRequest("GET", "/test", nil)
|
|
test.IsEqualString(t, getIpAddress(r), "192.0.2.1")
|
|
r = httptest.NewRequest("GET", "/test", nil)
|
|
r.RemoteAddr = "127.0.0.1:1234"
|
|
test.IsEqualString(t, getIpAddress(r), "127.0.0.1")
|
|
r.RemoteAddr = "invalid"
|
|
test.IsEqualString(t, getIpAddress(r), "Unknown IP")
|
|
r.Header.Add("X-REAL-IP", "1.1.1.1")
|
|
test.IsEqualString(t, getIpAddress(r), "1.1.1.1")
|
|
r.Header.Add("X-FORWARDED-FOR", "1.1.1.2")
|
|
test.IsEqualString(t, getIpAddress(r), "1.1.1.2")
|
|
}
|
|
|
|
func TestInit(t *testing.T) {
|
|
Init("test")
|
|
test.IsEqualString(t, logPath, "test/log.txt")
|
|
}
|
|
|
|
func TestAddString(t *testing.T) {
|
|
test.FileDoesNotExist(t, "test/log.txt")
|
|
createLogEntry(categoryInfo, "Hello", true)
|
|
test.FileExists(t, "test/log.txt")
|
|
content, _ := os.ReadFile("test/log.txt")
|
|
test.IsEqualBool(t, strings.Contains(string(content), "UTC [info] Hello"), true)
|
|
}
|
|
|
|
func TestAddDownload(t *testing.T) {
|
|
file := models.File{
|
|
Id: "testId",
|
|
Name: "testName",
|
|
}
|
|
r := httptest.NewRequest("GET", "/test", nil)
|
|
r.Header.Set("User-Agent", "testAgent")
|
|
r.Header.Add("X-REAL-IP", "1.1.1.1")
|
|
LogDownload(file, r, true)
|
|
// Need sleep, as LogDownload() is non-blocking
|
|
time.Sleep(500 * time.Millisecond)
|
|
content, _ := os.ReadFile("test/log.txt")
|
|
test.IsEqualBool(t, strings.Contains(string(content), "UTC [download] testName, IP 1.1.1.1, ID testId, Useragent testAgent"), true)
|
|
r.Header.Add("X-REAL-IP", "2.2.2.2")
|
|
LogDownload(file, r, false)
|
|
// Need sleep, as LogDownload() is non-blocking
|
|
time.Sleep(500 * time.Millisecond)
|
|
content, _ = os.ReadFile("test/log.txt")
|
|
test.IsEqualBool(t, strings.Contains(string(content), "2.2.2.2"), false)
|
|
}
|