Function TestBuildDMLDeleteQuery

go/sql/builder_test.go:740–805

(t *testing.T)

}

func TestBuildDMLDeleteQuery(t *testing.T) {
	databaseName := "mydb"
	tableName := "tbl"
	tableColumns := NewColumnList([]string{"id", "name", "rank", "position", "age"})
	args := []interface{}{3, "testname", "first", 17, 23}
	{
		uniqueKeyColumns := NewColumnList([]string{"position"})
		builder, err := NewDMLDeleteQueryBuilder(databaseName, tableName, tableColumns, uniqueKeyColumns)
		require.NoError(t, err)

		query, uniqueKeyArgs, err := builder.BuildQuery(args)
		require.NoError(t, err)
		expected := `
			delete /* gh-ost mydb.tbl */
				from
					mydb.tbl
				where
					((position = ?))
		`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{17}, uniqueKeyArgs)
	}
	{
		uniqueKeyColumns := NewColumnList([]string{"name", "position"})
		builder, err := NewDMLDeleteQueryBuilder(databaseName, tableName, tableColumns, uniqueKeyColumns)
		require.NoError(t, err)

		query, uniqueKeyArgs, err := builder.BuildQuery(args)
		require.NoError(t, err)
		expected := `
			delete /* gh-ost mydb.tbl */
				from
					mydb.tbl
				where
					((name = ?) and (position = ?))
		`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{"testname", 17}, uniqueKeyArgs)
	}
	{
		uniqueKeyColumns := NewColumnList([]string{"position", "name"})
		builder, err := NewDMLDeleteQueryBuilder(databaseName, tableName, tableColumns, uniqueKeyColumns)
		require.NoError(t, err)

		query, uniqueKeyArgs, err := builder.BuildQuery(args)
		require.NoError(t, err)
		expected := `
			delete /* gh-ost mydb.tbl */
				from
					mydb.tbl
				where
					((position = ?) and (name = ?))
		`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{17, "testname"}, uniqueKeyArgs)
	}
	{
		uniqueKeyColumns := NewColumnList([]string{"position", "name"})