diff --git a/internal/2016/DayFive/code_test.go b/internal/2016/DayFive/code_test.go index e5847b1..8a8100e 100644 --- a/internal/2016/DayFive/code_test.go +++ b/internal/2016/DayFive/code_test.go @@ -32,3 +32,27 @@ func TestPartOne(t *testing.T) { t.Errorf("PartOne() printed %q, want %q", got, expected) } } + +func TestPartTwo(t *testing.T) { + expected := "05ace8e3" + + oldStdout := os.Stdout + r, w, err := os.Pipe() + if err != nil { + t.Fatalf("Failed to create pipe: %v", err) + } + os.Stdout = w + + PartTwo(testInput) + + _ = w.Close() + os.Stdout = oldStdout + + var buffer bytes.Buffer + _, _ = buffer.ReadFrom(r) + got := strings.TrimSpace(buffer.String()) + + if got != expected { + t.Errorf("PartTwo() printed %q, want %q", got, expected) + } +}