rewrite TestSource

main_test.go

@@ -685,168 +685,132 @@ func TestAuthIpFail(t *testing.T) {
 	require.NotEqual(t, 0, cnt1.wait())
 }
 
-func TestSourceRtsp(t *testing.T) {
+func TestSource(t *testing.T) {
-	for _, proto := range []string{
+	for _, source := range []string{
-		"udp",
+		"rtsp_udp",
-		"tcp",
+		"rtsp_tcp",
+		"rtsps",
+		"rtmp",
 	} {
-		t.Run(proto, func(t *testing.T) {
+		t.Run(source, func(t *testing.T) {
-			p1, ok := testProgram("paths:\n" +
+			switch source {
-				"  all:\n" +
+			case "rtsp_udp", "rtsp_tcp":
-				"    readUser: testuser\n" +
+				p1, ok := testProgram("rtspPort: 8555\n" +
-				"    readPass: testpass\n")
+					"rtpPort: 8100\n" +
-			require.Equal(t, true, ok)
+					"rtcpPort: 8101\n" +
-			defer p1.close()
+					"paths:\n" +
+					"  all:\n" +
+					"    readUser: testuser\n" +
+					"    readPass: testpass\n")
+				require.Equal(t, true, ok)
+				defer p1.close()
 
-			time.Sleep(1 * time.Second)
+				cnt1, err := newContainer("ffmpeg", "source", []string{
+					"-re",
+					"-stream_loop", "-1",
+					"-i", "emptyvideo.ts",
+					"-c", "copy",
+					"-f", "rtsp",
+					"-rtsp_transport", "udp",
+					"rtsp://" + ownDockerIP + ":8555/teststream",
+				})
+				require.NoError(t, err)
+				defer cnt1.close()
 
-			cnt1, err := newContainer("ffmpeg", "source", []string{
+				p2, ok := testProgram("paths:\n" +
-				"-re",
+					"  proxied:\n" +
-				"-stream_loop", "-1",
+					"    source: rtsp://testuser:testpass@localhost:8555/teststream\n" +
-				"-i", "emptyvideo.ts",
+					"    sourceProtocol: " + source[len("rtsp_"):] + "\n" +
-				"-c", "copy",
+					"    sourceOnDemand: yes\n")
-				"-f", "rtsp",
+				require.Equal(t, true, ok)
-				"-rtsp_transport", "udp",
+				defer p2.close()
-				"rtsp://" + ownDockerIP + ":8554/teststream",
-			})
-			require.NoError(t, err)
-			defer cnt1.close()
 
-			time.Sleep(1 * time.Second)
+			case "rtsps":
+				serverCertFpath, err := writeTempFile(serverCert)
+				require.NoError(t, err)
+				defer os.Remove(serverCertFpath)
 
-			p2, ok := testProgram("rtspPort: 8555\n" +
+				serverKeyFpath, err := writeTempFile(serverKey)
-				"rtpPort: 8100\n" +
+				require.NoError(t, err)
-				"rtcpPort: 8101\n" +
+				defer os.Remove(serverKeyFpath)
-				"\n" +
+
-				"paths:\n" +
+				p, ok := testProgram("rtspPort: 8555\n" +
-				"  proxied:\n" +
+					"rtpPort: 8100\n" +
-				"    source: rtsp://testuser:testpass@localhost:8554/teststream\n" +
+					"rtcpPort: 8101\n" +
-				"    sourceProtocol: " + proto + "\n" +
+					"readTimeout: 20s\n" +
-				"    sourceOnDemand: yes\n")
+					"protocols: [tcp]\n" +
-			require.Equal(t, true, ok)
+					"encryption: yes\n" +
-			defer p2.close()
+					"serverCert: " + serverCertFpath + "\n" +
+					"serverKey: " + serverKeyFpath + "\n" +
+					"paths:\n" +
+					"  all:\n" +
+					"    readUser: testuser\n" +
+					"    readPass: testpass\n")
+				require.Equal(t, true, ok)
+				defer p.close()
+
+				cnt1, err := newContainer("ffmpeg", "source", []string{
+					"-re",
+					"-stream_loop", "-1",
+					"-i", "emptyvideo.ts",
+					"-c", "copy",
+					"-f", "rtsp",
+					"rtsps://" + ownDockerIP + ":8555/teststream",
+				})
+				require.NoError(t, err)
+				defer cnt1.close()
+
+				time.Sleep(1 * time.Second)
+
+				p2, ok := testProgram("paths:\n" +
+					"  proxied:\n" +
+					"    source: rtsps://testuser:testpass@localhost:8555/teststream\n" +
+					"    sourceOnDemand: yes\n")
+				require.Equal(t, true, ok)
+				defer p2.close()
+
+			case "rtmp":
+				cnt1, err := newContainer("nginx-rtmp", "rtmpserver", []string{})
+				require.NoError(t, err)
+				defer cnt1.close()
+
+				cnt2, err := newContainer("ffmpeg", "source", []string{
+					"-re",
+					"-stream_loop", "-1",
+					"-i", "emptyvideo.ts",
+					"-c", "copy",
+					"-f", "flv",
+					"rtmp://" + cnt1.ip() + "/stream/test",
+				})
+				require.NoError(t, err)
+				defer cnt2.close()
+
+				time.Sleep(1 * time.Second)
+
+				p, ok := testProgram("paths:\n" +
+					"  proxied:\n" +
+					"    source: rtmp://" + cnt1.ip() + "/stream/test\n" +
+					"    sourceOnDemand: yes\n")
+				require.Equal(t, true, ok)
+				defer p.close()
+			}
 
 			time.Sleep(1 * time.Second)
 
-			cnt2, err := newContainer("ffmpeg", "dest", []string{
+			cnt3, err := newContainer("ffmpeg", "dest", []string{
 				"-rtsp_transport", "udp",
-				"-i", "rtsp://" + ownDockerIP + ":8555/proxied",
+				"-i", "rtsp://" + ownDockerIP + ":8554/proxied",
 				"-vframes", "1",
 				"-f", "image2",
 				"-y", "/dev/null",
 			})
 			require.NoError(t, err)
-			defer cnt2.close()
+			defer cnt3.close()
-
+			require.Equal(t, 0, cnt3.wait())
-			require.Equal(t, 0, cnt2.wait())
 		})
 	}
 }
 
-func TestSourceRtsps(t *testing.T) {
-	serverCertFpath, err := writeTempFile(serverCert)
-	require.NoError(t, err)
-	defer os.Remove(serverCertFpath)
-
-	serverKeyFpath, err := writeTempFile(serverKey)
-	require.NoError(t, err)
-	defer os.Remove(serverKeyFpath)
-
-	p, ok := testProgram("readTimeout: 20s\n" +
-		"protocols: [tcp]\n" +
-		"encryption: yes\n" +
-		"serverCert: " + serverCertFpath + "\n" +
-		"serverKey: " + serverKeyFpath + "\n" +
-		"paths:\n" +
-		"  all:\n" +
-		"    readUser: testuser\n" +
-		"    readPass: testpass\n")
-	require.Equal(t, true, ok)
-	defer p.close()
-
-	time.Sleep(1 * time.Second)
-
-	cnt1, err := newContainer("ffmpeg", "source", []string{
-		"-re",
-		"-stream_loop", "-1",
-		"-i", "emptyvideo.ts",
-		"-c", "copy",
-		"-f", "rtsp",
-		"rtsps://" + ownDockerIP + ":8555/teststream",
-	})
-	require.NoError(t, err)
-	defer cnt1.close()
-
-	time.Sleep(1 * time.Second)
-
-	p2, ok := testProgram("rtspPort: 8556\n" +
-		"rtpPort: 8100\n" +
-		"rtcpPort: 8101\n" +
-		"\n" +
-		"paths:\n" +
-		"  proxied:\n" +
-		"    source: rtsps://testuser:testpass@localhost:8555/teststream\n" +
-		"    sourceOnDemand: yes\n")
-	require.Equal(t, true, ok)
-	defer p2.close()
-
-	time.Sleep(1 * time.Second)
-
-	cnt2, err := newContainer("ffmpeg", "dest", []string{
-		"-rtsp_transport", "udp",
-		"-i", "rtsp://" + ownDockerIP + ":8556/proxied",
-		"-vframes", "1",
-		"-f", "image2",
-		"-y", "/dev/null",
-	})
-	require.NoError(t, err)
-	defer cnt2.close()
-
-	require.Equal(t, 0, cnt2.wait())
-}
-
-func TestSourceRtmp(t *testing.T) {
-	cnt1, err := newContainer("nginx-rtmp", "rtmpserver", []string{})
-	require.NoError(t, err)
-	defer cnt1.close()
-
-	time.Sleep(1 * time.Second)
-
-	cnt2, err := newContainer("ffmpeg", "source", []string{
-		"-re",
-		"-stream_loop", "-1",
-		"-i", "emptyvideo.ts",
-		"-c", "copy",
-		"-f", "flv",
-		"rtmp://" + cnt1.ip() + "/stream/test",
-	})
-	require.NoError(t, err)
-	defer cnt2.close()
-
-	time.Sleep(1 * time.Second)
-
-	p, ok := testProgram("paths:\n" +
-		"  proxied:\n" +
-		"    source: rtmp://" + cnt1.ip() + "/stream/test\n" +
-		"    sourceOnDemand: yes\n")
-	require.Equal(t, true, ok)
-	defer p.close()
-
-	time.Sleep(1 * time.Second)
-
-	cnt3, err := newContainer("ffmpeg", "dest", []string{
-		"-rtsp_transport", "udp",
-		"-i", "rtsp://" + ownDockerIP + ":8554/proxied",
-		"-vframes", "1",
-		"-f", "image2",
-		"-y", "/dev/null",
-	})
-	require.NoError(t, err)
-	defer cnt3.close()
-
-	require.Equal(t, 0, cnt3.wait())
-}
-
 func TestRedirect(t *testing.T) {
 	p1, ok := testProgram("paths:\n" +
 		"  path1:\n" +