4e909185d1
* Add GetRecords function to Kinesis Utils * Add sample .env to showcase ENV vars needed * Fix RedisCheckpoint to look for empty string (default from Redis) * Extract example code into its own repository
129 lines
2.5 KiB
Go
129 lines
2.5 KiB
Go
package buffers
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
func TestConsumeRecord(t *testing.T) {
|
|
var r1, s1 = []byte("Record1"), "Seq1"
|
|
var r2, s2 = []byte("Recrod2"), "Seq2"
|
|
|
|
b := MsgBuffer{}
|
|
b.ConsumeRecord(r1, s1)
|
|
|
|
if b.NumMessagesInBuffer() != 1 {
|
|
t.Errorf("NumRecordsInBuffer() want %v", 1)
|
|
}
|
|
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if b.NumMessagesInBuffer() != 2 {
|
|
t.Errorf("NumMessagesInBuffer() want %v", 2)
|
|
}
|
|
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if b.NumMessagesInBuffer() != 2 {
|
|
t.Errorf("NumMessagesInBuffer() want %v", 2)
|
|
}
|
|
}
|
|
|
|
func TestSequenceExists(t *testing.T) {
|
|
var r1, s1 = []byte("Record1"), "Seq1"
|
|
var r2, s2 = []byte("Recrod2"), "Seq2"
|
|
|
|
b := MsgBuffer{}
|
|
b.ConsumeRecord(r1, s1)
|
|
|
|
if b.SequenceExists(s1) != true {
|
|
t.Errorf("SequenceExists() want %v", true)
|
|
}
|
|
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if b.SequenceExists(s2) != true {
|
|
t.Errorf("SequenceExists() want %v", true)
|
|
}
|
|
}
|
|
|
|
func TestFlush(t *testing.T) {
|
|
var r1, s1 = []byte("Record"), "SeqNum"
|
|
b := MsgBuffer{}
|
|
b.ConsumeRecord(r1, s1)
|
|
|
|
b.Flush()
|
|
|
|
if b.NumMessagesInBuffer() != 0 {
|
|
t.Errorf("NumMessagesInBuffer() want %v", 0)
|
|
}
|
|
}
|
|
|
|
func TestLastSequenceNumber(t *testing.T) {
|
|
var r1, s1 = []byte("Record1"), "Seq1"
|
|
var r2, s2 = []byte("Recrod2"), "Seq2"
|
|
|
|
b := MsgBuffer{}
|
|
b.ConsumeRecord(r1, s1)
|
|
|
|
if b.LastSequenceNumber() != s1 {
|
|
t.Errorf("LastSequenceNumber() want %v", s1)
|
|
}
|
|
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if b.LastSequenceNumber() != s2 {
|
|
t.Errorf("LastSequenceNumber() want %v", s2)
|
|
}
|
|
}
|
|
|
|
func TestFirstSequenceNumber(t *testing.T) {
|
|
var r1, s1 = []byte("Record1"), "Seq1"
|
|
var r2, s2 = []byte("Recrod2"), "Seq2"
|
|
|
|
b := MsgBuffer{}
|
|
b.ConsumeRecord(r1, s1)
|
|
|
|
if b.FirstSequenceNumber() != s1 {
|
|
t.Errorf("FirstSequenceNumber() want %v", s1)
|
|
}
|
|
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if b.FirstSequenceNumber() != s1 {
|
|
t.Errorf("FirstSequenceNumber() want %v", s1)
|
|
}
|
|
}
|
|
|
|
func TestShouldFlush(t *testing.T) {
|
|
const n = 2
|
|
var r1, s1 = []byte("Record1"), "Seq1"
|
|
var r2, s2 = []byte("Recrod2"), "Seq2"
|
|
|
|
b := MsgBuffer{NumMessagesToBuffer: n}
|
|
b.ConsumeRecord(r1, s1)
|
|
|
|
if b.ShouldFlush() != false {
|
|
t.Errorf("ShouldFlush() want %v", false)
|
|
}
|
|
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if b.ShouldFlush() != true {
|
|
t.Errorf("ShouldFlush() want %v", true)
|
|
}
|
|
}
|
|
|
|
func TestData(t *testing.T) {
|
|
var r1, s1 = []byte("Record1\n"), "Seq1"
|
|
var r2, s2 = []byte("Record2\n"), "Seq2"
|
|
var body = []byte("Record1\nRecord2\n")
|
|
|
|
b := MsgBuffer{}
|
|
b.ConsumeRecord(r1, s1)
|
|
b.ConsumeRecord(r2, s2)
|
|
|
|
if !bytes.Equal(b.Data(), body) {
|
|
t.Errorf("Data() want %v", body)
|
|
}
|
|
}
|