package tests
import (
"context"
"github.com/navidrome/navidrome/model"
)
type MockedPropertyRepo struct {
model.PropertyRepository
Error error
Data map[string]string
}
func (p *MockedPropertyRepo) init() {
if p.Data == nil {
p.Data = make(map[string]string)
}
}
func (p *MockedPropertyRepo) Put(_ context.Context, id string, value string) error {
if p.Error != nil {
return p.Error
}
p.init()
p.Data[id] = value
return nil
}
func (p *MockedPropertyRepo) Get(_ context.Context, id string) (string, error) {
if p.Error != nil {
return "", p.Error
}
p.init()
if v, ok := p.Data[id]; ok {
return v, nil
}
return "", model.ErrNotFound
}
func (p *MockedPropertyRepo) Delete(_ context.Context, id string) error {
if p.Error != nil {
return p.Error
}
p.init()
if _, ok := p.Data[id]; ok {
delete(p.Data, id)
return nil
}
return model.ErrNotFound
}
func (p *MockedPropertyRepo) DefaultGet(ctx context.Context, id string, defaultValue string) (string, error) {
if p.Error != nil {
return "", p.Error
}
p.init()
v, err := p.Get(ctx, id)
if err != nil {
return defaultValue, nil
}
return v, nil
}
