logs_test.go: replaceWithStaticTimestamps breaks for short PIDs

pkg/logs/logs_test.go

@@ -290,7 +290,7 @@ E0000 00:00:00.000000   00000 logs_test.go:000] "Contextual error" err="fake-err
 
 var (
 	timestampRegexpStdLog = regexp.MustCompile(`\d{4}/\d{2}/\d{2} \d{2}:\d{2}:\d{2}`)
-	timestampRegexpKlog   = regexp.MustCompile(`\d{4} \d{2}:\d{2}:\d{2}\.\d{6}   \d{5}`)
+	timestampRegexpKlog   = regexp.MustCompile(`\d{4} \d{2}:\d{2}:\d{2}\.\d{6} +\d+`)
 	timestampRegexpJSON   = regexp.MustCompile(`"ts":\d+\.\d+`)
 	fileAndLineRegexpJSON = regexp.MustCompile(`"caller":"([^"]+).go:\d+"`)
 	fileAndLineRegexpKlog = regexp.MustCompile(` ([^:]+).go:\d+`)
@@ -318,6 +318,25 @@ func replaceWithStaticTimestamps(input string) string {
 	return input
 }
 
+func Test_replaceWithStaticTimestamps(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    string
+		expected string
+	}{
+		{
+			name:     "klog",
+			input:    `I1018 15:20:42.861239    2386 logs_test.go:13] "Contextual Info Level 3" logger="foo" key="value"`,
+			expected: `I0000 00:00:00.000000   00000 logs_test.go:000] "Contextual Info Level 3" logger="foo" key="value"`,
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			assert.Equal(t, test.expected, replaceWithStaticTimestamps(test.input))
+		})
+	}
+}
+
 func TestLogToSlogWriter(t *testing.T) {
 	// This test makes sure that all the agent's Log.Fatalf calls are correctly
 	// translated to slog.Error calls.