Fix finding playlist from db

bot.go

@@ -324,6 +324,9 @@ func (app *App) AutomateSection(title string) error {
 				} else {
 					playlistId, tracks := findPlaylist(changedWikiText)
 					currentTracks, err := app.db.FindPlaylistBySection(section.title)
+					if err != nil {
+						log.Println("Failed to find tracks from DB", err)
+					}
 					log.Println("Checking if playlist needs updating", currentTracks, tracks)
 					if len(tracks) > 0 && (err != nil || reflect.DeepEqual(currentTracks, tracks)) {
 

db.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 	_ "github.com/lib/pq"
 	"log"
+	"os"
 )
 
 type DB struct {
@@ -13,7 +14,14 @@ type DB struct {
 }
 
 func InitDatabase() *DB {
-	connStr := "user=levyraati dbname=levyraati sslmode=disable"
+	if os.Getenv("PGUSER") == "" {
+		os.Setenv("PGUSER", "levyraati")
+	}
+	if os.Getenv("PGDATABASE") == "" {
+		os.Setenv("PGDATABASE", "levyraati")
+	}
+
+	connStr := "sslmode=disable"
 	var err error
 	database, err := sql.Open("postgres", connStr)
 	if err != nil {
@@ -191,12 +199,12 @@ func (db *DB) FindPlaylistBySection(sectionName string) ([]string, error) {
 	WHERE r.section = $1`
 	row := db.database.QueryRow(query, sectionName)
 	var tracksJson []byte
-	var tracks []string
+	tracks := make([]string, 0)
 	err := row.Scan(&tracksJson)
 	if err != nil {
 		return nil, err
 	}
-	err = json.Unmarshal(tracksJson, tracks)
+	err = json.Unmarshal(tracksJson, &tracks)
 	if err != nil {
 		return nil, err
 	}

db_test.go

@@ -0,0 +1,108 @@
+package main
+
+import (
+	"io/ioutil"
+	"os"
+	"testing"
+)
+
+var db *DB
+
+func initTestCase(t *testing.T) {
+	os.Setenv("PGDATABASE", "levyraati_test")
+	os.Setenv("PGUSER", "levyraati_test")
+	os.Setenv("PGPASSWORD", "levyraati_test")
+	db = InitDatabase()
+
+	execSQLFile(t, "sql/clear-schema.sql")
+	execSQLFile(t, "sql/database.sql")
+	execSQLFile(t, "sql/database-test.sql")
+}
+
+func execSQLFile(t *testing.T, filename string) {
+	data, err := ioutil.ReadFile(filename)
+	if err != nil {
+		t.Fatal("Unable to read setup file", filename)
+	}
+	_, err = db.database.Exec(string(data))
+	if err != nil {
+		t.Fatal("Unable to setup database", err)
+	}
+}
+
+func TestDbConnWorks(t *testing.T) {
+	initTestCase(t)
+	if db == nil {
+		t.Error("Database not initialized")
+	}
+}
+
+func TestFindAllEntries(t *testing.T) {
+	initTestCase(t)
+
+	songs, err := db.FindAllEntries("Lamperi")
+	if err != nil {
+		t.Error("Error while searching for entries")
+	}
+	if len(songs) != 52 {
+		t.Error("Should have found 52 entries, got", len(songs))
+	}
+}
+
+func TestFindAllPanels(t *testing.T) {
+	initTestCase(t)
+
+	panels, err := db.FindAllPanels()
+	if err != nil {
+		t.Error("Error while searching for panels")
+	}
+	if len(panels) != 1 {
+		t.Error("Should have found 1 panel, got", len(panels))
+	}
+}
+
+func TestFindPlaylistBySection(t *testing.T) {
+	initTestCase(t)
+
+	playlistEntries, err := db.FindPlaylistBySection("Viikko 07")
+	if err != nil {
+		t.Error("Error while searching for playlist")
+	}
+	if len(playlistEntries) != 0 {
+		t.Error("Should have found 0 playlistEntries, got", len(playlistEntries))
+	}
+}
+
+func TestUpdatePlaylistBySection(t *testing.T) {
+	initTestCase(t)
+
+	updated, err := db.UpdatePlaylistBySection("Viikko 07", []string{"ID1", "ID2", "ID3", "ID4"})
+	if err != nil {
+		t.Error("Error while updating for panels", err)
+	}
+	if !updated {
+		t.Error("Could not update section")
+	}
+	playlistEntries, err := db.FindPlaylistBySection("Viikko 07")
+	if err != nil {
+		t.Error("Error while searching for playlist", err)
+	}
+	if len(playlistEntries) != 4 {
+		t.Error("Should have found 4 playlistEntries, got", len(playlistEntries))
+	}
+}
+func TestUpdatePlaylistBySectionUnknownSection(t *testing.T) {
+	initTestCase(t)
+
+	updated, err := db.UpdatePlaylistBySection("Viikko joskus", []string{"ID1", "ID2", "ID3", "ID4"})
+	if err != nil {
+		t.Error("Error while updating for panels", err)
+	}
+	if updated {
+		t.Error("Should not update unknown section")
+	}
+	_, err = db.FindPlaylistBySection("Viikko joskus")
+	if err == nil {
+		t.Error("Should be error with unknown section", err)
+	}
+}