package ast

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestArg2Map(t *testing.T) {
	defs := ArgumentDefinitionList{
		{
			Name:         "A",
			Type:         NamedType("String", nil),
			DefaultValue: &Value{Kind: StringValue, Raw: "defaultA"},
		},
		{Name: "B", Type: NamedType("String", nil)},
	}

	t.Run("defaults", func(t *testing.T) {
		args := arg2map(defs, ArgumentList{}, nil)
		require.Equal(t, "defaultA", args["A"])
		require.NotContains(t, args, "B")
	})

	t.Run("values", func(t *testing.T) {
		args := arg2map(defs, ArgumentList{
			{Name: "A", Value: &Value{Kind: StringValue, Raw: "valA"}},
			{Name: "B", Value: &Value{Kind: StringValue, Raw: "valB"}},
		}, nil)
		require.Equal(t, "valA", args["A"])
		require.Equal(t, "valB", args["B"])
	})

	t.Run("nulls", func(t *testing.T) {
		args := arg2map(defs, ArgumentList{
			{Name: "A", Value: &Value{Kind: NullValue}},
			{Name: "B", Value: &Value{Kind: NullValue}},
		}, nil)
		require.Nil(t, args["A"])
		require.Nil(t, args["B"])
		require.Contains(t, args, "A")
		require.Contains(t, args, "B")
	})

	t.Run("undefined variables", func(t *testing.T) {
		args := arg2map(defs, ArgumentList{
			{Name: "A", Value: &Value{Kind: Variable, Raw: "VarA"}},
			{Name: "B", Value: &Value{Kind: Variable, Raw: "VarB"}},
		}, map[string]any{})
		require.Equal(t, "defaultA", args["A"])
		require.NotContains(t, args, "B")
	})

	t.Run("nil variables", func(t *testing.T) {
		args := arg2map(defs, ArgumentList{
			{Name: "A", Value: &Value{Kind: Variable, Raw: "VarA"}},
			{Name: "B", Value: &Value{Kind: Variable, Raw: "VarB"}},
		}, map[string]any{
			"VarA": nil,
			"VarB": nil,
		})
		require.Nil(t, args["A"])
		require.Nil(t, args["B"])
		require.Contains(t, args, "A")
		require.Contains(t, args, "B")
	})

	t.Run("defined variables", func(t *testing.T) {
		args := arg2map(defs, ArgumentList{
			{Name: "A", Value: &Value{Kind: Variable, Raw: "VarA"}},
			{Name: "B", Value: &Value{Kind: Variable, Raw: "VarB"}},
		}, map[string]any{
			"VarA": "varvalA",
			"VarB": "varvalB",
		})
		require.Equal(t, "varvalA", args["A"])
		require.Equal(t, "varvalB", args["B"])
	})
}