From 89b2ec90f2eee067e01a463afd2fc674194bfc58 Mon Sep 17 00:00:00 2001
From: Kharec <sandro@cazzaniga.fr>
Date: Sat, 6 Dec 2025 23:57:23 +0100
Subject: [PATCH] test: add unit test for part two

---
 internal/2018/DayFour/code_test.go | 49 ++++++++++++++++++------------
 1 file changed, 29 insertions(+), 20 deletions(-)

diff --git a/internal/2018/DayFour/code_test.go b/internal/2018/DayFour/code_test.go
index 1345e03..07cf994 100644
--- a/internal/2018/DayFour/code_test.go
+++ b/internal/2018/DayFour/code_test.go
@@ -2,29 +2,38 @@ package dayfour
 
 import "testing"
 
+var testInput = []string{
+	"[1518-11-01 00:00] Guard #10 begins shift",
+	"[1518-11-01 00:05] falls asleep",
+	"[1518-11-01 00:25] wakes up",
+	"[1518-11-01 00:30] falls asleep",
+	"[1518-11-01 00:55] wakes up",
+	"[1518-11-01 23:58] Guard #99 begins shift",
+	"[1518-11-02 00:40] falls asleep",
+	"[1518-11-02 00:50] wakes up",
+	"[1518-11-03 00:05] Guard #10 begins shift",
+	"[1518-11-03 00:24] falls asleep",
+	"[1518-11-03 00:29] wakes up",
+	"[1518-11-04 00:02] Guard #99 begins shift",
+	"[1518-11-04 00:36] falls asleep",
+	"[1518-11-04 00:46] wakes up",
+	"[1518-11-05 00:03] Guard #99 begins shift",
+	"[1518-11-05 00:45] falls asleep",
+	"[1518-11-05 00:55] wakes up",
+}
+
 func TestPartOne(t *testing.T) {
-	input := []string{
-		"[1518-11-01 00:00] Guard #10 begins shift",
-		"[1518-11-01 00:05] falls asleep",
-		"[1518-11-01 00:25] wakes up",
-		"[1518-11-01 00:30] falls asleep",
-		"[1518-11-01 00:55] wakes up",
-		"[1518-11-01 23:58] Guard #99 begins shift",
-		"[1518-11-02 00:40] falls asleep",
-		"[1518-11-02 00:50] wakes up",
-		"[1518-11-03 00:05] Guard #10 begins shift",
-		"[1518-11-03 00:24] falls asleep",
-		"[1518-11-03 00:29] wakes up",
-		"[1518-11-04 00:02] Guard #99 begins shift",
-		"[1518-11-04 00:36] falls asleep",
-		"[1518-11-04 00:46] wakes up",
-		"[1518-11-05 00:03] Guard #99 begins shift",
-		"[1518-11-05 00:45] falls asleep",
-		"[1518-11-05 00:55] wakes up",
-	}
 	expected := 240
-	got := PartOne(input)
+	got := PartOne(testInput)
 	if got != expected {
 		t.Errorf("PartOne() = %d, want %d", got, expected)
 	}
 }
+
+func TestPartTwo(t *testing.T) {
+	expected := 4455
+	got := PartTwo(testInput)
+	if got != expected {
+		t.Errorf("PartTwo() = %d, want %d", got, expected)
+	}
+}