build_request_form_test.go

package bot

import (
	"bytes"
	"mime/multipart"
	"strings"
	"testing"

	"github.com/go-telegram/bot/models"
)

func assertFormData(t *testing.T, data, expect string) {
	if !strings.Contains(expect, "\r\n") {
		expect = strings.ReplaceAll(expect, "\n", "\r\n")
	}

	if data != expect {
		t.Fatalf("\nExpect:\n%s\nGot:\n%s\n", expect, data)
	}
}

func Test_buildRequestForm(t *testing.T) {
	params := struct {
		String                     string                     `json:"string"`
		InputFileUpload            *models.InputFileUpload    `json:"input_file_upload"`
		InputFileString            *models.InputFileString    `json:"input_file_string"`
		InputMediaSlice            []models.InputMedia        `json:"input_media_slice"`
		InlineQueryResultSlice     []models.InlineQueryResult `json:"inline_query_result_slice"`
		DefaultInt                 int                        `json:"default_int"`
		InputMediaInterface        models.InputMedia          `json:"input_media_interface"`
		InlineQueryResultInterface models.InlineQueryResult   `json:"inline_query_result_interface"`
		InputStickerSlice          []models.InputSticker      `json:"input_sticker_slice"`
		NoJSONTag1                 string
		NoJSONTag2                 string `json:"-"`
		OmitEmptyString            string `json:"omit_empty_string,omitempty"`
	}{
		String:          "foo",
		InputFileUpload: &models.InputFileUpload{Filename: "file.png", Data: strings.NewReader("content input file upload")},
		InputFileString: &models.InputFileString{Data: "content input file string"},
		InputMediaSlice: []models.InputMedia{
			&models.InputMediaPhoto{Media: "attach://foo.png", Caption: "bar", MediaAttachment: strings.NewReader("media attachment")},
			&models.InputMediaPhoto{Media: "foobar", Caption: "bar"},
		},
		InlineQueryResultSlice: []models.InlineQueryResult{
			&models.InlineQueryResultArticle{Title: "foo", Description: "bar", InputMessageContent: &models.InputTextMessageContent{MessageText: "foo"}},
		},
		DefaultInt:                 42,
		InputMediaInterface:        &models.InputMediaPhoto{Media: "foo", Caption: "bar", ParseMode: "baz"},
		InlineQueryResultInterface: &models.InlineQueryResultArticle{Title: "foo", Description: "bar", InputMessageContent: &models.InputTextMessageContent{MessageText: "foo"}},
		InputStickerSlice: []models.InputSticker{
			{
				Sticker:   "attach://sticker.png",
				Format:    "foo",
				EmojiList: []string{"bar"},
				StickerAttachment: strings.NewReader("sticker file"),
			},
			{
				Sticker:   "foo",
				Format:    "bar",
				EmojiList: []string{"baz"},
			},
		},
		NoJSONTag1:      "foo",
		NoJSONTag2:      "bar",
		OmitEmptyString: "",
	}

	buf := bytes.NewBuffer(nil)
	form := multipart.NewWriter(buf)
	form.SetBoundary("XXX") //nolint

	fieldsCount, errBuild := buildRequestForm(form, &params)
	if errBuild != nil {
		t.Error(errBuild)
		return
	}
	if err := form.Close(); err != nil {
		t.Errorf("failed to close form: %v", err)
	}

	expect := `--XXX
Content-Disposition: form-data; name="string"

foo
--XXX
Content-Disposition: form-data; name="input_file_upload"; filename="file.png"
Content-Type: application/octet-stream

content input file upload
--XXX
Content-Disposition: form-data; name="input_file_string"

content input file string
--XXX
Content-Disposition: form-data; name="foo.png"; filename="foo.png"
Content-Type: application/octet-stream

media attachment
--XXX
Content-Disposition: form-data; name="input_media_slice"

[{"type":"photo","media":"attach://foo.png","caption":"bar"},{"type":"photo","media":"foobar","caption":"bar"}]
--XXX
Content-Disposition: form-data; name="inline_query_result_slice"

[{"type":"article","id":"","title":"foo","input_message_content":{"message_text":"foo"},"description":"bar"}]
--XXX
Content-Disposition: form-data; name="default_int"

42
--XXX
Content-Disposition: form-data; name="input_media_interface"

{"type":"photo","media":"foo","caption":"bar","parse_mode":"baz"}
--XXX
Content-Disposition: form-data; name="inline_query_result_interface"

{"type":"article","id":"","title":"foo","input_message_content":{"message_text":"foo"},"description":"bar"}
--XXX
Content-Disposition: form-data; name="sticker.png"; filename="sticker.png"
Content-Type: application/octet-stream

sticker file
--XXX
Content-Disposition: form-data; name="input_sticker_slice"

[{"sticker":"attach://sticker.png","format":"foo","emoji_list":["bar"]},{"sticker":"foo","format":"bar","emoji_list":["baz"]}]
--XXX--
`
	assertEqualInt(t, fieldsCount, 7)
	assertFormData(t, buf.String(), expect)
}