github.com/Jeffail/benthos/v3@v3.65.0/internal/batch/count_test.go (about)

     1  package batch
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/Jeffail/benthos/v3/lib/message"
     7  	"github.com/stretchr/testify/assert"
     8  )
     9  
    10  func TestCount(t *testing.T) {
    11  	p1 := message.NewPart([]byte("foo bar"))
    12  
    13  	p2 := WithCollapsedCount(p1, 2)
    14  	p3 := WithCollapsedCount(p2, 3)
    15  	p4 := WithCollapsedCount(p1, 4)
    16  
    17  	assert.Equal(t, 1, CollapsedCount(p1))
    18  	assert.Equal(t, 2, CollapsedCount(p2))
    19  	assert.Equal(t, 4, CollapsedCount(p3))
    20  	assert.Equal(t, 4, CollapsedCount(p4))
    21  }
    22  
    23  func TestMessageCount(t *testing.T) {
    24  	m := message.New([][]byte{
    25  		[]byte("FOO"),
    26  		[]byte("BAR"),
    27  		[]byte("BAZ"),
    28  	})
    29  
    30  	assert.Equal(t, 3, MessageCollapsedCount(m))
    31  }