segmento/internal/segment/segment_test.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))
}