mirror of
https://github.com/livekit/livekit.git
synced 2026-04-04 06:05:45 +00:00
37 lines
832 B
Go
37 lines
832 B
Go
package buffer
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDataStats(t *testing.T) {
|
|
stats := NewDataStats(DataStatsParam{WindowDuration: time.Second})
|
|
|
|
time.Sleep(time.Millisecond)
|
|
r := stats.ToProtoAggregateOnly()
|
|
require.Equal(t, r.StartTime.AsTime().UnixNano(), stats.startTime.UnixNano())
|
|
require.NotZero(t, r.EndTime)
|
|
require.NotZero(t, r.Duration)
|
|
r.StartTime = nil
|
|
r.EndTime = nil
|
|
r.Duration = 0
|
|
require.Zero(t, *r)
|
|
|
|
stats.Update(100, time.Now().UnixNano())
|
|
r = stats.ToProtoActive()
|
|
require.EqualValues(t, 100, r.Bytes)
|
|
require.NotZero(t, r.Bitrate)
|
|
|
|
// wait for window duration
|
|
time.Sleep(time.Second)
|
|
r = stats.ToProtoActive()
|
|
require.Zero(t, *r)
|
|
stats.Stop()
|
|
r = stats.ToProtoAggregateOnly()
|
|
require.EqualValues(t, 100, r.Bytes)
|
|
require.NotZero(t, r.Bitrate)
|
|
}
|