github.com/argoproj/argo-events@v1.9.1/common/saramaconfig_test.go (about)

     1  package common
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  )
     8  
     9  func TestGetSaramaConfigFromYAMLString(t *testing.T) {
    10  	t.Run("YAML Config", func(t *testing.T) {
    11  		var yamlExample = string(`
    12  admin:
    13    retry:
    14      max: 105
    15  producer:
    16    maxMessageBytes: 800
    17  consumer:
    18    fetch: 
    19      min: 2
    20  net:
    21    MaxOpenRequests: 5
    22  `)
    23  		conf, err := GetSaramaConfigFromYAMLString(yamlExample)
    24  		assert.NoError(t, err)
    25  		assert.Equal(t, 800, conf.Producer.MaxMessageBytes)
    26  		assert.Equal(t, 105, conf.Admin.Retry.Max)
    27  		assert.Equal(t, int32(2), conf.Consumer.Fetch.Min)
    28  		assert.Equal(t, 5, conf.Net.MaxOpenRequests)
    29  	})
    30  	t.Run("Empty config", func(t *testing.T) {
    31  		conf, err := GetSaramaConfigFromYAMLString("")
    32  		assert.NoError(t, err)
    33  		assert.Equal(t, 1000000, conf.Producer.MaxMessageBytes)
    34  		assert.Equal(t, 5, conf.Admin.Retry.Max)
    35  		assert.Equal(t, int32(1), conf.Consumer.Fetch.Min)
    36  		assert.Equal(t, 5, conf.Net.MaxOpenRequests)
    37  	})
    38  
    39  	t.Run("NON yaml config", func(t *testing.T) {
    40  		_, err := GetSaramaConfigFromYAMLString("welcome")
    41  		assert.Error(t, err)
    42  	})
    43  }