External/mediamtx
1
0
Fork 1
mirror of https://github.com/bluenviron/mediamtx.git synced 2025-12-20 02:00:05 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

rtmp: split net.Conn from rtmp.Conn

Browse source
This commit is contained in:
aler9 2022-07-09 17:25:33 +02:00
parent bf1f45df32
commit 67e8a01d56
12 changed files with 141 additions and 123 deletions
Download patch file Download diff file Expand all files Collapse all files

20
internal/core/rtmp_server_test.go
View file

@ -1,8 +1,9 @@
package core
import (
"context"
"io"
"net"
"net/url"
"testing"
"time"
@ -134,10 +135,13 @@ func TestRTMPServerAuth(t *testing.T) {
defer a.close()
}
conn, err := rtmp.DialContext(context.Background(),
"rtmp://127.0.0.1/teststream?user=testreader&pass=testpass&param=value")
u, err := url.Parse("rtmp://127.0.0.1:1935/teststream?user=testreader&pass=testpass&param=value")
require.NoError(t, err)
defer conn.Close()
nconn, err := net.Dial("tcp", u.Host)
require.NoError(t, err)
defer nconn.Close()
conn := rtmp.NewClientConn(nconn, u)
err = conn.ClientHandshake(true)
require.NoError(t, err)
@ -219,9 +223,13 @@ func TestRTMPServerAuthFail(t *testing.T) {
time.Sleep(1 * time.Second)
conn, err := rtmp.DialContext(context.Background(), "rtmp://127.0.0.1/teststream?user=testuser&pass=testpass")
u, err := url.Parse("rtmp://127.0.0.1:1935/teststream?user=testuser&pass=testpass")
require.NoError(t, err)
defer conn.Close()
nconn, err := net.Dial("tcp", u.Host)
require.NoError(t, err)
defer nconn.Close()
conn := rtmp.NewClientConn(nconn, u)
err = conn.ClientHandshake(true)
require.Equal(t, err, io.EOF)