github.com/argoproj/argo-events@v1.9.1/eventbus/jetstream/eventsource/source_jetstream_test.go (about)

     1  package eventsource
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/argoproj/argo-events/eventbus/common"
     7  	"github.com/argoproj/argo-events/pkg/apis/eventbus/v1alpha1"
     8  	"github.com/stretchr/testify/assert"
     9  	"go.uber.org/zap"
    10  )
    11  
    12  const (
    13  	testURL          = "test-url"
    14  	testEventSource  = "test-event-source-name"
    15  	testStreamConfig = "test-stream-config"
    16  )
    17  
    18  func TestNewSourceJetstream(t *testing.T) {
    19  	logger := zap.NewExample().Sugar()
    20  
    21  	auth := &common.Auth{}
    22  	sourceJetstream, err := NewSourceJetstream(testURL, testEventSource, testStreamConfig, auth, logger)
    23  	assert.NotNil(t, sourceJetstream)
    24  	assert.Nil(t, err)
    25  }
    26  
    27  func TestSourceJetstream_Connect(t *testing.T) {
    28  	logger := zap.NewExample().Sugar()
    29  
    30  	auth := &common.Auth{}
    31  	sourceJetstream, err := NewSourceJetstream(testURL, testEventSource, testStreamConfig, auth, logger)
    32  	assert.NotNil(t, sourceJetstream)
    33  	assert.Nil(t, err)
    34  
    35  	conn, err := sourceJetstream.Connect("test-client-id")
    36  	assert.Nil(t, conn)
    37  	assert.NotNil(t, err)
    38  }
    39  
    40  func TestSourceJetstream_Initialize_Failure(t *testing.T) {
    41  	logger := zap.NewExample().Sugar()
    42  
    43  	auth := &common.Auth{
    44  		Strategy: v1alpha1.AuthStrategyNone,
    45  	}
    46  	sourceJetstream, err := NewSourceJetstream(testURL, testEventSource, testStreamConfig, auth, logger)
    47  	assert.NotNil(t, sourceJetstream)
    48  	assert.Nil(t, err)
    49  
    50  	err = sourceJetstream.Initialize()
    51  	assert.NotNil(t, err)
    52  }