78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package forwarding
|
|
|
|
import (
|
|
"bufio"
|
|
"os/exec"
|
|
"testing"
|
|
"time"
|
|
|
|
"gitlab.crans.org/nounous/ghostream/stream"
|
|
"gitlab.crans.org/nounous/ghostream/stream/srt"
|
|
)
|
|
|
|
// TestServeSRT Serve a SRT server, stream content during 5 seconds and ensure that it is well received
|
|
func TestForwardStream(t *testing.T) {
|
|
// Start virtual RTMP server with ffmpeg
|
|
forwardedFfmpeg := exec.Command("ffmpeg", "-hide_banner", "-loglevel", "error",
|
|
"-y", // allow overwrite /dev/null
|
|
"-listen", "1", "-i", "rtmp://127.0.0.1:1936/live/app", "-f", "null", "-c", "copy", "/dev/null")
|
|
forwardingErrOutput, err := forwardedFfmpeg.StderrPipe()
|
|
if err != nil {
|
|
t.Fatal("Error while querying ffmpeg forwardingOutput:", err)
|
|
}
|
|
if err := forwardedFfmpeg.Start(); err != nil {
|
|
t.Fatal("Error while starting forwarding stream ffmpeg instance:", err)
|
|
}
|
|
|
|
go func() {
|
|
scanner := bufio.NewScanner(forwardingErrOutput)
|
|
for scanner.Scan() {
|
|
t.Fatalf("ffmpeg virtual RTMP server returned %s", scanner.Text())
|
|
}
|
|
}()
|
|
|
|
cfg := make(map[string][]string)
|
|
cfg["demo"] = []string{"rtmp://127.0.0.1:1936/live/app"}
|
|
|
|
// Register forwarding stream list
|
|
streams := make(map[string]*stream.Stream)
|
|
go Serve(streams, cfg)
|
|
|
|
// Serve SRT Server without authentification backend
|
|
go srt.Serve(streams, nil, &srt.Options{Enabled: true, ListenAddress: ":9712", MaxClients: 2})
|
|
|
|
ffmpeg := exec.Command("ffmpeg", "-hide_banner", "-loglevel", "error",
|
|
"-re", "-f", "lavfi", "-i", "testsrc=size=640x480:rate=10",
|
|
"-f", "flv", "srt://127.0.0.1:9712?streamid=demo:")
|
|
|
|
errOutput, err := ffmpeg.StderrPipe()
|
|
if err != nil {
|
|
t.Fatal("Error while querying ffmpeg forwardingOutput:", err)
|
|
}
|
|
|
|
if err := ffmpeg.Start(); err != nil {
|
|
t.Fatal("Error while starting ffmpeg:", err)
|
|
}
|
|
|
|
go func() {
|
|
scanner := bufio.NewScanner(errOutput)
|
|
for scanner.Scan() {
|
|
t.Fatalf("ffmpeg virtual source returned %s", scanner.Text())
|
|
}
|
|
}()
|
|
|
|
time.Sleep(5 * time.Second) // Delay is in nanoseconds, here 5s
|
|
|
|
err = ffmpeg.Process.Kill()
|
|
if err != nil {
|
|
t.Fatalf("Error while killing ffmpeg: already died?")
|
|
}
|
|
|
|
err = forwardedFfmpeg.Process.Kill()
|
|
if err != nil {
|
|
t.Fatalf("Error while killing ffmpeg: already died?")
|
|
}
|
|
|
|
// TODO Kill SRT server
|
|
}
|