package connector

import (
	"testing"

	"github.com/aws/aws-sdk-go/service/kinesis"
	"github.com/bmizerany/assert"
)

func BenchmarkBufferLifecycle(b *testing.B) {
	buf := Buffer{MaxBatchCount: 1000}
	seq := "1"
	rec := &kinesis.Record{SequenceNumber: &seq}

	for i := 0; i < b.N; i++ {
		buf.AddRecord(rec)

		if buf.ShouldFlush() {
			buf.Flush()
		}
	}
}

func Test_FirstSeq(t *testing.T) {
	b := Buffer{}
	s1, s2 := "1", "2"
	r1 := &kinesis.Record{SequenceNumber: &s1}
	r2 := &kinesis.Record{SequenceNumber: &s2}

	b.AddRecord(r1)
	assert.Equal(t, b.FirstSeq(), "1")

	b.AddRecord(r2)
	assert.Equal(t, b.FirstSeq(), "1")
}

func Test_LastSeq(t *testing.T) {
	b := Buffer{}
	s1, s2 := "1", "2"
	r1 := &kinesis.Record{SequenceNumber: &s1}
	r2 := &kinesis.Record{SequenceNumber: &s2}

	b.AddRecord(r1)
	assert.Equal(t, b.LastSeq(), "1")

	b.AddRecord(r2)
	assert.Equal(t, b.LastSeq(), "2")
}

func Test_ShouldFlush(t *testing.T) {
	b := Buffer{MaxBatchCount: 2}
	s1, s2 := "1", "2"
	r1 := &kinesis.Record{SequenceNumber: &s1}
	r2 := &kinesis.Record{SequenceNumber: &s2}

	b.AddRecord(r1)
	assert.Equal(t, b.ShouldFlush(), false)

	b.AddRecord(r2)
	assert.Equal(t, b.ShouldFlush(), true)
}