package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-23 23:23:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

add tests against ffmpeg in listen mode (#5)

Browse Source
This commit is contained in:
aler9
2020-10-15 23:15:54 +02:00
parent 5a08848530
commit 4bcdc863e4
1 changed files with 176 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
connclientdial_test.go
View File

@@ -130,10 +130,29 @@ func TestConnClientDialReadTCP(t *testing.T) {
} }
func TestConnClientDialPublishUDP(t *testing.T) { func TestConnClientDialPublishUDP(t *testing.T) {
for _, server := range []string{
"rtsp-simple-server",
"ffmpeg",
} {
t.Run(server, func(t *testing.T) {
switch server {
case "rtsp-simple-server":
cnt1, err := newContainer("rtsp-simple-server", "server", []string{}) cnt1, err := newContainer("rtsp-simple-server", "server", []string{})
require.NoError(t, err) require.NoError(t, err)
defer cnt1.close() defer cnt1.close()
default:
cnt0, err := newContainer("rtsp-simple-server", "server0", []string{})
require.NoError(t, err)
defer cnt0.close()
cnt1, err := newContainer("ffmpeg", "server", []string{
"-fflags nobuffer -re -rtsp_flags listen -i rtsp://localhost:8555/teststream -c copy -f rtsp rtsp://localhost:8554/teststream",
})
require.NoError(t, err)
defer cnt1.close()
}
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
publishDone := make(chan struct{}) publishDone := make(chan struct{})
@@ -165,7 +184,11 @@ func TestConnClientDialPublishUDP(t *testing.T) {
track, err := NewTrackH264(0, sps, pps) track, err := NewTrackH264(0, sps, pps)
require.NoError(t, err) require.NoError(t, err)
conn, err = DialPublish("rtsp://localhost:8554/teststream", port := "8554"
if server == "ffmpeg" {
port = "8555"
}
conn, err = DialPublish("rtsp://localhost:"+port+"/teststream",
StreamProtocolUDP, Tracks{track}) StreamProtocolUDP, Tracks{track})
require.NoError(t, err) require.NoError(t, err)
@@ -183,6 +206,9 @@ func TestConnClientDialPublishUDP(t *testing.T) {
} }
}() }()
if server == "ffmpeg" {
time.Sleep(5 * time.Second)
}
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
cnt3, err := newContainer("ffmpeg", "read", []string{ cnt3, err := newContainer("ffmpeg", "read", []string{
@@ -197,13 +223,34 @@ func TestConnClientDialPublishUDP(t *testing.T) {
code := cnt3.wait() code := cnt3.wait()
require.Equal(t, 0, code) require.Equal(t, 0, code)
})
}
} }
func TestConnClientDialPublishTCP(t *testing.T) { func TestConnClientDialPublishTCP(t *testing.T) {
for _, server := range []string{
"rtsp-simple-server",
"ffmpeg",
} {
t.Run(server, func(t *testing.T) {
switch server {
case "rtsp-simple-server":
cnt1, err := newContainer("rtsp-simple-server", "server", []string{}) cnt1, err := newContainer("rtsp-simple-server", "server", []string{})
require.NoError(t, err) require.NoError(t, err)
defer cnt1.close() defer cnt1.close()
default:
cnt0, err := newContainer("rtsp-simple-server", "server0", []string{})
require.NoError(t, err)
defer cnt0.close()
cnt1, err := newContainer("ffmpeg", "server", []string{
"-fflags nobuffer -re -rtsp_flags listen -i rtsp://localhost:8555/teststream -c copy -f rtsp rtsp://localhost:8554/teststream",
})
require.NoError(t, err)
defer cnt1.close()
}
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
publishDone := make(chan struct{}) publishDone := make(chan struct{})
@@ -235,7 +282,11 @@ func TestConnClientDialPublishTCP(t *testing.T) {
track, err := NewTrackH264(0, sps, pps) track, err := NewTrackH264(0, sps, pps)
require.NoError(t, err) require.NoError(t, err)
conn, err = DialPublish("rtsp://localhost:8554/teststream", port := "8554"
if server == "ffmpeg" {
port = "8555"
}
conn, err = DialPublish("rtsp://localhost:"+port+"/teststream",
StreamProtocolTCP, Tracks{track}) StreamProtocolTCP, Tracks{track})
require.NoError(t, err) require.NoError(t, err)
@@ -253,6 +304,9 @@ func TestConnClientDialPublishTCP(t *testing.T) {
} }
}() }()
if server == "ffmpeg" {
time.Sleep(5 * time.Second)
}
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
cnt3, err := newContainer("ffmpeg", "read", []string{ cnt3, err := newContainer("ffmpeg", "read", []string{
@@ -267,4 +321,6 @@ func TestConnClientDialPublishTCP(t *testing.T) {
code := cnt3.wait() code := cnt3.wait()
require.Equal(t, 0, code) require.Equal(t, 0, code)
})
}
} }