mirror of
https://github.com/aler9/gortsplib
synced 2025-10-05 15:16:51 +08:00
server: test invalid session
This commit is contained in:
@@ -199,3 +199,11 @@ type ErrServerSessionLinkedToOtherConn struct{}
|
|||||||
func (e ErrServerSessionLinkedToOtherConn) Error() string {
|
func (e ErrServerSessionLinkedToOtherConn) Error() string {
|
||||||
return "session is linked to another connection"
|
return "session is linked to another connection"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ErrServerInvalidSession is an error that can be returned by a server.
|
||||||
|
type ErrServerInvalidSession struct{}
|
||||||
|
|
||||||
|
// Error implements the error interface.
|
||||||
|
func (e ErrServerInvalidSession) Error() string {
|
||||||
|
return "invalid session"
|
||||||
|
}
|
||||||
|
@@ -659,3 +659,57 @@ func TestServerGetSetParameter(t *testing.T) {
|
|||||||
require.Equal(t, base.StatusOK, res.StatusCode)
|
require.Equal(t, base.StatusOK, res.StatusCode)
|
||||||
require.Equal(t, []byte("param1: 123456\r\n"), res.Body)
|
require.Equal(t, []byte("param1: 123456\r\n"), res.Body)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestServerErrorInvalidSession(t *testing.T) {
|
||||||
|
for _, method := range []base.Method{
|
||||||
|
base.Play,
|
||||||
|
base.Record,
|
||||||
|
base.Pause,
|
||||||
|
base.Teardown,
|
||||||
|
} {
|
||||||
|
t.Run(string(method), func(t *testing.T) {
|
||||||
|
s := &Server{
|
||||||
|
Handler: &testServerHandler{
|
||||||
|
onPlay: func(ctx *ServerHandlerOnPlayCtx) (*base.Response, error) {
|
||||||
|
return &base.Response{
|
||||||
|
StatusCode: base.StatusOK,
|
||||||
|
}, nil
|
||||||
|
},
|
||||||
|
onRecord: func(ctx *ServerHandlerOnRecordCtx) (*base.Response, error) {
|
||||||
|
return &base.Response{
|
||||||
|
StatusCode: base.StatusOK,
|
||||||
|
}, nil
|
||||||
|
},
|
||||||
|
onPause: func(ctx *ServerHandlerOnPauseCtx) (*base.Response, error) {
|
||||||
|
return &base.Response{
|
||||||
|
StatusCode: base.StatusOK,
|
||||||
|
}, nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := s.Start("127.0.0.1:8554")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer s.Close()
|
||||||
|
|
||||||
|
conn, err := net.Dial("tcp", "localhost:8554")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer conn.Close()
|
||||||
|
bconn := bufio.NewReadWriter(bufio.NewReader(conn), bufio.NewWriter(conn))
|
||||||
|
|
||||||
|
err = base.Request{
|
||||||
|
Method: method,
|
||||||
|
URL: base.MustParseURL("rtsp://localhost:8554/teststream"),
|
||||||
|
Header: base.Header{
|
||||||
|
"CSeq": base.HeaderValue{"1"},
|
||||||
|
},
|
||||||
|
}.Write(bconn.Writer)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
var res base.Response
|
||||||
|
err = res.Read(bconn.Reader)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, base.StatusBadRequest, res.StatusCode)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@@ -333,7 +333,7 @@ func (sc *ServerConn) handleRequest(req *base.Request) (*base.Response, error) {
|
|||||||
if ss == nil {
|
if ss == nil {
|
||||||
return &base.Response{
|
return &base.Response{
|
||||||
StatusCode: base.StatusBadRequest,
|
StatusCode: base.StatusBadRequest,
|
||||||
}, fmt.Errorf("terminated")
|
}, liberrors.ErrServerInvalidSession{}
|
||||||
}
|
}
|
||||||
|
|
||||||
rres := make(chan requestRes)
|
rres := make(chan requestRes)
|
||||||
@@ -359,7 +359,7 @@ func (sc *ServerConn) handleRequest(req *base.Request) (*base.Response, error) {
|
|||||||
if ss == nil {
|
if ss == nil {
|
||||||
return &base.Response{
|
return &base.Response{
|
||||||
StatusCode: base.StatusBadRequest,
|
StatusCode: base.StatusBadRequest,
|
||||||
}, fmt.Errorf("terminated")
|
}, liberrors.ErrServerInvalidSession{}
|
||||||
}
|
}
|
||||||
|
|
||||||
rres := make(chan requestRes)
|
rres := make(chan requestRes)
|
||||||
@@ -385,7 +385,7 @@ func (sc *ServerConn) handleRequest(req *base.Request) (*base.Response, error) {
|
|||||||
if ss == nil {
|
if ss == nil {
|
||||||
return &base.Response{
|
return &base.Response{
|
||||||
StatusCode: base.StatusBadRequest,
|
StatusCode: base.StatusBadRequest,
|
||||||
}, fmt.Errorf("terminated")
|
}, liberrors.ErrServerInvalidSession{}
|
||||||
}
|
}
|
||||||
|
|
||||||
rres := make(chan requestRes)
|
rres := make(chan requestRes)
|
||||||
@@ -408,7 +408,7 @@ func (sc *ServerConn) handleRequest(req *base.Request) (*base.Response, error) {
|
|||||||
if ss == nil {
|
if ss == nil {
|
||||||
return &base.Response{
|
return &base.Response{
|
||||||
StatusCode: base.StatusBadRequest,
|
StatusCode: base.StatusBadRequest,
|
||||||
}, fmt.Errorf("terminated")
|
}, liberrors.ErrServerInvalidSession{}
|
||||||
}
|
}
|
||||||
|
|
||||||
rres := make(chan requestRes)
|
rres := make(chan requestRes)
|
||||||
|
Reference in New Issue
Block a user