lal/pkg/innertest/dump_ps_test.go

// Copyright 2022, 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 innertest

import (
	"net"
	"os"
	"testing"

	"github.com/q191201771/lal/pkg/base"
	"github.com/q191201771/naza/pkg/bele"
	"github.com/q191201771/naza/pkg/nazalog"
	"github.com/q191201771/naza/pkg/nazanet"
)

// TestRe_PsPubSession
//
// 重放业务方的ps流。
//
// 本测试函数模拟客户端，读取业务方对的dumpfile，重新推送给lalserver。
//
// 步骤：
//
// 1. 让业务方提供lalserver录制下来的dumpfile文件
// 2. 将dumpfile存放在下面filename变量处，或者修改下面filename变量值
// 3. 启动lalserver
// 4. 调用HTTP API
// curl -H "Content-Type:application/json" -X POST -d '{"stream_name": "test110", "port": 10002, "timeout_ms": 10000}' http://127.0.0.1:8083/api/ctrl/start_rtp_pub
// 5. 执行该测试
// go test -test.run TestDump_PsPub
func TestDump_PsPub(t *testing.T) {
	filename := "/tmp/record.psdata"
	isTcpFlag := 0

	b, err := os.ReadFile(filename)
	if len(b) == 0 || err != nil {
		return
	}

	testPushFile("127.0.0.1:10002", filename, isTcpFlag)
}

// ---------------------------------------------------------------------------------------------------------------------

// testPushFile 创建udp客户端，向 addr 地址发送 filename 文件中的包
func testPushFile(addr string, filename string, isTcpFlag int) {
	var udpConn *nazanet.UdpConnection
	var tcpConn net.Conn
	var err error

	if isTcpFlag != 0 {
		tcpConn, err = net.Dial("tcp", addr)
		nazalog.Assert(nil, err)
	} else {
		udpConn, err = nazanet.NewUdpConnection(func(option *nazanet.UdpConnectionOption) {
			option.RAddr = addr
		})
		nazalog.Assert(nil, err)
	}

	df := base.NewDumpFile()
	err = df.OpenToRead(filename)
	nazalog.Assert(nil, err)

	lb := make([]byte, 2)
	for {
		m, err := df.ReadOneMessage()
		if err != nil {
			nazalog.Errorf("%+v", err)
			break
		}
		nazalog.Debugf("%s", m.DebugString())
		if m.Typ == base.DumpTypeInnerFileHeaderData {
			continue
		}

		if isTcpFlag != 0 {
			bele.BePutUint16(lb, uint16(m.Len))
			_, err = tcpConn.Write(lb)
			nazalog.Assert(nil, err)
			_, err = tcpConn.Write(m.Body)
			nazalog.Assert(nil, err)
		} else {
			udpConn.Write(m.Body)
		}

		//time.Sleep(10 * time.Millisecond)
	}
}