From 0d2cb5efd9aea750468d17ac562df1c12e9f08f6 Mon Sep 17 00:00:00 2001
From: Micheal Wilkinson <micheal.wilkinson@xc.betfanatics.com>
Date: Tue, 17 Mar 2026 21:47:44 +0000
Subject: [PATCH] Test deleting the notes

---
 internal/repository/notes.go      |  6 ++-
 internal/repository/notes_test.go | 69 +++++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+), 1 deletion(-)

diff --git a/internal/repository/notes.go b/internal/repository/notes.go
index 12ce97b..289480c 100644
--- a/internal/repository/notes.go
+++ b/internal/repository/notes.go
@@ -20,7 +20,7 @@ func (r *NoteRepository) CreateNote(content string) (models.Note, error) {
 		return models.Note{}, fmt.Errorf("content cannot be empty")
 	}
 	note := models.Note{
-		Content: content,
+		Content:    content,
 		LastUpdate: time.Now(),
 	}
 	return r.store.SaveNote(note)
@@ -46,3 +46,7 @@ func (r *NoteRepository) UpdateNote(id int, content string) (models.Note, error)
 	note.LastUpdate = time.Now()
 	return r.store.SaveNote(note)
 }
+
+func (r *NoteRepository) DeleteNote(id int) error {
+	return fmt.Errorf("not implemented")
+}
diff --git a/internal/repository/notes_test.go b/internal/repository/notes_test.go
index 197e3a4..bac1ace 100644
--- a/internal/repository/notes_test.go
+++ b/internal/repository/notes_test.go
@@ -265,3 +265,72 @@ func TestUpdateNote(t *testing.T) {
 		})
 	}
 }
+
+func TestDeleteNote(t *testing.T) {
+	testcases := []struct {
+		name          string
+		id            int
+		setupNotes    []models.Note
+		expectedNotes []models.Note
+		expectedError bool
+	}{
+		{
+			name: "Delete existing note",
+			id:   1,
+			setupNotes: []models.Note{
+				{ID: 1, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "First note"},
+				{ID: 2, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "Second note"},
+			},
+			expectedNotes: []models.Note{
+				{ID: 2, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "Second note"},
+			},
+			expectedError: false,
+		},
+		{
+			name: "Delete non-existing note",
+			id:   999,
+			setupNotes: []models.Note{
+				{ID: 1, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "First note"},
+				{ID: 2, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "Second note"},
+			},
+			expectedNotes: []models.Note{
+				{ID: 1, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "First note"},
+				{ID: 2, LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC), Content: "Second note"},
+			},
+			expectedError: true,
+		},
+	}
+	for _, tc := range testcases {
+		t.Run(tc.name, func(t *testing.T) {
+			store := &mockNoteStore{Notes: tc.setupNotes}
+			repo := repository.NewNoteRepository(store)
+			err := repo.DeleteNote(tc.id)
+			if tc.expectedError {
+				if err == nil {
+					t.Errorf("expected an error but got none")
+				}
+			} else {
+				if err != nil {
+					t.Errorf("unexpected error: %v", err)
+				}
+			}
+			notes, _ := repo.ListNotes()
+			if len(notes) != len(tc.expectedNotes) {
+				t.Errorf("expected %d notes but got %d", len(tc.expectedNotes), len(notes))
+			}
+			for i, note := range notes {
+				if len(tc.expectedNotes) <= i {
+					t.Errorf("unexpected extra note: %v", note)
+					continue
+				}
+				if note.Content != tc.expectedNotes[i].Content {
+					t.Errorf("expected content %q but got %q", tc.expectedNotes[i].Content, note.Content)
+				}
+				if note.ID != tc.expectedNotes[i].ID {
+					t.Errorf("expected ID %d but got %d", tc.expectedNotes[i].ID, note.ID)
+				}
+				// Purposefully not checking time as it doesn't make sense in context
+			}
+		})
+	}
+}