60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package tests
|
|
|
|
import "github.com/navidrome/navidrome/model"
|
|
|
|
type MockedUserPropsRepo struct {
|
|
model.UserPropsRepository
|
|
Error error
|
|
data map[string]string
|
|
}
|
|
|
|
func (p *MockedUserPropsRepo) init() {
|
|
if p.data == nil {
|
|
p.data = make(map[string]string)
|
|
}
|
|
}
|
|
|
|
func (p *MockedUserPropsRepo) Put(userId, key string, value string) error {
|
|
if p.Error != nil {
|
|
return p.Error
|
|
}
|
|
p.init()
|
|
p.data[userId+key] = value
|
|
return nil
|
|
}
|
|
|
|
func (p *MockedUserPropsRepo) Get(userId, key string) (string, error) {
|
|
if p.Error != nil {
|
|
return "", p.Error
|
|
}
|
|
p.init()
|
|
if v, ok := p.data[userId+key]; ok {
|
|
return v, nil
|
|
}
|
|
return "", model.ErrNotFound
|
|
}
|
|
|
|
func (p *MockedUserPropsRepo) Delete(userId, key string) error {
|
|
if p.Error != nil {
|
|
return p.Error
|
|
}
|
|
p.init()
|
|
if _, ok := p.data[userId+key]; ok {
|
|
delete(p.data, userId+key)
|
|
return nil
|
|
}
|
|
return model.ErrNotFound
|
|
}
|
|
|
|
func (p *MockedUserPropsRepo) DefaultGet(userId, key string, defaultValue string) (string, error) {
|
|
if p.Error != nil {
|
|
return "", p.Error
|
|
}
|
|
p.init()
|
|
v, err := p.Get(userId, key)
|
|
if err != nil {
|
|
return defaultValue, nil //nolint:nilerr
|
|
}
|
|
return v, nil
|
|
}
|