fix expected stream url

m3u/generate.go

@@ -9,6 +9,10 @@ import (
 	"net/http"
 )
 
+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) {
 	w.Header().Set("Content-Type", "text/plain") // Set the Content-Type header to M3U
 	w.Header().Set("Access-Control-Allow-Origin", "*")
@@ -34,7 +38,7 @@ func GenerateM3UContent(w http.ResponseWriter, r *http.Request) {
 		}
 
 		// Write stream URL
-		_, err = fmt.Fprintf(w, "%s/%s.mp4\n", baseUrl, utils.GetStreamUID(stream.Title))
+		_, err = fmt.Fprintf(w, "%s", generateStreamURL(baseUrl, stream.Title))
 		if err != nil {
 			continue
 		}

m3u/m3u_test.go

@@ -48,10 +48,10 @@ func TestGenerateM3UContent(t *testing.T) {
 	}
 
 	// Check the generated M3U content
-	expectedContent := `#EXTM3U
+	expectedContent := fmt.Sprintf(`#EXTM3U
 #EXTINF:-1 tvg-id="1" tvg-name="TestStream" tvg-logo="http://example.com/logo.png" group-title="TestGroup",TestStream
-http://localhost/stream/teststream.mp4
-`
+%s
+`, generateStreamURL("http:///stream", "TestStream"))
 	if rr.Body.String() != expectedContent {
 		t.Errorf("handler returned unexpected body: got %v want %v",
 			rr.Body.String(), expectedContent)