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 }