More tests, more refactoring, more reflection (ugh)

2016-02-29 22:17:54 -05:00
parent 3f72466e24
commit 64b9b3f13a
7 changed files with 179 additions and 44 deletions
@@ -10,7 +10,7 @@ type Album struct {

 func NewAlbumRepository() *Album {
 	r := &Album{}
-	r.table = "album"
+	r.init("album", &models.Album{})
 	return r
 }

@@ -22,9 +22,9 @@ func (r *Album) Put(m *models.Album) error {
 }

 func (r *Album) Get(id string) (*models.Album, error) {
-	rec := &models.Album{}
-	err := r.loadEntity(id, rec)
-	return rec, err
+	var rec interface{}
+	rec, err := r.readEntity(id)
+	return rec.(*models.Album), err
 }

 func (r *Album) GetByName(name string) (*models.Album, error) {