Move user properties (like session keys) to their own table

This commit is contained in:
Deluan
2021-06-23 16:47:32 -04:00
parent 265f33ed9d
commit 5001518260
12 changed files with 248 additions and 46 deletions
+8
View File
@@ -16,6 +16,7 @@ type MockDataStore struct {
MockedPlayer model.PlayerRepository
MockedShare model.ShareRepository
MockedTranscoding model.TranscodingRepository
MockedUserProps model.UserPropsRepository
}
func (db *MockDataStore) Album(context.Context) model.AlbumRepository {
@@ -58,6 +59,13 @@ func (db *MockDataStore) PlayQueue(context.Context) model.PlayQueueRepository {
return struct{ model.PlayQueueRepository }{}
}
func (db *MockDataStore) UserProps(context.Context) model.UserPropsRepository {
if db.MockedUserProps == nil {
db.MockedUserProps = &MockedUserPropsRepo{}
}
return db.MockedUserProps
}
func (db *MockDataStore) Property(context.Context) model.PropertyRepository {
if db.MockedProperty == nil {
db.MockedProperty = &MockedPropertyRepo{}
+60
View File
@@ -0,0 +1,60 @@
package tests
import "github.com/navidrome/navidrome/model"
type MockedUserPropsRepo struct {
model.UserPropsRepository
UserID string
data map[string]string
err error
}
func (p *MockedUserPropsRepo) init() {
if p.data == nil {
p.data = make(map[string]string)
}
}
func (p *MockedUserPropsRepo) Put(key string, value string) error {
if p.err != nil {
return p.err
}
p.init()
p.data[p.UserID+"_"+key] = value
return nil
}
func (p *MockedUserPropsRepo) Get(key string) (string, error) {
if p.err != nil {
return "", p.err
}
p.init()
if v, ok := p.data[p.UserID+"_"+key]; ok {
return v, nil
}
return "", model.ErrNotFound
}
func (p *MockedUserPropsRepo) Delete(key string) error {
if p.err != nil {
return p.err
}
p.init()
if _, ok := p.data[p.UserID+"_"+key]; ok {
delete(p.data, p.UserID+"_"+key)
return nil
}
return model.ErrNotFound
}
func (p *MockedUserPropsRepo) DefaultGet(key string, defaultValue string) (string, error) {
if p.err != nil {
return "", p.err
}
p.init()
v, err := p.Get(p.UserID + "_" + key)
if err != nil {
return defaultValue, nil
}
return v, nil
}