-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Expand file tree
/
Copy pathdebuglogs_internal_test.go
More file actions
103 lines (84 loc) · 3.16 KB
/
Copy pathdebuglogs_internal_test.go
File metadata and controls
103 lines (84 loc) · 3.16 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package agent
import (
"net"
"net/http"
"net/http/httptest"
"os"
"path/filepath"
"runtime"
"testing"
"time"
"github.com/stretchr/testify/require"
"cdr.dev/slog/v3"
"cdr.dev/slog/v3/sloggers/slogtest"
)
func TestHandleHTTPDebugLogsWithAfterCapsResponse(t *testing.T) {
t.Parallel()
logDir := t.TempDir()
activePath := filepath.Join(logDir, "coder-agent.log")
f, err := os.Create(activePath)
require.NoError(t, err)
// A huge active log must not starve the rotated logs.
require.NoError(t, f.Truncate(debugLogsCombinedMaxBytes+1))
require.NoError(t, f.Close())
rotatedPath := filepath.Join(logDir, "coder-agent-2026-05-17T20-00-00.000.log")
require.NoError(t, os.WriteFile(rotatedPath, []byte("rotated marker\n"), 0o600))
rotatedModTime := time.Now().Add(-time.Minute)
require.NoError(t, os.Chtimes(rotatedPath, rotatedModTime, rotatedModTime))
a := &agent{
logger: slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).Leveled(slog.LevelDebug),
logDir: logDir,
}
req := httptest.NewRequest(http.MethodGet, "/debug/logs?after="+time.Now().Add(-time.Hour).UTC().Format(time.RFC3339Nano), nil)
res := httptest.NewRecorder()
a.HandleHTTPDebugLogs(res, req)
require.Equal(t, http.StatusOK, res.Code)
body := res.Body.String()
// Active is capped at its own limit, so the rotated log still fits.
require.Less(t, int64(len(body)), int64(debugLogsCombinedMaxBytes))
require.Contains(t, body, "coder-agent.log")
require.Contains(t, body, "rotated marker")
}
func TestHandleHTTPDebugLogsWithAfterOpenFailure(t *testing.T) {
t.Parallel()
if runtime.GOOS == "windows" {
t.Skip("unix sockets only")
}
logDir, err := os.MkdirTemp("/tmp", "coder-debuglogs-")
require.NoError(t, err)
t.Cleanup(func() {
_ = os.RemoveAll(logDir)
})
activePath := filepath.Join(logDir, "coder-agent.log")
listener, err := net.Listen("unix", activePath)
require.NoError(t, err)
t.Cleanup(func() {
_ = listener.Close()
})
a := &agent{
logger: slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).Leveled(slog.LevelDebug),
logDir: logDir,
}
req := httptest.NewRequest(http.MethodGet, "/debug/logs?after="+time.Now().Add(-time.Minute).UTC().Format(time.RFC3339Nano), nil)
res := httptest.NewRecorder()
a.HandleHTTPDebugLogs(res, req)
require.Equal(t, http.StatusInternalServerError, res.Code)
require.Contains(t, res.Body.String(), "could not open log file")
}
func TestRotatedAgentLogFilesReadsLogDirLiterally(t *testing.T) {
t.Parallel()
root := t.TempDir()
logDir := filepath.Join(root, "logs[abc]")
require.NoError(t, os.Mkdir(logDir, 0o700))
activePath := filepath.Join(logDir, "coder-agent.log")
rotatedPath := filepath.Join(logDir, "coder-agent-2026-05-18T00-00-00.000.log")
require.NoError(t, os.WriteFile(activePath, []byte("active log"), 0o600))
require.NoError(t, os.WriteFile(rotatedPath, []byte("rotated log"), 0o600))
dirRoot, err := os.OpenRoot(logDir)
require.NoError(t, err)
t.Cleanup(func() { _ = dirRoot.Close() })
files, err := rotatedAgentLogFiles(t.Context(), slogtest.Make(t, nil), dirRoot, time.Now().Add(-time.Minute))
require.NoError(t, err)
require.Len(t, files, 1)
require.Equal(t, "coder-agent-2026-05-18T00-00-00.000.log", files[0].name)
}