fix lint errors

database/database_test.go

@@ -10,7 +10,6 @@ func TestSaveAndLoadFromSQLite(t *testing.T) {
 	if err != nil {
 		t.Errorf("InitializeSQLite returned error: %v", err)
 	}
-	defer DeleteSQLite(db, "test")
 
 	// Test LoadFromSQLite with existing data in the database
 	expected := []StreamInfo{{
@@ -74,6 +73,11 @@ func TestSaveAndLoadFromSQLite(t *testing.T) {
 			t.Errorf("GetStreams returned %+v, expected %+v", result[i], expectedStream)
 		}
 	}
+
+	err = DeleteSQLite(db, "test")
+	if err != nil {
+		t.Errorf("DeleteSQLite returned error: %v", err)
+	}
 }
 
 // streamInfoEqual checks if two StreamInfo objects are equal.

m3u/generate.go

@@ -1,6 +1,7 @@
 package m3u
 
 import (
+	"database/sql"
 	"fmt"
 	"log"
 	"m3u-stream-merger/database"
@@ -12,8 +13,8 @@ func generateStreamURL(baseUrl string, title string) string {
 	return fmt.Sprintf("%s/%s.mp4\n", baseUrl, utils.GetStreamUID(title))
 }
 
-func GenerateM3UContent(w http.ResponseWriter, r *http.Request) {
-	streams, err := database.GetStreams()
+func GenerateM3UContent(w http.ResponseWriter, r *http.Request, db *sql.DB) {
+	streams, err := database.GetStreams(db)
 	if err != nil {
 		log.Println(fmt.Errorf("GetStreams error: %v", err))
 	}

m3u/m3u_test.go

@@ -24,7 +24,6 @@ func TestGenerateM3UContent(t *testing.T) {
 	if err != nil {
 		t.Errorf("InitializeSQLite returned error: %v", err)
 	}
-	defer database.DeleteSQLite(db, "test")
 
 	_, err = database.InsertStream(db, stream)
 	if err != nil {
@@ -39,7 +38,9 @@ func TestGenerateM3UContent(t *testing.T) {
 
 	// Create a ResponseRecorder to record the response
 	rr := httptest.NewRecorder()
-	handler := http.HandlerFunc(GenerateM3UContent)
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		GenerateM3UContent(w, r, db)
+	})
 
 	// Call the ServeHTTP method of the handler to execute the test
 	handler.ServeHTTP(rr, req)
@@ -65,6 +66,11 @@ func TestGenerateM3UContent(t *testing.T) {
 		t.Errorf("handler returned unexpected body: got %v want %v",
 			rr.Body.String(), expectedContent)
 	}
+
+	err = database.DeleteSQLite(db, "test")
+	if err != nil {
+		t.Errorf("DeleteSQLite returned error: %v", err)
+	}
 }
 
 func TestParseM3UFromURL(t *testing.T) {
@@ -92,10 +98,9 @@ http://example.com/fox
 	if err != nil {
 		t.Errorf("InitializeSQLite returned error: %v", err)
 	}
-	defer database.DeleteSQLite(db, "test")
 
 	// Test the parseM3UFromURL function with the mock server URL
-	err = ParseM3UFromURL(mockServer.URL, 0, 1)
+	err = ParseM3UFromURL(db, mockServer.URL, 0, 1)
 	if err != nil {
 		t.Errorf("Error parsing M3U from URL: %v", err)
 	}
@@ -149,6 +154,11 @@ http://example.com/fox
 			t.FailNow()
 		}
 	}
+
+	err = database.DeleteSQLite(db, "test")
+	if err != nil {
+		t.Errorf("DeleteSQLite returned error: %v", err)
+	}
 }
 
 // streamInfoEqual checks if two StreamInfo objects are equal.

main.go

@@ -88,7 +88,9 @@ func main() {
 	}
 
 	// HTTP handlers
-	http.HandleFunc("/playlist.m3u", m3u.GenerateM3UContent)
+	http.HandleFunc("/playlist.m3u", func(w http.ResponseWriter, r *http.Request) {
+		m3u.GenerateM3UContent(w, r, db)
+	})
 	http.HandleFunc("/stream/", func(w http.ResponseWriter, r *http.Request) {
 		mp4Handler(w, r, db)
 	})