31 lines
602 B
Go
31 lines
602 B
Go
package repository_test
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.hrafn.xyz/aether/notes/internal/models"
|
|
)
|
|
|
|
type mockNoteStore struct {
|
|
Notes []models.Note
|
|
}
|
|
|
|
func (m *mockNoteStore) SaveNote(note models.Note) (models.Note, error) {
|
|
note.ID = len(m.Notes) + 1
|
|
m.Notes = append(m.Notes, note)
|
|
return note, nil
|
|
}
|
|
|
|
func (m *mockNoteStore) GetNoteByID(id int) (models.Note, error) {
|
|
for _, note := range m.Notes {
|
|
if note.ID == id {
|
|
return note, nil
|
|
}
|
|
}
|
|
return models.Note{}, fmt.Errorf("note with ID %d not found", id)
|
|
}
|
|
|
|
func (m *mockNoteStore) GetAllNotes() ([]models.Note, error) {
|
|
return m.Notes, nil
|
|
}
|