github.com/instill-ai/component@v0.16.0-beta/pkg/base/operator_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/operatorDef.json
    13  	operatorDefJSON []byte
    14  	//go:embed testdata/operatorTasks.json
    15  	operatorTasksJSON []byte
    16  	//go:embed testdata/wantOperatorDefinition.json
    17  	wantOperatorDefinitionJSON []byte
    18  )
    19  
    20  func TestOperator_ListOperatorDefinitions(t *testing.T) {
    21  	c := qt.New(t)
    22  	logger := zap.NewNop()
    23  
    24  	op := BaseOperator{
    25  		Logger: logger,
    26  	}
    27  
    28  	err := op.LoadOperatorDefinition(operatorDefJSON, operatorTasksJSON, nil)
    29  	c.Assert(err, qt.IsNil)
    30  
    31  	got, err := op.GetOperatorDefinition(nil, nil)
    32  	c.Assert(err, qt.IsNil)
    33  	c.Check(wantOperatorDefinitionJSON, qt.JSONEquals, got)
    34  }