ODWRTW/canape
2
2
Fork 0
Code Issues 3 Pull Requests 1 Releases Activity

Add show.NotFound error

Browse Source
This commit is contained in:
Nicolas Duhamel 2016-02-17 20:02:55 +01:00
parent 5f60096591
commit 6a60ca3c6e
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
shows/shows.go
View File

@ -29,6 +29,10 @@ const (
WHERE shows.imdbid=$1;` WHERE shows.imdbid=$1;`
) )
var (
NotFound = fmt.Errorf("Not found")
)
type Show struct { type Show struct {
sqly.BaseTable sqly.BaseTable
polochon.Show polochon.Show
@ -41,6 +45,9 @@ func Get(db *sqlx.DB, imdbID string) (*Show, error) {
s := &Show{} s := &Show{}
err := db.QueryRowx(getShowQuery, imdbID).StructScan(s) err := db.QueryRowx(getShowQuery, imdbID).StructScan(s)
if err != nil { if err != nil {
if err.Error() == "sql: no rows in result set" {
return nil, NotFound
}
return nil, err return nil, err
} }
return s, nil return s, nil

8
shows/shows_test.go
View File

@ -85,6 +85,12 @@ func init() {
func TestAddRemoveShow(t *testing.T) { func TestAddRemoveShow(t *testing.T) {
sqly.RunWithLastestMigration(db, pgdsn, t, func(db *sqlx.DB, t *testing.T) { sqly.RunWithLastestMigration(db, pgdsn, t, func(db *sqlx.DB, t *testing.T) {
//Get unkown show
_, err := Get(db, "polp")
if err != NotFound {
t.Fatal("NotFound error expected here")
}
nfo := strings.NewReader(showNFO1) nfo := strings.NewReader(showNFO1)
s := &polochon.Show{} s := &polochon.Show{}
polochon.ReadNFO(nfo, s) polochon.ReadNFO(nfo, s)
@ -102,7 +108,7 @@ func TestAddRemoveShow(t *testing.T) {
show := Show{Show: *s} show := Show{Show: *s}
err := show.Add(db) err = show.Add(db)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }