streamsql/window/counting_window_test.go

package window

import (
	"context"
	"testing"
	"time"

	"github.com/rulego/streamsql/model"
	"github.com/stretchr/testify/require"

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

func TestCountingWindow(t *testing.T) {
	_, cancel := context.WithCancel(context.Background())
	defer cancel()

	// Test case 1: Normal operation
	cw, _ := NewCountingWindow(model.WindowConfig{
		Params: map[string]interface{}{
			"count": 3,
			"callback": func(results []interface{}) {
				t.Logf("Received results: %v", results)
			},
		},
	})
	go cw.Start()

	for i := 0; i < 3; i++ {
		cw.Add(i)
		time.Sleep(100 * time.Millisecond)
	}

	// Trigger one more element to check threshold
	cw.Add(3)

	results := make(chan []model.Row)
	go func() {
		for res := range cw.OutputChan() {
			results <- res
		}
	}()

	select {
	case res := <-results:
		assert.Len(t, res, 3)
		raw := make([]interface{}, len(res))
		for _, row := range res {
			raw = append(raw, row.Data)
		}
		assert.Contains(t, raw, 0)
		assert.Contains(t, raw, 1)
		assert.Contains(t, raw, 2)
	case <-time.After(2 * time.Second):
		t.Error("No results received within timeout")
	}

	// Test case 2: Reset
	cw.Reset()
	assert.Len(t, cw.dataBuffer, 0)
}

func TestCountingWindowBadThreshold(t *testing.T) {
	_, err := CreateWindow(model.WindowConfig{
		Type: "counting",
		Params: map[string]interface{}{
			"count": 0,
		},
	})
	require.Error(t, err)
}