package function

import (
	"testing"

	"github.com/src-d/go-mysql-server/sql"
	"github.com/src-d/go-mysql-server/sql/expression"
	"github.com/stretchr/testify/require"
)

func TestConcatWithSeparator(t *testing.T) {
	t.Run("multiple arguments", func(t *testing.T) {
		require := require.New(t)
		f, err := NewConcatWithSeparator(
			expression.NewLiteral(",", sql.Text),
			expression.NewLiteral("foo", sql.Text),
			expression.NewLiteral(5, sql.Text),
			expression.NewLiteral(true, sql.Boolean),
		)
		require.NoError(err)

		v, err := f.Eval(sql.NewEmptyContext(), nil)
		require.NoError(err)
		require.Equal("foo,5,true", v)
	})

	t.Run("some argument is empty", func(t *testing.T) {
		require := require.New(t)
		f, err := NewConcatWithSeparator(
			expression.NewLiteral(",", sql.Text),
			expression.NewLiteral("foo", sql.Text),
			expression.NewLiteral("", sql.Text),
			expression.NewLiteral(true, sql.Boolean),
		)
		require.NoError(err)

		v, err := f.Eval(sql.NewEmptyContext(), nil)
		require.NoError(err)
		require.Equal("foo,,true", v)
	})

	t.Run("some argument is nil", func(t *testing.T) {
		require := require.New(t)
		f, err := NewConcatWithSeparator(
			expression.NewLiteral(",", sql.Text),
			expression.NewLiteral("foo", sql.Text),
			expression.NewLiteral(nil, sql.Text),
			expression.NewLiteral(true, sql.Boolean),
		)
		require.NoError(err)

		v, err := f.Eval(sql.NewEmptyContext(), nil)
		require.NoError(err)
		require.Equal("foo,true", v)
	})

	t.Run("separator is nil", func(t *testing.T) {
		require := require.New(t)
		f, err := NewConcatWithSeparator(
			expression.NewLiteral(nil, sql.Text),
			expression.NewLiteral("foo", sql.Text),
			expression.NewLiteral(5, sql.Text),
			expression.NewLiteral(true, sql.Boolean),
		)
		require.NoError(err)

		v, err := f.Eval(sql.NewEmptyContext(), nil)
		require.NoError(err)
		require.Equal(nil, v)
	})

	t.Run("concat_ws array", func(t *testing.T) {
		require := require.New(t)
		f, err := NewConcatWithSeparator(
			expression.NewLiteral([]interface{}{",", 5, "bar", true}, sql.Array(sql.Text)),
		)
		require.NoError(err)

		v, err := f.Eval(sql.NewEmptyContext(), nil)
		require.NoError(err)
		require.Equal("5,bar,true", v)
	})
}

func TestNewConcatWithSeparator(t *testing.T) {
	require := require.New(t)

	_, err := NewConcatWithSeparator(expression.NewLiteral(nil, sql.Array(sql.Text)))
	require.NoError(err)

	_, err = NewConcatWithSeparator(expression.NewLiteral(nil, sql.Array(sql.Text)), expression.NewLiteral(nil, sql.Int64))
	require.Error(err)
	require.True(ErrConcatArrayWithOthers.Is(err))

	_, err = NewConcatWithSeparator(expression.NewLiteral(nil, sql.Tuple(sql.Text, sql.Text)))
	require.Error(err)
	require.True(sql.ErrInvalidType.Is(err))

	_, err = NewConcatWithSeparator(
		expression.NewLiteral(nil, sql.Text),
		expression.NewLiteral(nil, sql.Boolean),
		expression.NewLiteral(nil, sql.Int64),
		expression.NewLiteral(nil, sql.Text),
	)
	require.NoError(err)
}