zishuo/rtsp-simple-server
1
0
Fork 0
mirror of https://github.com/aler9/rtsp-simple-server synced 2025-10-06 08:06:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

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)