Skip to content

Commit

Permalink
add ut
Browse files Browse the repository at this point in the history
  • Loading branch information
sdojjy committed Apr 23, 2024
1 parent 0576660 commit 65e8415
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions pkg/sink/codec/open/open_protocol_encoder_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -225,6 +225,7 @@ func TestEncodeDecodeE2E(t *testing.T) {
topic := "test"

codecConfig := common.NewConfig(config.ProtocolOpen)
codecConfig.OpenOutputOldValue = false
builder, err := NewBatchEncoderBuilder(ctx, codecConfig)
require.NoError(t, err)
encoder := builder.Build()
Expand Down Expand Up @@ -422,3 +423,41 @@ func TestE2EClaimCheckMessage(t *testing.T) {
require.Equal(t, column.Value, decodedColumn.Value, colName)
}
}

func TestOutputOldValueFalse(t *testing.T) {
helper := entry.NewSchemaTestHelper(t)
defer helper.Close()

_ = helper.DDL2Event(`create table test.t(a varchar(10) primary key, b varchar(10))`)
event := helper.DML2Event(`insert into test.t values ("aa", "bb")`, "test", "t")
event.PreColumns = event.Columns

ctx := context.Background()
topic := "test"

codecConfig := common.NewConfig(config.ProtocolOpen)
codecConfig.OpenOutputOldValue = false
builder, err := NewBatchEncoderBuilder(ctx, codecConfig)
require.NoError(t, err)
encoder := builder.Build()

err = encoder.AppendRowChangedEvent(ctx, topic, event, func() {})
require.NoError(t, err)

message := encoder.Build()[0]

decoder, err := NewBatchDecoder(ctx, codecConfig, nil)
require.NoError(t, err)

err = decoder.AddKeyValue(message.Key, message.Value)
require.NoError(t, err)

messageType, hasNext, err := decoder.HasNext()
require.NoError(t, err)
require.True(t, hasNext)
require.Equal(t, messageType, model.MessageTypeRow)

decoded, err := decoder.NextRowChangedEvent()
require.NoError(t, err)
require.Nil(t, decoded.PreColumns)
}

0 comments on commit 65e8415

Please sign in to comment.