96 lines
2.0 KiB
Go
96 lines
2.0 KiB
Go
//go:build enable_highlevel_tests
|
|
// +build enable_highlevel_tests
|
|
|
|
package testhighlevel
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestHLSServerRead(t *testing.T) {
|
|
p, ok := newInstance("paths:\n" +
|
|
" all_others:\n")
|
|
require.Equal(t, true, ok)
|
|
defer p.Close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "rtsp",
|
|
"rtsp://127.0.0.1:8554/test/stream",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-i", "http://127.0.0.1:8888/test/stream/index.m3u8",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.Equal(t, 0, cnt2.wait())
|
|
}
|
|
|
|
func TestHLSServerAuth(t *testing.T) {
|
|
for _, result := range []string{
|
|
"success",
|
|
"fail",
|
|
} {
|
|
t.Run(result, func(t *testing.T) {
|
|
conf := "paths:\n" +
|
|
" all_others:\n" +
|
|
" readUser: testreader\n" +
|
|
" readPass: testpass\n" +
|
|
" readIPs: [127.0.0.0/16]\n"
|
|
|
|
p, ok := newInstance(conf)
|
|
require.Equal(t, true, ok)
|
|
defer p.Close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "rtsp",
|
|
"rtsp://testpublisher:testpass@127.0.0.1:8554/teststream?param=value",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
var usr string
|
|
if result == "success" {
|
|
usr = "testreader"
|
|
} else {
|
|
usr = "testreader2"
|
|
}
|
|
|
|
tr := &http.Transport{}
|
|
defer tr.CloseIdleConnections()
|
|
hc := &http.Client{Transport: tr}
|
|
|
|
res, err := hc.Get("http://" + usr + ":testpass@127.0.0.1:8888/teststream/index.m3u8?param=value")
|
|
require.NoError(t, err)
|
|
defer res.Body.Close()
|
|
|
|
if result == "success" {
|
|
require.Equal(t, http.StatusOK, res.StatusCode)
|
|
} else {
|
|
require.Equal(t, http.StatusUnauthorized, res.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
}
|