server_data_publish_handler_test.go

//
// Copyright (c) 2018- yutopp (yutopp@gmail.com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at  https://www.boost.org/LICENSE_1_0.txt)
//

package rtmp

import (
	"testing"

	"github.com/yutopp/go-rtmp/message"
)

func BenchmarkHandlePublisherVideoMessage(b *testing.B) {
	rwc := &rwcMock{}
	c := newConn(rwc, nil)

	s := newStream(42, c)
	s.handler.ChangeState(streamStateServerPublish)

	chunkStreamID := 0
	timestamp := uint32(0)
	msg := &message.VideoMessage{}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		_ = s.handle(chunkStreamID, timestamp, msg)
	}
}