Add show.NotFound error

shows/shows.go

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

shows/shows_test.go

@@ -85,6 +85,12 @@ func init() {
 
 func TestAddRemoveShow(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)
 		s := &polochon.Show{}
 		polochon.ReadNFO(nfo, s)
@@ -102,7 +108,7 @@ func TestAddRemoveShow(t *testing.T) {
 
 		show := Show{Show: *s}
 
-		err := show.Add(db)
+		err = show.Add(db)
 		if err != nil {
 			t.Fatal(err)
 		}