mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.0 KiB
Go
59 lines
1.0 KiB
Go
4 years ago
|
// Copyright 2021, Chef. All rights reserved.
|
||
|
|
// https://github.com/q191201771/lal
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by a MIT-style license
|
||
|
|
// that can be found in the License file.
|
||
|
|
//
|
||
|
|
// Author: Chef (191201771@qq.com)
|
||
|
|
|
||
|
|
package hls
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/q191201771/lal/pkg/base"
|
||
|
|
"github.com/q191201771/lal/pkg/mpegts"
|
||
|
|
"github.com/q191201771/naza/pkg/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
fh []byte
|
||
|
|
poped []base.RTMPMsg
|
||
|
|
)
|
||
|
|
|
||
|
|
type qo struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *qo) OnPATPMT(b []byte) {
|
||
|
|
fh = b
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *qo) OnPop(msg base.RTMPMsg) {
|
||
|
|
poped = append(poped, msg)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestQueue(t *testing.T) {
|
||
|
|
goldenRTMPMsg := []base.RTMPMsg{
|
||
|
|
{
|
||
|
|
Header: base.RTMPHeader{
|
||
|
|
MsgTypeID: base.RTMPTypeIDAudio,
|
||
|
|
},
|
||
|
|
Payload: []byte{0xAF},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Header: base.RTMPHeader{
|
||
|
|
MsgTypeID: base.RTMPTypeIDVideo,
|
||
|
|
},
|
||
|
|
Payload: []byte{0x17},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
q := &qo{}
|
||
|
|
queue := NewQueue(8, q)
|
||
|
|
for i := range goldenRTMPMsg {
|
||
|
|
queue.Push(goldenRTMPMsg[i])
|
||
|
|
}
|
||
|
|
assert.Equal(t, mpegts.FixedFragmentHeader, fh)
|
||
|
|
assert.Equal(t, goldenRTMPMsg, poped)
|
||
|
|
}
|