23 lines
574 B
Go
23 lines
574 B
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSegment(t *testing.T) {
|
|
segment := newSegment(10*time.Second, 0)
|
|
|
|
require.Equal(t, time.Duration(0), segment.Duration())
|
|
require.True(t, segment.CanWrite(9*time.Second))
|
|
require.True(t, segment.CanWrite(10*time.Second))
|
|
require.False(t, segment.CanWrite(11*time.Second))
|
|
|
|
d := segment.IncrementDuration(10 * time.Second)
|
|
require.Equal(t, segment.Duration(), d)
|
|
require.Equal(t, 10*time.Second, segment.Duration())
|
|
require.False(t, segment.CanWrite(1*time.Millisecond))
|
|
}
|