Fix tests for Cover service

This commit is contained in:
Deluan
2020-04-09 12:13:54 -04:00
parent 4fc88f23e9
commit 5265d0234f
5 changed files with 44 additions and 21 deletions
+9 -6
View File
@@ -14,7 +14,7 @@ func CreateMockArtistRepo() *MockArtist {
type MockArtist struct {
model.ArtistRepository
data map[string]*model.Artist
data map[string]model.Artist
err bool
}
@@ -22,19 +22,22 @@ func (m *MockArtist) SetError(err bool) {
m.err = err
}
func (m *MockArtist) SetData(j string, size int) {
m.data = make(map[string]*model.Artist)
var l = make([]model.Artist, size)
func (m *MockArtist) SetData(j string) {
m.data = make(map[string]model.Artist)
var l = model.Artists{}
err := json.Unmarshal([]byte(j), &l)
if err != nil {
fmt.Println("ERROR: ", err)
}
for _, a := range l {
m.data[a.ID] = &a
m.data[a.ID] = a
}
}
func (m *MockArtist) Exists(id string) (bool, error) {
if m.err {
return false, errors.New("Error!")
}
_, found := m.data[id]
return found, nil
}
@@ -44,7 +47,7 @@ func (m *MockArtist) Get(id string) (*model.Artist, error) {
return nil, errors.New("Error!")
}
if d, ok := m.data[id]; ok {
return d, nil
return &d, nil
}
return nil, model.ErrNotFound
}