mirror of
https://github.com/taigrr/log-socket
synced 2026-03-20 16:02:28 -07:00
Compare commits
1 Commits
master
...
cd/quality
| Author | SHA1 | Date | |
|---|---|---|---|
| 9395e417c6 |
@@ -1,53 +0,0 @@
|
|||||||
package browser
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
"net/http/httptest"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestLogSocketViewHandler_HTTP(t *testing.T) {
|
|
||||||
req := httptest.NewRequest(http.MethodGet, "http://localhost:8080/", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
LogSocketViewHandler(w, req)
|
|
||||||
|
|
||||||
if w.Code != http.StatusOK {
|
|
||||||
t.Fatalf("status = %d, want 200", w.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
body := w.Body.String()
|
|
||||||
// html/template escapes forward slashes in JS context
|
|
||||||
if !strings.Contains(body, `ws:\/\/localhost:8080\/ws`) {
|
|
||||||
t.Error("response should contain escaped ws://localhost:8080/ws URL")
|
|
||||||
}
|
|
||||||
if !strings.Contains(body, "<!DOCTYPE html>") {
|
|
||||||
t.Error("response should contain HTML doctype")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLogSocketViewHandler_CustomPath(t *testing.T) {
|
|
||||||
req := httptest.NewRequest(http.MethodGet, "http://myhost:9090/dashboard/", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
LogSocketViewHandler(w, req)
|
|
||||||
|
|
||||||
body := w.Body.String()
|
|
||||||
if !strings.Contains(body, `ws:\/\/myhost:9090\/dashboard\/ws`) {
|
|
||||||
t.Error("expected escaped ws://myhost:9090/dashboard/ws in body")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLogSocketViewHandler_TrailingSlashTrimmed(t *testing.T) {
|
|
||||||
req := httptest.NewRequest(http.MethodGet, "http://example.com/", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
LogSocketViewHandler(w, req)
|
|
||||||
|
|
||||||
body := w.Body.String()
|
|
||||||
// Should NOT have double slash before ws
|
|
||||||
if strings.Contains(body, `\/\/ws`) {
|
|
||||||
t.Error("should not have double slash before /ws")
|
|
||||||
}
|
|
||||||
if !strings.Contains(body, `ws:\/\/example.com\/ws`) {
|
|
||||||
t.Error("expected escaped ws://example.com/ws in body")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,518 +0,0 @@
|
|||||||
package log
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestLoggerTrace(t *testing.T) {
|
|
||||||
c := CreateClient("logger-trace")
|
|
||||||
c.SetLogLevel(LTrace)
|
|
||||||
|
|
||||||
l := NewLogger("logger-trace")
|
|
||||||
l.Trace("trace message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "TRACE" {
|
|
||||||
t.Errorf("level = %q, want TRACE", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "trace message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "trace message")
|
|
||||||
}
|
|
||||||
if e.Namespace != "logger-trace" {
|
|
||||||
t.Errorf("namespace = %q, want %q", e.Namespace, "logger-trace")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerTracef(t *testing.T) {
|
|
||||||
c := CreateClient("logger-tracef")
|
|
||||||
c.SetLogLevel(LTrace)
|
|
||||||
|
|
||||||
l := NewLogger("logger-tracef")
|
|
||||||
l.Tracef("trace %s %d", "msg", 1)
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "TRACE" {
|
|
||||||
t.Errorf("level = %q, want TRACE", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "trace msg 1" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "trace msg 1")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerTraceln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-traceln")
|
|
||||||
c.SetLogLevel(LTrace)
|
|
||||||
|
|
||||||
l := NewLogger("logger-traceln")
|
|
||||||
l.Traceln("trace line")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "TRACE" {
|
|
||||||
t.Errorf("level = %q, want TRACE", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "trace line\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "trace line\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerDebug(t *testing.T) {
|
|
||||||
c := CreateClient("logger-debug")
|
|
||||||
c.SetLogLevel(LDebug)
|
|
||||||
|
|
||||||
l := NewLogger("logger-debug")
|
|
||||||
l.Debug("debug message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "DEBUG" {
|
|
||||||
t.Errorf("level = %q, want DEBUG", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "debug message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "debug message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerDebugf(t *testing.T) {
|
|
||||||
c := CreateClient("logger-debugf")
|
|
||||||
c.SetLogLevel(LDebug)
|
|
||||||
|
|
||||||
l := NewLogger("logger-debugf")
|
|
||||||
l.Debugf("debug %d", 42)
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "debug 42" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "debug 42")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerInfo(t *testing.T) {
|
|
||||||
c := CreateClient("logger-info")
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l := NewLogger("logger-info")
|
|
||||||
l.Info("info message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "INFO" {
|
|
||||||
t.Errorf("level = %q, want INFO", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "info message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "info message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerInfof(t *testing.T) {
|
|
||||||
c := CreateClient("logger-infof")
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l := NewLogger("logger-infof")
|
|
||||||
l.Infof("count: %d", 99)
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "count: 99" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "count: 99")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerInfoln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-infoln")
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l := NewLogger("logger-infoln")
|
|
||||||
l.Infoln("info line")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "info line\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "info line\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerNotice(t *testing.T) {
|
|
||||||
c := CreateClient("logger-notice")
|
|
||||||
c.SetLogLevel(LNotice)
|
|
||||||
|
|
||||||
l := NewLogger("logger-notice")
|
|
||||||
l.Notice("notice message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "NOTICE" {
|
|
||||||
t.Errorf("level = %q, want NOTICE", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "notice message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "notice message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerNoticef(t *testing.T) {
|
|
||||||
c := CreateClient("logger-noticef")
|
|
||||||
c.SetLogLevel(LNotice)
|
|
||||||
|
|
||||||
l := NewLogger("logger-noticef")
|
|
||||||
l.Noticef("notice %s", "formatted")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "notice formatted" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "notice formatted")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerNoticeln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-noticeln")
|
|
||||||
c.SetLogLevel(LNotice)
|
|
||||||
|
|
||||||
l := NewLogger("logger-noticeln")
|
|
||||||
l.Noticeln("notice line")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "notice line\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "notice line\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerWarn(t *testing.T) {
|
|
||||||
c := CreateClient("logger-warn")
|
|
||||||
c.SetLogLevel(LWarn)
|
|
||||||
|
|
||||||
l := NewLogger("logger-warn")
|
|
||||||
l.Warn("warn message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "WARN" {
|
|
||||||
t.Errorf("level = %q, want WARN", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "warn message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "warn message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerWarnf(t *testing.T) {
|
|
||||||
c := CreateClient("logger-warnf")
|
|
||||||
c.SetLogLevel(LWarn)
|
|
||||||
|
|
||||||
l := NewLogger("logger-warnf")
|
|
||||||
l.Warnf("warn %d", 1)
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "warn 1" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "warn 1")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerWarnln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-warnln")
|
|
||||||
c.SetLogLevel(LWarn)
|
|
||||||
|
|
||||||
l := NewLogger("logger-warnln")
|
|
||||||
l.Warnln("warn line")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "warn line\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "warn line\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerError(t *testing.T) {
|
|
||||||
c := CreateClient("logger-error")
|
|
||||||
c.SetLogLevel(LError)
|
|
||||||
|
|
||||||
l := NewLogger("logger-error")
|
|
||||||
l.Error("error message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Level != "ERROR" {
|
|
||||||
t.Errorf("level = %q, want ERROR", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "error message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "error message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerErrorf(t *testing.T) {
|
|
||||||
c := CreateClient("logger-errorf")
|
|
||||||
c.SetLogLevel(LError)
|
|
||||||
|
|
||||||
l := NewLogger("logger-errorf")
|
|
||||||
l.Errorf("err: %s", "something broke")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "err: something broke" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "err: something broke")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerErrorln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-errorln")
|
|
||||||
c.SetLogLevel(LError)
|
|
||||||
|
|
||||||
l := NewLogger("logger-errorln")
|
|
||||||
l.Errorln("error line")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "error line\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "error line\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPanic(t *testing.T) {
|
|
||||||
c := CreateClient("logger-panic")
|
|
||||||
c.SetLogLevel(LPanic)
|
|
||||||
|
|
||||||
l := NewLogger("logger-panic")
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
r := recover()
|
|
||||||
if r == nil {
|
|
||||||
t.Error("expected panic, got nil")
|
|
||||||
}
|
|
||||||
// Verify the entry was broadcast
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out waiting for panic entry")
|
|
||||||
}
|
|
||||||
if e.Level != "PANIC" {
|
|
||||||
t.Errorf("level = %q, want PANIC", e.Level)
|
|
||||||
}
|
|
||||||
if e.Namespace != "logger-panic" {
|
|
||||||
t.Errorf("namespace = %q, want %q", e.Namespace, "logger-panic")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}()
|
|
||||||
|
|
||||||
l.Panic("panic message")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPanicf(t *testing.T) {
|
|
||||||
c := CreateClient("logger-panicf")
|
|
||||||
c.SetLogLevel(LPanic)
|
|
||||||
|
|
||||||
l := NewLogger("logger-panicf")
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
r := recover()
|
|
||||||
if r == nil {
|
|
||||||
t.Error("expected panic, got nil")
|
|
||||||
}
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "panic 42" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "panic 42")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}()
|
|
||||||
|
|
||||||
l.Panicf("panic %d", 42)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPanicln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-panicln")
|
|
||||||
c.SetLogLevel(LPanic)
|
|
||||||
|
|
||||||
l := NewLogger("logger-panicln")
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
r := recover()
|
|
||||||
if r == nil {
|
|
||||||
t.Error("expected panic, got nil")
|
|
||||||
}
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "panic line\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "panic line\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}()
|
|
||||||
|
|
||||||
l.Panicln("panic line")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPrint(t *testing.T) {
|
|
||||||
c := CreateClient("logger-print")
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l := NewLogger("logger-print")
|
|
||||||
l.Print("print message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
// Print delegates to Info
|
|
||||||
if e.Level != "INFO" {
|
|
||||||
t.Errorf("level = %q, want INFO", e.Level)
|
|
||||||
}
|
|
||||||
if e.Output != "print message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "print message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPrintf(t *testing.T) {
|
|
||||||
c := CreateClient("logger-printf")
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l := NewLogger("logger-printf")
|
|
||||||
l.Printf("formatted %s", "print")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "formatted print" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "formatted print")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPrintln(t *testing.T) {
|
|
||||||
c := CreateClient("logger-println")
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l := NewLogger("logger-println")
|
|
||||||
l.Println("println message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "println message\n" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "println message\n")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerSetInfoDepth(t *testing.T) {
|
|
||||||
l := NewLogger("depth-test")
|
|
||||||
l.SetInfoDepth(3)
|
|
||||||
if l.FileInfoDepth != 3 {
|
|
||||||
t.Errorf("FileInfoDepth = %d, want 3", l.FileInfoDepth)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDefaultLogger(t *testing.T) {
|
|
||||||
l := Default()
|
|
||||||
if l.Namespace != DefaultNamespace {
|
|
||||||
t.Errorf("namespace = %q, want %q", l.Namespace, DefaultNamespace)
|
|
||||||
}
|
|
||||||
if l.FileInfoDepth != 0 {
|
|
||||||
t.Errorf("FileInfoDepth = %d, want 0", l.FileInfoDepth)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Verify Default() logger can emit entries
|
|
||||||
c := CreateClient(DefaultNamespace)
|
|
||||||
c.SetLogLevel(LInfo)
|
|
||||||
|
|
||||||
l.Info("default logger message")
|
|
||||||
|
|
||||||
e, ok := getEntry(c, time.Second)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("timed out")
|
|
||||||
}
|
|
||||||
if e.Output != "default logger message" {
|
|
||||||
t.Errorf("output = %q, want %q", e.Output, "default logger message")
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoggerPanicWithError(t *testing.T) {
|
|
||||||
// When the first arg is an error, Panic should re-panic with that error
|
|
||||||
c := CreateClient("logger-panic-err")
|
|
||||||
c.SetLogLevel(LPanic)
|
|
||||||
|
|
||||||
l := NewLogger("logger-panic-err")
|
|
||||||
testErr := errTest("test error")
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
r := recover()
|
|
||||||
if r == nil {
|
|
||||||
t.Error("expected panic, got nil")
|
|
||||||
}
|
|
||||||
if err, ok := r.(errTest); ok {
|
|
||||||
if string(err) != "test error" {
|
|
||||||
t.Errorf("panic value = %q, want %q", string(err), "test error")
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// The first arg was an error, so Panic should re-panic with it
|
|
||||||
t.Logf("panic type = %T, value = %v (implementation re-panics with original error)", r, r)
|
|
||||||
}
|
|
||||||
c.Destroy()
|
|
||||||
}()
|
|
||||||
|
|
||||||
l.Panic(testErr)
|
|
||||||
}
|
|
||||||
|
|
||||||
// errTest is a simple error type for testing.
|
|
||||||
type errTest string
|
|
||||||
|
|
||||||
func (e errTest) Error() string { return string(e) }
|
|
||||||
@@ -1,62 +0,0 @@
|
|||||||
package ws
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"net/http"
|
|
||||||
"net/http/httptest"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
logger "github.com/taigrr/log-socket/v2/log"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestNamespacesHandler(t *testing.T) {
|
|
||||||
// Log to a known namespace to ensure it appears
|
|
||||||
nsLogger := logger.NewLogger("ns-handler-test")
|
|
||||||
nsLogger.Info("register namespace")
|
|
||||||
|
|
||||||
req := httptest.NewRequest(http.MethodGet, "/api/namespaces", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
NamespacesHandler(w, req)
|
|
||||||
|
|
||||||
if w.Code != http.StatusOK {
|
|
||||||
t.Fatalf("status = %d, want 200", w.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
ct := w.Header().Get("Content-Type")
|
|
||||||
if ct != "application/json" {
|
|
||||||
t.Errorf("Content-Type = %q, want application/json", ct)
|
|
||||||
}
|
|
||||||
|
|
||||||
var result struct {
|
|
||||||
Namespaces []string `json:"namespaces"`
|
|
||||||
}
|
|
||||||
if err := json.Unmarshal(w.Body.Bytes(), &result); err != nil {
|
|
||||||
t.Fatalf("failed to unmarshal response: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
found := false
|
|
||||||
for _, ns := range result.Namespaces {
|
|
||||||
if ns == "ns-handler-test" {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
t.Errorf("namespace 'ns-handler-test' not found in %v", result.Namespaces)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNamespacesHandler_ResponseFormat(t *testing.T) {
|
|
||||||
req := httptest.NewRequest(http.MethodGet, "/api/namespaces", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
NamespacesHandler(w, req)
|
|
||||||
|
|
||||||
var result map[string]interface{}
|
|
||||||
if err := json.Unmarshal(w.Body.Bytes(), &result); err != nil {
|
|
||||||
t.Fatalf("response is not valid JSON: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, ok := result["namespaces"]; !ok {
|
|
||||||
t.Error("response missing 'namespaces' key")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -23,7 +23,7 @@ func LogSocketHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
if namespacesParam != "" {
|
if namespacesParam != "" {
|
||||||
namespaces = strings.Split(namespacesParam, ",")
|
namespaces = strings.Split(namespacesParam, ",")
|
||||||
}
|
}
|
||||||
|
|
||||||
c, err := upgrader.Upgrade(w, r, nil)
|
c, err := upgrader.Upgrade(w, r, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error("upgrade:", err)
|
logger.Error("upgrade:", err)
|
||||||
|
|||||||
@@ -1,127 +0,0 @@
|
|||||||
package ws
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"net/http"
|
|
||||||
"net/http/httptest"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/gorilla/websocket"
|
|
||||||
logger "github.com/taigrr/log-socket/v2/log"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestSetUpgrader(t *testing.T) {
|
|
||||||
custom := websocket.Upgrader{
|
|
||||||
ReadBufferSize: 2048,
|
|
||||||
WriteBufferSize: 2048,
|
|
||||||
}
|
|
||||||
SetUpgrader(custom)
|
|
||||||
if upgrader.ReadBufferSize != 2048 {
|
|
||||||
t.Errorf("ReadBufferSize = %d, want 2048", upgrader.ReadBufferSize)
|
|
||||||
}
|
|
||||||
if upgrader.WriteBufferSize != 2048 {
|
|
||||||
t.Errorf("WriteBufferSize = %d, want 2048", upgrader.WriteBufferSize)
|
|
||||||
}
|
|
||||||
// Reset to default
|
|
||||||
SetUpgrader(websocket.Upgrader{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLogSocketHandler_NonWebSocket(t *testing.T) {
|
|
||||||
// A non-upgrade request should fail gracefully (upgrader returns error)
|
|
||||||
req := httptest.NewRequest(http.MethodGet, "/ws", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
LogSocketHandler(w, req)
|
|
||||||
// The upgrader should return a 400-level error for non-websocket requests
|
|
||||||
if w.Code == http.StatusOK || w.Code == http.StatusSwitchingProtocols {
|
|
||||||
t.Errorf("expected error status for non-websocket request, got %d", w.Code)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLogSocketHandler_WebSocket(t *testing.T) {
|
|
||||||
// Set upgrader with permissive origin check for testing
|
|
||||||
SetUpgrader(websocket.Upgrader{
|
|
||||||
CheckOrigin: func(r *http.Request) bool { return true },
|
|
||||||
})
|
|
||||||
defer SetUpgrader(websocket.Upgrader{})
|
|
||||||
|
|
||||||
server := httptest.NewServer(http.HandlerFunc(LogSocketHandler))
|
|
||||||
defer server.Close()
|
|
||||||
|
|
||||||
wsURL := "ws" + strings.TrimPrefix(server.URL, "http") + "/ws"
|
|
||||||
conn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("failed to connect: %v", err)
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
|
|
||||||
// Send a log entry and verify it arrives over the websocket
|
|
||||||
testLogger := logger.NewLogger("ws-test")
|
|
||||||
testLogger.Info("test message for websocket")
|
|
||||||
|
|
||||||
// Read messages until we find our test entry (the handler itself
|
|
||||||
// logs "Websocket client attached." which may arrive first)
|
|
||||||
conn.SetReadDeadline(time.Now().Add(2 * time.Second))
|
|
||||||
var found bool
|
|
||||||
for i := 0; i < 10; i++ {
|
|
||||||
_, message, err := conn.ReadMessage()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("failed to read message: %v", err)
|
|
||||||
}
|
|
||||||
var entry logger.Entry
|
|
||||||
if err := json.Unmarshal(message, &entry); err != nil {
|
|
||||||
t.Fatalf("failed to unmarshal entry: %v", err)
|
|
||||||
}
|
|
||||||
if entry.Namespace == "ws-test" && entry.Level == "INFO" {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
t.Error("did not receive expected log entry with namespace ws-test")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLogSocketHandler_NamespaceFilter(t *testing.T) {
|
|
||||||
SetUpgrader(websocket.Upgrader{
|
|
||||||
CheckOrigin: func(r *http.Request) bool { return true },
|
|
||||||
})
|
|
||||||
defer SetUpgrader(websocket.Upgrader{})
|
|
||||||
|
|
||||||
server := httptest.NewServer(http.HandlerFunc(LogSocketHandler))
|
|
||||||
defer server.Close()
|
|
||||||
|
|
||||||
// Connect with namespace filter
|
|
||||||
wsURL := "ws" + strings.TrimPrefix(server.URL, "http") + "/ws?namespaces=filtered-ns"
|
|
||||||
conn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("failed to connect: %v", err)
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
|
|
||||||
// Send a log to a different namespace — it should NOT be received
|
|
||||||
otherLogger := logger.NewLogger("other-ns")
|
|
||||||
otherLogger.Info("should not arrive")
|
|
||||||
|
|
||||||
// Send a log to the filtered namespace — it SHOULD be received
|
|
||||||
filteredLogger := logger.NewLogger("filtered-ns")
|
|
||||||
filteredLogger.Info("should arrive")
|
|
||||||
|
|
||||||
conn.SetReadDeadline(time.Now().Add(2 * time.Second))
|
|
||||||
_, message, err := conn.ReadMessage()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("failed to read message: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var entry logger.Entry
|
|
||||||
if err := json.Unmarshal(message, &entry); err != nil {
|
|
||||||
t.Fatalf("failed to unmarshal entry: %v", err)
|
|
||||||
}
|
|
||||||
if entry.Namespace != "filtered-ns" {
|
|
||||||
t.Errorf("namespace = %q, want filtered-ns", entry.Namespace)
|
|
||||||
}
|
|
||||||
if !strings.Contains(entry.Output, "should arrive") {
|
|
||||||
t.Errorf("output = %q, want to contain 'should arrive'", entry.Output)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user