// Copyright 2021 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package v1

import (
	"testing"

	timodel "github.com/pingcap/tidb/pkg/meta/model"
	"github.com/pingcap/tidb/pkg/parser/charset"
	"github.com/pingcap/tidb/pkg/parser/mysql"
	"github.com/pingcap/tiflow/cdc/model"
	"github.com/stretchr/testify/require"
)

func TestRowRedoConvert(t *testing.T) {
	t.Parallel()

	row := &RowChangedEvent{
		StartTs:  100,
		CommitTs: 120,
		Table:    &TableName{Schema: "test", Table: "table1", TableID: 57},
		PreColumns: []*Column{{
			Name:  "a1",
			Type:  mysql.TypeLong,
			Flag:  model.BinaryFlag | model.MultipleKeyFlag | model.HandleKeyFlag,
			Value: int64(1),
		}, {
			Name:  "a2",
			Type:  mysql.TypeVarchar,
			Value: []byte("char"),
		}, {
			Name:  "a3",
			Type:  mysql.TypeLong,
			Flag:  model.BinaryFlag | model.MultipleKeyFlag | model.HandleKeyFlag,
			Value: int64(1),
		}, {
			Name:    "a4",
			Type:    mysql.TypeTinyBlob,
			Charset: charset.CharsetGBK,
			Value:   []byte("你好"),
		}, nil},
		Columns: []*Column{{
			Name:  "a1",
			Type:  mysql.TypeLong,
			Flag:  model.BinaryFlag | model.MultipleKeyFlag | model.HandleKeyFlag,
			Value: int64(2),
		}, {
			Name:  "a2",
			Type:  mysql.TypeVarchar,
			Value: []byte("char-updated"),
		}, {
			Name:  "a3",
			Type:  mysql.TypeLong,
			Flag:  model.BinaryFlag | model.MultipleKeyFlag | model.HandleKeyFlag,
			Value: int64(2),
		}, {
			Name:    "a4",
			Type:    mysql.TypeTinyBlob,
			Charset: charset.CharsetGBK,
			Value:   []byte("世界"),
		}, nil},
		IndexColumns: [][]int{{1, 3}},
	}

	redoLog := &RedoLog{RedoRow: &RedoRowChangedEvent{Row: row}}
	PreMarshal(redoLog)
	require.Equal(t, 5, len(redoLog.RedoRow.PreColumns))
	require.Equal(t, 5, len(redoLog.RedoRow.Columns))
	data, err := redoLog.MarshalMsg(nil)
	require.Nil(t, err)

	redoLog2 := &RedoLog{}
	_, err = redoLog2.UnmarshalMsg(data)
	require.Nil(t, err)
	PostUnmarshal(redoLog2)
	require.Equal(t, row, redoLog2.RedoRow.Row)
}

func TestRowRedoConvertWithEmptySlice(t *testing.T) {
	t.Parallel()

	row := &RowChangedEvent{
		StartTs:  100,
		CommitTs: 120,
		Table:    &TableName{Schema: "test", Table: "table1", TableID: 57},
		PreColumns: []*Column{{
			Name:  "a1",
			Type:  mysql.TypeLong,
			Flag:  model.BinaryFlag | model.MultipleKeyFlag | model.HandleKeyFlag,
			Value: int64(1),
		}, {
			Name:  "a2",
			Type:  mysql.TypeVarchar,
			Value: []byte(""), // empty slice should be marshal and unmarshal safely
		}},
		Columns: []*Column{{
			Name:  "a1",
			Type:  mysql.TypeLong,
			Flag:  model.BinaryFlag | model.MultipleKeyFlag | model.HandleKeyFlag,
			Value: int64(2),
		}, {
			Name:  "a2",
			Type:  mysql.TypeVarchar,
			Value: []byte(""),
		}},
		IndexColumns: [][]int{{1}},
	}

	redoLog := &RedoLog{RedoRow: &RedoRowChangedEvent{Row: row}}
	PreMarshal(redoLog)
	data, err := redoLog.MarshalMsg(nil)
	require.Nil(t, err)

	redoLog2 := &RedoLog{}
	_, err = redoLog2.UnmarshalMsg(data)
	require.Nil(t, err)
	PostUnmarshal(redoLog2)
	require.Equal(t, row, redoLog2.RedoRow.Row)
}

func TestDDLRedoConvert(t *testing.T) {
	t.Parallel()

	ddl := &DDLEvent{
		StartTs:  1020,
		CommitTs: 1030,
		Type:     timodel.ActionAddColumn,
		Query:    "ALTER TABLE test.t1 ADD COLUMN a int",
	}

	redoLog := &RedoLog{RedoDDL: &RedoDDLEvent{DDL: ddl}}
	PreMarshal(redoLog)
	data, err := redoLog.MarshalMsg(nil)
	require.Nil(t, err)

	redoLog2 := &RedoLog{}
	_, err = redoLog2.UnmarshalMsg(data)
	require.Nil(t, err)
	PostUnmarshal(redoLog2)
	require.Equal(t, ddl, redoLog2.RedoDDL.DDL)
}