Function TestShowCreateSequence

dumpling/export/sql_test.go:368–406

(t *testing.T)

}

func TestShowCreateSequence(t *testing.T) {
	conf := defaultConfigForTest(t)
	db, mock, err := sqlmock.New()
	require.NoError(t, err)
	defer func() {
		require.NoError(t, db.Close())
	}()

	conn, err := db.Conn(context.Background())
	require.NoError(t, err)
	tctx := tcontext.Background().WithLogger(appLogger)
	baseConn := newBaseConn(conn, true, nil)

	conf.ServerInfo.ServerType = version.ServerTypeTiDB
	mock.ExpectQuery("SHOW CREATE SEQUENCE `test`.`s`").
		WillReturnRows(sqlmock.NewRows([]string{"Sequence", "Create Sequence"}).
			AddRow("s", "CREATE SEQUENCE `s` start with 1 minvalue 1 maxvalue 9223372036854775806 increment by 1 cache 1000 nocycle ENGINE=InnoDB"))
	mock.ExpectQuery("SHOW TABLE `test`.`s` NEXT_ROW_ID").
		WillReturnRows(sqlmock.NewRows([]string{"DB_NAME", "TABLE_NAME", "COLUMN_NAME", "NEXT_GLOBAL_ROW_ID", "ID_TYPE"}).
			AddRow("test", "s", nil, 1001, "SEQUENCE"))

	createSequenceSQL, err := ShowCreateSequence(tctx, baseConn, "test", "s", conf)
	require.NoError(t, err)
	require.Equal(t, "CREATE SEQUENCE `s` start with 1 minvalue 1 maxvalue 9223372036854775806 increment by 1 cache 1000 nocycle ENGINE=InnoDB;\nSELECT SETVAL(`s`,1001);\n", createSequenceSQL)
	require.NoError(t, mock.ExpectationsWereMet())

	conf.ServerInfo.ServerType = version.ServerTypeMariaDB
	mock.ExpectQuery("SHOW CREATE SEQUENCE `test`.`s`").
		WillReturnRows(sqlmock.NewRows([]string{"Table", "Create Table"}).
			AddRow("s", "CREATE SEQUENCE `s` start with 1 minvalue 1 maxvalue 9223372036854775806 increment by 1 cache 1000 nocycle ENGINE=InnoDB"))
	mock.ExpectQuery("SELECT NEXT_NOT_CACHED_VALUE FROM `test`.`s`").
		WillReturnRows(sqlmock.NewRows([]string{"next_not_cached_value"}).
			AddRow(1001))

	createSequenceSQL, err = ShowCreateSequence(tctx, baseConn, "test", "s", conf)
	require.NoError(t, err)
	require.Equal(t, "CREATE SEQUENCE `s` start with 1 minvalue 1 maxvalue 9223372036854775806 increment by 1 cache 1000 nocycle ENGINE=InnoDB;\nSELECT SETVAL(`s`,1001);\n", createSequenceSQL)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestShowCreatePolicy(t *testing.T) {
	db, mock, err := sqlmock.New()