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
}