25 lines
620 B
Go
25 lines
620 B
Go
package segment_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"segmento/internal/segment"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSegment(t *testing.T) {
|
|
segment := 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))
|
|
}
|