package broadcast

import (
	"context"
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"git.sr.ht/~blallo/logz/interface"
	"git.sr.ht/~blallo/logz/testlogger"
)

func TestLineRing(t *testing.T) {
	assert := assert.New(t)
	require := require.New(t)

	ctx := context.TODO()
	lines, err := NewLineRing(3)
	require.NoError(err)

	logLevel := logz.LogInfo
	if os.Getenv("DEBUG") != "" {
		logLevel = logz.LogDebug
	}

	logger := testlogger.New(t).SetLevel(logLevel)
	lines.WithLogger(logger)

	go lines.Run(ctx)
	defer lines.Close()

	testLines := []string{
		"test line 1",
		"test line 2",
		"test line 3",
		"test line 4",
		"test line 5",
	}
	pIn := lines.File()

	for _, line := range testLines[:3] {
		pIn.Write([]byte(line))
		pIn.Write([]byte("\n"))
	}

	assert.Eventually(
		func() bool {
			return assert.Equal(testLines[:3], lines.Lines())
		}, 100*time.Millisecond, time.Millisecond,
	)

	for _, line := range testLines[3:] {
		pIn.Write([]byte(line))
		pIn.Write([]byte("\n"))
	}

	assert.Eventually(
		func() bool {
			return assert.Equal(append(testLines[2:3], testLines[3:]...), lines.Lines())
		}, 100*time.Millisecond, time.Millisecond,
	)
}