Fix create first login
This commit is contained in:
@@ -2,8 +2,6 @@ package tests
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/base64"
|
||||
"strings"
|
||||
|
||||
"github.com/navidrome/navidrome/model"
|
||||
)
|
||||
@@ -14,6 +12,7 @@ type MockDataStore struct {
|
||||
MockedArtist model.ArtistRepository
|
||||
MockedMediaFile model.MediaFileRepository
|
||||
MockedUser model.UserRepository
|
||||
MockedProperty model.PropertyRepository
|
||||
MockedPlayer model.PlayerRepository
|
||||
MockedTranscoding model.TranscodingRepository
|
||||
}
|
||||
@@ -59,7 +58,10 @@ func (db *MockDataStore) PlayQueue(context.Context) model.PlayQueueRepository {
|
||||
}
|
||||
|
||||
func (db *MockDataStore) Property(context.Context) model.PropertyRepository {
|
||||
return struct{ model.PropertyRepository }{}
|
||||
if db.MockedProperty == nil {
|
||||
db.MockedProperty = &mockedPropertyRepo{}
|
||||
}
|
||||
return db.MockedProperty
|
||||
}
|
||||
|
||||
func (db *MockDataStore) User(context.Context) model.UserRepository {
|
||||
@@ -94,32 +96,3 @@ func (db *MockDataStore) Resource(ctx context.Context, m interface{}) model.Reso
|
||||
func (db *MockDataStore) GC(ctx context.Context, rootFolder string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
type mockedUserRepo struct {
|
||||
model.UserRepository
|
||||
data map[string]*model.User
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) CountAll(qo ...model.QueryOptions) (int64, error) {
|
||||
return int64(len(u.data)), nil
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) Put(usr *model.User) error {
|
||||
if u.data == nil {
|
||||
u.data = make(map[string]*model.User)
|
||||
}
|
||||
if usr.ID == "" {
|
||||
usr.ID = base64.StdEncoding.EncodeToString([]byte(usr.UserName))
|
||||
}
|
||||
usr.Password = usr.NewPassword
|
||||
u.data[strings.ToLower(usr.UserName)] = usr
|
||||
return nil
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) FindByUsername(username string) (*model.User, error) {
|
||||
usr, ok := u.data[strings.ToLower(username)]
|
||||
if !ok {
|
||||
return nil, model.ErrNotFound
|
||||
}
|
||||
return usr, nil
|
||||
}
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
package tests
|
||||
|
||||
import "github.com/navidrome/navidrome/model"
|
||||
|
||||
type mockedPropertyRepo struct {
|
||||
model.UserRepository
|
||||
data map[string]string
|
||||
err error
|
||||
}
|
||||
|
||||
func (p *mockedPropertyRepo) init() {
|
||||
if p.data == nil {
|
||||
p.data = make(map[string]string)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *mockedPropertyRepo) Put(id string, value string) error {
|
||||
if p.err != nil {
|
||||
return p.err
|
||||
}
|
||||
p.init()
|
||||
p.data[id] = value
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *mockedPropertyRepo) Get(id string) (string, error) {
|
||||
if p.err != nil {
|
||||
return "", p.err
|
||||
}
|
||||
p.init()
|
||||
if v, ok := p.data[id]; ok {
|
||||
return v, nil
|
||||
}
|
||||
return "", model.ErrNotFound
|
||||
}
|
||||
|
||||
func (p *mockedPropertyRepo) DefaultGet(id string, defaultValue string) (string, error) {
|
||||
if p.err != nil {
|
||||
return "", p.err
|
||||
}
|
||||
p.init()
|
||||
v, err := p.Get(id)
|
||||
if err != nil {
|
||||
return defaultValue, nil
|
||||
}
|
||||
return v, nil
|
||||
}
|
||||
@@ -0,0 +1,41 @@
|
||||
package tests
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"strings"
|
||||
|
||||
"github.com/navidrome/navidrome/model"
|
||||
)
|
||||
|
||||
type mockedUserRepo struct {
|
||||
model.UserRepository
|
||||
data map[string]*model.User
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) CountAll(qo ...model.QueryOptions) (int64, error) {
|
||||
return int64(len(u.data)), nil
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) Put(usr *model.User) error {
|
||||
if u.data == nil {
|
||||
u.data = make(map[string]*model.User)
|
||||
}
|
||||
if usr.ID == "" {
|
||||
usr.ID = base64.StdEncoding.EncodeToString([]byte(usr.UserName))
|
||||
}
|
||||
usr.Password = usr.NewPassword
|
||||
u.data[strings.ToLower(usr.UserName)] = usr
|
||||
return nil
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) FindByUsername(username string) (*model.User, error) {
|
||||
usr, ok := u.data[strings.ToLower(username)]
|
||||
if !ok {
|
||||
return nil, model.ErrNotFound
|
||||
}
|
||||
return usr, nil
|
||||
}
|
||||
|
||||
func (u *mockedUserRepo) UpdateLastLoginAt(id string) error {
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user