github.com/Jeffail/benthos/v3@v3.65.0/lib/message/batch/docs_test.go (about)

     1  package batch_test
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/Jeffail/benthos/v3/internal/docs"
     7  	"github.com/Jeffail/benthos/v3/lib/message/batch"
     8  	"github.com/stretchr/testify/assert"
     9  	"github.com/stretchr/testify/require"
    10  	"gopkg.in/yaml.v3"
    11  )
    12  
    13  func TestBatchPolicySanit(t *testing.T) {
    14  	conf := batch.NewPolicyConfig()
    15  
    16  	var node yaml.Node
    17  	require.NoError(t, node.Encode(conf))
    18  	require.NoError(t, batch.FieldSpec().SanitiseYAML(&node, docs.SanitiseConfig{
    19  		RemoveTypeField: true,
    20  	}))
    21  
    22  	expSanit := `count: 0
    23  byte_size: 0
    24  period: ""
    25  check: ""
    26  processors: []
    27  `
    28  
    29  	b, err := yaml.Marshal(node)
    30  	require.NoError(t, err)
    31  	assert.Equal(t, expSanit, string(b))
    32  }