mg-transport-api-client-go/v1/template_test.go

package v1

import (
	"encoding/json"
	"testing"

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

func TestTemplateItem_MarshalJSON(t *testing.T) {
	text := TemplateItem{
		Type: TemplateItemTypeText,
		Text: "text item",
	}

	variable := TemplateItem{
		Type:    TemplateItemTypeVar,
		VarType: TemplateVarFirstName,
	}

	emptyVariable := TemplateItem{
		Type:    TemplateItemTypeVar,
		VarType: "",
	}

	data, err := json.Marshal(text)
	assert.NoError(t, err)
	assert.Equal(t, "\""+text.Text+"\"", string(data))

	data, err = json.Marshal(variable)
	assert.NoError(t, err)
	assert.Equal(t, `{"var":"first_name"}`, string(data))

	data, err = json.Marshal(emptyVariable)
	assert.NoError(t, err)
	assert.Equal(t, "{}", string(data))
}

func TestTemplateItem_UnmarshalJSON(t *testing.T) {
	var (
		textResult          TemplateItem
		variableResult      TemplateItem
		emptyVariableResult TemplateItem
	)

	text := []byte("\"text block\"")
	variable := []byte(`{"var":"first_name"}`)
	emptyVariable := []byte("{}")

	require.NoError(t, json.Unmarshal(text, &textResult))
	require.NoError(t, json.Unmarshal(variable, &variableResult))
	require.NoError(t, json.Unmarshal(emptyVariable, &emptyVariableResult))

	assert.Equal(t, TemplateItemTypeText, textResult.Type)
	assert.Equal(t, string(text)[1:11], textResult.Text)

	assert.Equal(t, TemplateItemTypeVar, variableResult.Type)
	assert.Equal(t, TemplateVarFirstName, variableResult.VarType)
	assert.Empty(t, variableResult.Text)

	assert.Equal(t, TemplateItemTypeVar, emptyVariableResult.Type)
	assert.Equal(t, TemplateVarCustom, emptyVariableResult.VarType)
	assert.Empty(t, emptyVariableResult.Text)
}

func TestUnmarshalMediaInteractiveTemplate(t *testing.T) {
	var template Template
	input := `{
	"code":"aaa#bbb#ru",
    "phone": "79252223456",
    "channel_id": 1,
    "headerParams": {
        "textVars": [
            "Johny",
            "1234C"
        ],
        "imageUrl": "http://example.com/intaro/d2354125",
        "videoUrl": "http://example.com/intaro/d2222",
        "documentUrl": "http://example.com/intaro/d4444"
    },
    "footer": "Scooter",
    "buttonParams": [
        {
            "type": "URL",
            "urlParameter": "222ddd"
        },
        {
            "type": "QUICK_REPLY",
            "text": "Yes"
        }
    ]
}`
	assert.NoError(t, json.Unmarshal([]byte(input), &template))

	assert.Equal(t, "aaa#bbb#ru", template.Code)
	assert.Equal(t, []string{"Johny", "1234C"}, template.HeaderParams.TextVars)
	assert.Equal(t, "http://example.com/intaro/d2354125", template.HeaderParams.ImageURL)
	assert.Equal(t, "http://example.com/intaro/d2222", template.HeaderParams.VideoURL)
	assert.Equal(t, "http://example.com/intaro/d4444", template.HeaderParams.DocumentURL)
	assert.Equal(t, "Scooter", *template.Footer)
	assert.Equal(t, URLButton, template.ButtonParams[0].ButtonType)
	assert.Equal(t, "222ddd", template.ButtonParams[0].URLParameter)
	assert.Equal(t, QuickReplyButton, template.ButtonParams[1].ButtonType)
	assert.Equal(t, "Yes", template.ButtonParams[1].Text)

	input = `{"footer": "Scooter"}`
	template = Template{}
	assert.NoError(t, json.Unmarshal([]byte(input), &template))
	assert.Nil(t, template.HeaderParams)
	assert.Empty(t, template.ButtonParams)
}