73 lines
2.5 KiB
Go
73 lines
2.5 KiB
Go
package diagnostics
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestShouldRetainRunDirectoryMatrix(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
input RetentionDecisionInput
|
|
want bool
|
|
}{
|
|
{name: "always success keeps", input: RetentionDecisionInput{RetentionMode: "always", RunSucceeded: true}, want: true},
|
|
{name: "always failure keeps", input: RetentionDecisionInput{RetentionMode: "always", RunSucceeded: false}, want: true},
|
|
{name: "never success keeps", input: RetentionDecisionInput{RetentionMode: "never", RunSucceeded: true}, want: true},
|
|
{name: "never failure keeps", input: RetentionDecisionInput{RetentionMode: "never", RunSucceeded: false}, want: true},
|
|
{name: "auto success no skips removes", input: RetentionDecisionInput{RetentionMode: "auto", RunSucceeded: true, HasSkippedCorrections: false}, want: false},
|
|
{name: "auto success skips keeps", input: RetentionDecisionInput{RetentionMode: "auto", RunSucceeded: true, HasSkippedCorrections: true}, want: true},
|
|
{name: "auto failure keeps", input: RetentionDecisionInput{RetentionMode: "auto", RunSucceeded: false}, want: true},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got := ShouldRetainRunDirectory(tc.input)
|
|
if got != tc.want {
|
|
t.Fatalf("unexpected retain decision: got=%v want=%v", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestApplyRetentionRemovesWhenDecisionSaysRemove(t *testing.T) {
|
|
workDir := t.TempDir()
|
|
runPath := filepath.Join(workDir, "run-test")
|
|
if err := os.Mkdir(runPath, 0o755); err != nil {
|
|
t.Fatalf("mkdir run path: %v", err)
|
|
}
|
|
|
|
runDir := &RunDirectory{path: runPath, retention: "auto"}
|
|
err := runDir.ApplyRetention(RetentionDecisionInput{RunSucceeded: true, HasSkippedCorrections: false})
|
|
if err != nil {
|
|
t.Fatalf("ApplyRetention failed: %v", err)
|
|
}
|
|
|
|
if _, err := os.Stat(runPath); !os.IsNotExist(err) {
|
|
t.Fatalf("expected run directory removed, stat err=%v", err)
|
|
}
|
|
}
|
|
|
|
func TestApplyRetentionReturnsRemovalError(t *testing.T) {
|
|
parent := t.TempDir()
|
|
runPath := filepath.Join(parent, "run-test")
|
|
if err := os.Mkdir(runPath, 0o755); err != nil {
|
|
t.Fatalf("mkdir run path: %v", err)
|
|
}
|
|
|
|
// Make parent non-writable so removing child fails.
|
|
if err := os.Chmod(parent, 0o500); err != nil {
|
|
t.Fatalf("chmod parent: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
_ = os.Chmod(parent, 0o700)
|
|
})
|
|
|
|
runDir := &RunDirectory{path: runPath, retention: "auto"}
|
|
err := runDir.ApplyRetention(RetentionDecisionInput{RunSucceeded: true, HasSkippedCorrections: false})
|
|
if err == nil {
|
|
t.Fatalf("expected removal error, got nil")
|
|
}
|
|
}
|