av/protocol/rtmp/parseurl_test.go

201 lines
5.0 KiB
Go
Raw Normal View History

/*
NAME
parseurl_test.go
DESCRIPTION
See Readme.md
AUTHOR
Dan Kortschak <dan@ausocean.org>
LICENSE
parseurl.go is Copyright (C) 2019 the Australian Ocean Lab (AusOcean)
It is free software: you can redistribute it and/or modify them
under the terms of the GNU General Public License as published by the
Free Software Foundation, either version 3 of the License, or (at your
option) any later version.
It is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
*/
package rtmp
import (
"testing"
)
var parseURLTests = []struct {
url string
wantProtocol int32
wantHost string
wantPort uint16
wantApp string
wantPlaypath string
wantErr error
}{
{
2019-03-25 04:21:03 +03:00
url: "rtmp://addr",
wantErr: errInvalidPath,
},
{
url: "rtmp://addr/",
wantErr: errInvalidElements,
},
{
url: "rtmp://addr/live2",
wantErr: errInvalidElements,
},
{
url: "rtmp://addr/live2/",
wantErr: errInvalidElements,
},
{
url: "rtmp://addr/appname/key",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "key",
},
{
url: "rtmp://addr/appname/instancename",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "instancename",
},
{
url: "rtmp://addr/appname/instancename/",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "instancename",
},
{
url: "rtmp://addr/appname/mp4:path.f4v",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path",
},
{
url: "rtmp://addr/appname/mp4:path.f4v?param1=value1&param2=value2",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path?param1=value1&param2=value2",
},
{
url: "rtmp://addr/appname/mp4:path/to/file.f4v",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path/to/file",
},
{
url: "rtmp://addr/appname/mp4:path/to/file.f4v?param1=value1&param2=value2",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path/to/file?param1=value1&param2=value2",
},
{
url: "rtmp://addr/appname/path.mp4",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path",
},
{
url: "rtmp://addr/appname/path.mp4?param1=value1&param2=value2",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path?param1=value1&param2=value2",
},
{
url: "rtmp://addr/appname/path/to/file.mp4",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path/to/file",
},
{
url: "rtmp://addr/appname/path/to/file.mp4?param1=value1&param2=value2",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "mp4:path/to/file?param1=value1&param2=value2",
},
{
url: "rtmp://addr/appname/path.flv",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "path",
},
{
url: "rtmp://addr/appname/path.flv?param1=value1&param2=value2",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "path?param1=value1&param2=value2",
},
{
url: "rtmp://addr/appname/path/to/file.flv",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "path/to/file",
},
{
url: "rtmp://addr/appname/path/to/file.flv?param1=value1&param2=value2",
wantHost: "addr",
wantPort: 1935,
wantApp: "appname",
wantPlaypath: "path/to/file?param1=value1&param2=value2",
},
}
func TestParseURL(t *testing.T) {
for _, test := range parseURLTests {
func() {
defer func() {
p := recover()
if p != nil {
t.Errorf("unexpected panic for %q: %v", test.url, p)
}
}()
protocol, host, port, app, playpath, err := parseURL(test.url)
if err != test.wantErr {
t.Errorf("unexpected error for %q: got:%v want:%v", test.url, err, test.wantErr)
return
}
if err != nil {
return
}
if protocol != test.wantProtocol {
t.Errorf("unexpected protocol for %q: got:%v want:%v", test.url, protocol, test.wantProtocol)
}
if host != test.wantHost {
t.Errorf("unexpected host for %q: got:%v want:%v", test.url, host, test.wantHost)
}
if port != test.wantPort {
t.Errorf("unexpected port for %q: got:%v want:%v", test.url, port, test.wantPort)
}
if app != test.wantApp {
t.Errorf("unexpected app for %q: got:%v want:%v", test.url, app, test.wantApp)
}
if playpath != test.wantPlaypath {
t.Errorf("unexpected playpath for %q: got:%v want:%v", test.url, playpath, test.wantPlaypath)
}
}()
}
}