github.com/instill-ai/component@v0.16.0-beta/pkg/base/connector_test.go (about)

     1  package base
     2  
     3  import (
     4  	_ "embed"
     5  	"testing"
     6  
     7  	qt "github.com/frankban/quicktest"
     8  	"go.uber.org/zap"
     9  )
    10  
    11  var (
    12  	//go:embed testdata/connectorDef.json
    13  	connectorDefJSON []byte
    14  	//go:embed testdata/connectorTasks.json
    15  	connectorTasksJSON []byte
    16  	//go:embed testdata/connectorAdditional.json
    17  	connectorAdditionalJSON []byte
    18  	//go:embed testdata/wantConnectorDefinition.json
    19  	wantConnectorDefinitionJSON []byte
    20  )
    21  
    22  func TestConnector_ListConnectorDefinitions(t *testing.T) {
    23  	c := qt.New(t)
    24  	logger := zap.NewNop()
    25  
    26  	conn := BaseConnector{
    27  		Logger: logger,
    28  	}
    29  
    30  	err := conn.LoadConnectorDefinition(
    31  		connectorDefJSON,
    32  		connectorTasksJSON,
    33  		map[string][]byte{"additional.json": connectorAdditionalJSON})
    34  	c.Assert(err, qt.IsNil)
    35  
    36  	got, err := conn.GetConnectorDefinition(nil, nil)
    37  	c.Assert(err, qt.IsNil)
    38  	c.Check(wantConnectorDefinitionJSON, qt.JSONEquals, got)
    39  }