kinesis-consumer/consumer_test.go
Harlow Ward c29698550f Add config options to Consumer
The Firehose service can take a max batch size of 500. While created the
example the need for finer grained configuration was necessary.
2016-02-09 22:31:15 -08:00

17 lines
293 B
Go

package connector
import (
"testing"
"github.com/bmizerany/assert"
)
func Test_Set(t *testing.T) {
defaultMaxBatchCount := 1000
assert.Equal(t, maxBatchCount, defaultMaxBatchCount)
c := NewConsumer("app", "stream")
c.Set("maxBatchCount", 100)
assert.Equal(t, maxBatchCount, 100)
}