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

package v1

import (
	"bytes"
	"encoding/base64"
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
	"testing"
	"time"

	"github.com/stretchr/testify/suite"
	"gopkg.in/h2non/gock.v1"
)

type MGClientTest struct {
	suite.Suite
}

func TestMGClient(t *testing.T) {
	suite.Run(t, new(MGClientTest))
}

func (t *MGClientTest) client() *MgClient {
	c := New("https://mg-test.retailcrm.pro", "mg_token")
	c.Debug = true

	return c
}

func (t *MGClientTest) gock() *gock.Request {
	return gock.New("https://mg-test.retailcrm.pro").MatchHeader("x-transport-token", "mg_token")
}

func (t *MGClientTest) transportURL(path string) string {
	return "/api/transport/v1/" + strings.TrimLeft(path, "/")
}

func (t *MGClientTest) Test_TransportChannels() {
	c := t.client()
	chName := "WhatsApp Channel"
	createdAt := "2021-11-22T08:20:46.479979Z"

	defer gock.Off()
	t.gock().
		Get(t.transportURL("channels")).
		Reply(http.StatusOK).
		JSON(
			[]ChannelListItem{
				{
					ID:         1,
					ExternalID: "external_id",
					Type:       "whatsapp",
					Name:       &chName,
					Settings: ChannelSettings{
						Status: Status{
							Delivered: ChannelFeatureNone,
							Read:      ChannelFeatureSend,
						},
						Text: ChannelSettingsText{
							Creating:      ChannelFeatureBoth,
							Editing:       ChannelFeatureBoth,
							Quoting:       ChannelFeatureBoth,
							Deleting:      ChannelFeatureReceive,
							MaxCharsCount: 4096,
						},
						Product: Product{
							Creating: ChannelFeatureReceive,
							Editing:  ChannelFeatureReceive,
						},
						Order: Order{
							Creating: ChannelFeatureReceive,
							Editing:  ChannelFeatureReceive,
						},
						File: ChannelSettingsFilesBase{
							Creating: ChannelFeatureBoth,
							Editing:  ChannelFeatureBoth,
							Quoting:  ChannelFeatureBoth,
							Deleting: ChannelFeatureReceive,
							Max:      1,
						},
						Image: ChannelSettingsFilesBase{
							Creating: ChannelFeatureBoth,
							Editing:  ChannelFeatureBoth,
							Quoting:  ChannelFeatureBoth,
							Deleting: ChannelFeatureReceive,
							Max:      1, // nolint:gomnd
						},
						Suggestions: ChannelSettingsSuggestions{
							Text:  ChannelFeatureBoth,
							Phone: ChannelFeatureBoth,
							Email: ChannelFeatureBoth,
						},
						CustomerExternalID: ChannelFeatureCustomerExternalIDPhone,
						SendingPolicy: SendingPolicy{
							NewCustomer: ChannelFeatureSendingPolicyTemplate,
						},
					},
					CreatedAt:     createdAt,
					UpdatedAt:     &createdAt,
					ActivatedAt:   createdAt,
					DeactivatedAt: nil,
					IsActive:      true,
				},
			},
		)

	data, status, err := c.TransportChannels(Channels{Active: true})
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)

	t.Assert().Len(data, 1)
}

func (t *MGClientTest) Test_ActivateTransportChannel() {
	c := t.client()
	ch := Channel{
		ID:   1,
		Type: "telegram",
		Name: "@my_shopping_bot",
		Settings: ChannelSettings{
			Status: Status{
				Delivered: ChannelFeatureNone,
				Read:      ChannelFeatureReceive,
			},
			Text: ChannelSettingsText{
				Creating:      ChannelFeatureBoth,
				Editing:       ChannelFeatureSend,
				Quoting:       ChannelFeatureReceive,
				Deleting:      ChannelFeatureBoth,
				MaxCharsCount: 2000,
			},
			Product: Product{
				Creating: ChannelFeatureSend,
				Deleting: ChannelFeatureSend,
			},
			Order: Order{
				Creating: ChannelFeatureBoth,
				Deleting: ChannelFeatureSend,
			},
			Image: ChannelSettingsFilesBase{
				Creating: ChannelFeatureBoth,
			},
			File: ChannelSettingsFilesBase{
				Creating: ChannelFeatureBoth,
			},
		},
	}

	defer gock.Off()
	t.gock().
		Post(t.transportURL("channels")).
		Reply(http.StatusCreated).
		JSON(
			ActivateResponse{
				ChannelID:   1,
				ExternalID:  "external_id_1",
				ActivatedAt: time.Now(),
			},
		)

	data, status, err := c.ActivateTransportChannel(ch)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusCreated, status)

	t.Assert().Equal(uint64(1), data.ChannelID)
	t.Assert().Equal("external_id_1", data.ExternalID)
	t.Assert().NotEmpty(data.ActivatedAt.String())
}

func (t *MGClientTest) Test_ActivateNewTransportChannel() {
	c := t.client()
	ch := Channel{
		Type: "telegram",
		Name: "@my_shopping_bot",
		Settings: ChannelSettings{
			Status: Status{
				Delivered: ChannelFeatureNone,
				Read:      ChannelFeatureBoth,
			},
			Text: ChannelSettingsText{
				Creating: ChannelFeatureBoth,
				Editing:  ChannelFeatureSend,
				Quoting:  ChannelFeatureBoth,
				Deleting: ChannelFeatureSend,
			},
			Product: Product{
				Creating: ChannelFeatureSend,
				Deleting: ChannelFeatureSend,
			},
			Order: Order{
				Creating: ChannelFeatureBoth,
				Deleting: ChannelFeatureSend,
			},
			Image: ChannelSettingsFilesBase{
				Creating: ChannelFeatureBoth,
			},
			File: ChannelSettingsFilesBase{
				Creating: ChannelFeatureBoth,
			},
		},
	}

	defer gock.Off()

	t.gock().
		Post(t.transportURL("channels")).
		Reply(http.StatusCreated).
		JSON(
			ActivateResponse{
				ChannelID:   1,
				ExternalID:  "external_id_1",
				ActivatedAt: time.Now(),
			},
		)

	t.gock().
		Delete(t.transportURL("channels/1")).
		Reply(http.StatusOK).
		JSON(
			DeleteResponse{
				ChannelID:     1,
				DeactivatedAt: time.Now(),
			},
		)

	data, status, err := c.ActivateTransportChannel(ch)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusCreated, status)

	t.Assert().Equal(uint64(1), data.ChannelID)
	t.Assert().Equal("external_id_1", data.ExternalID)
	t.Assert().NotEmpty(data.ActivatedAt.String())

	deleteData, status, err := c.DeactivateTransportChannel(data.ChannelID)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().NotEmpty(deleteData.DeactivatedAt.String())
	t.Assert().Equal(uint64(1), deleteData.ChannelID)
}

func (t *MGClientTest) Test_UpdateTransportChannel() {
	c := t.client()
	ch := Channel{
		ID:   1,
		Name: "@my_shopping_bot_2",
		Settings: ChannelSettings{
			Status: Status{
				Delivered: ChannelFeatureNone,
				Read:      ChannelFeatureBoth,
			},
			Text: ChannelSettingsText{
				Creating: ChannelFeatureBoth,
				Editing:  ChannelFeatureBoth,
				Quoting:  ChannelFeatureBoth,
				Deleting: ChannelFeatureBoth,
			},
			Product: Product{
				Creating: ChannelFeatureSend,
				Deleting: ChannelFeatureSend,
			},
			Order: Order{
				Creating: ChannelFeatureBoth,
				Deleting: ChannelFeatureSend,
			},
			Image: ChannelSettingsFilesBase{
				Creating: ChannelFeatureBoth,
			},
			File: ChannelSettingsFilesBase{
				Creating: ChannelFeatureBoth,
			},
		},
	}

	defer gock.Off()
	t.gock().
		Put(t.transportURL("channels/1")).
		Reply(http.StatusOK).
		JSON(
			UpdateResponse{
				ChannelID:  uint64(1),
				ExternalID: "external_id_1",
				UpdatedAt:  time.Now(),
			},
		)

	data, status, err := c.UpdateTransportChannel(ch)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().Equal(uint64(1), data.ChannelID)
	t.Assert().Equal("external_id_1", data.ExternalID)
	t.Assert().NotEmpty(data.UpdatedAt.String())
}

func (t *MGClientTest) Test_TransportTemplates() {
	c := t.client()

	defer gock.Off()
	t.gock().
		Get(t.transportURL("templates")).
		Reply(http.StatusOK).
		JSON(
			[]Template{
				{
					Code:      "tpl_code",
					ChannelID: 1,
					Name:      "Test Template",
					Enabled:   true,
					Type:      TemplateTypeText,
					Template: []TemplateItem{
						{
							Type: TemplateItemTypeText,
							Text: "Hello, ",
						},
						{
							Type:    TemplateItemTypeVar,
							VarType: TemplateVarFirstName,
						},
						{
							Type: TemplateItemTypeText,
							Text: "! We're glad to see you back in our store.",
						},
					},
				},
			},
		)

	data, status, err := c.TransportTemplates()
	t.Assert().NoError(err, fmt.Sprintf("%d %s", status, err))
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().Len(data, 1)

	for _, item := range data {
		for _, tpl := range item.Template {
			if tpl.Type == TemplateItemTypeText {
				t.Assert().Empty(tpl.VarType)
			} else {
				t.Assert().Empty(tpl.Text)
				t.Assert().NotEmpty(tpl.VarType)

				if _, ok := templateVarAssoc[tpl.VarType]; !ok {
					t.T().Errorf("unknown TemplateVar type %s", tpl.VarType)
				}
			}
		}
	}
}

func (t *MGClientTest) Test_ActivateTemplate() {
	c := t.client()
	req := ActivateTemplateRequest{
		Code: "tplCode",
		Type: TemplateTypeText,
		UpdateTemplateRequest: UpdateTemplateRequest{
			Name: "tplCode",
			Template: []TemplateItem{
				{
					Type: TemplateItemTypeText,
					Text: "Hello ",
				},
				{
					Type:    TemplateItemTypeVar,
					VarType: TemplateVarFirstName,
				},
				{
					Type: TemplateItemTypeText,
					Text: "!",
				},
			},
			RejectionReason:    "",
			VerificationStatus: TemplateStatusApproved,
		},
	}

	defer gock.Off()
	t.gock().
		Post("/channels/1/templates").
		Reply(http.StatusCreated).
		JSON(map[string]interface{}{})

	status, err := c.ActivateTemplate(1, req)
	t.Assert().NoError(err, fmt.Sprintf("%d %s", status, err))
	t.Assert().Equal(http.StatusCreated, status)
}

func (t *MGClientTest) Test_UpdateTemplate() {
	c := t.client()
	tpl := UpdateTemplateRequest{
		Name: "updated name",
		Template: []TemplateItem{
			{
				Type: TemplateItemTypeText,
				Text: "Welcome ",
			},
			{
				Type:    TemplateItemTypeVar,
				VarType: TemplateVarFirstName,
			},
			{
				Type: TemplateItemTypeText,
				Text: "!",
			},
		},
	}

	defer gock.Off()

	t.gock().
		Filter(
			func(r *http.Request) bool {
				return r.Method == http.MethodPut &&
					r.URL.Path == "/api/transport/v1/channels/1/templates/encodable#code"
			},
		).
		Reply(http.StatusOK).
		JSON(map[string]interface{}{})

	t.gock().
		Get(t.transportURL("templates")).
		Reply(http.StatusOK).
		JSON([]ActivateTemplateRequest{ActivateTemplateRequest{
			UpdateTemplateRequest: tpl,
			Code:                  "encodable#code",
			Type:                  TemplateTypeText,
		}})

	status, err := c.UpdateTemplate(1, "encodable#code", tpl)
	t.Assert().NoError(err, fmt.Sprintf("%d %s", status, err))

	templates, status, err := c.TransportTemplates()
	t.Assert().NoError(err, fmt.Sprintf("%d %s", status, err))

	for _, template := range templates {
		if template.Code == "encodable#code" {
			t.Assert().Equal(tpl.Name, template.Name)
		}
	}
}

func (t *MGClientTest) Test_UpdateTemplateFail() {
	c := t.client()
	tpl := UpdateTemplateRequest{
		Name: "updated name",
		Template: []TemplateItem{
			{
				Type: TemplateItemTypeText,
				Text: "Welcome ",
			},
			{
				Type:    TemplateItemTypeVar,
				VarType: TemplateVarFirstName,
			},
			{
				Type: TemplateItemTypeText,
				Text: "!",
			},
		},
	}

	defer gock.Off()
	t.gock().
		Reply(http.StatusBadRequest).
		JSON(
			map[string][]string{
				"errors": {"Some weird error message..."},
			},
		)

	status, err := c.UpdateTemplate(1, "encodable#code", tpl)
	t.Assert().Error(err, fmt.Sprintf("%d %s", status, err))
}

func (t *MGClientTest) Test_DeactivateTemplate() {
	c := t.client()

	defer gock.Off()
	t.gock().
		Filter(
			func(r *http.Request) bool {
				return r.Method == http.MethodDelete &&
					r.URL.Path == t.transportURL("channels/1/templates/test_template#code")
			},
		).
		Reply(http.StatusOK).
		JSON(map[string]interface{}{})

	status, err := c.DeactivateTemplate(1, "test_template#code")
	t.Assert().NoError(err, fmt.Sprintf("%d %s", status, err))
}

func (t *MGClientTest) Test_TextMessages() {
	c := t.client()

	snd := SendData{
		Message: Message{
			ExternalID: "external_id",
			Type:       MsgTypeText,
			Text:       "hello!",
			PageLink:   "https://example.loca/catalog/1",
		},
		Originator: OriginatorCustomer,
		Customer: Customer{
			ExternalID: "6",
			Nickname:   "octopus",
			Firstname:  "Joe",
			Utm: &Utm{
				Source: "test-source",
				Term:   "",
			},
		},
		Channel:        1,
		ExternalChatID: "24798237492374",
	}

	defer gock.Off()
	t.gock().
		Post(t.transportURL("messages")).
		Filter(func(request *http.Request) bool {
			data, err := ioutil.ReadAll(request.Body)
			if err != nil {
				return false
			}
			request.Body = ioutil.NopCloser(bytes.NewReader(data))

			var snd SendData
			t.Require().NoError(json.Unmarshal(data, &snd))
			return t.Assert().Equal("https://example.loca/catalog/1", snd.Message.PageLink)
		}).
		Reply(http.StatusOK).
		JSON(
			MessagesResponse{
				MessageID: 1,
				Time:      time.Now(),
			},
		)

	data, status, err := c.Messages(snd)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().NotEmpty(data.Time.String())
	t.Assert().Equal(1, data.MessageID)
}

func (t *MGClientTest) Test_ImageMessages() {
	c := t.client()

	defer gock.Off()

	t.gock().
		Post(t.transportURL("files/upload_by_url")).
		Reply(http.StatusOK).
		JSON(
			UploadFileResponse{
				ID:        "1",
				Hash:      "1",
				Type:      "image/png",
				MimeType:  "",
				Size:      1024,
				CreatedAt: time.Now(),
			},
		)

	t.gock().
		Post(t.transportURL("messages")).
		Reply(http.StatusOK).
		JSON(
			MessagesResponse{
				MessageID: 1,
				Time:      time.Now(),
			},
		)

	uploadFileResponse, st, err := c.UploadFileByURL(
		UploadFileByUrlRequest{
			Url: "https://via.placeholder.com/1",
		},
	)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, st)
	t.Assert().Equal("1", uploadFileResponse.ID)

	snd := SendData{
		Message: Message{
			ExternalID: "file",
			Type:       MsgTypeImage,
			Items:      []Item{{ID: uploadFileResponse.ID}},
		},
		Originator: OriginatorCustomer,
		Customer: Customer{
			ExternalID: "6",
			Nickname:   "octopus",
			Firstname:  "Joe",
		},
		Channel:        1,
		ExternalChatID: "24798237492374",
	}

	data, status, err := c.Messages(snd)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().NotEmpty(data.Time.String())
	t.Assert().Equal(1, data.MessageID)
}

func (t *MGClientTest) Test_UpdateMessages() {
	c := t.client()

	sndU := EditMessageRequest{
		EditMessageRequestMessage{
			ExternalID: "editing",
			Text:       "hello hello!",
			PageLink:   "https://example.local/1",
		},
		1,
	}

	defer gock.Off()
	t.gock().
		Put(t.transportURL("messages")).
		Filter(func(request *http.Request) bool {
			data, err := ioutil.ReadAll(request.Body)
			if err != nil {
				return false
			}
			request.Body = ioutil.NopCloser(bytes.NewReader(data))

			var snd SendData
			t.Require().NoError(json.Unmarshal(data, &snd))
			return t.Assert().Equal("https://example.local/1", snd.Message.PageLink)
		}).
		Reply(http.StatusOK).
		JSON(
			MessagesResponse{
				MessageID: 1,
				Time:      time.Now(),
			},
		)

	dataU, status, err := c.UpdateMessages(sndU)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().NotEmpty(dataU.Time.String())
	t.Assert().Equal(1, dataU.MessageID)
}

func (t *MGClientTest) Test_ReadUntil() {
	c := t.client()
	req := MarkMessagesReadUntilRequest{
		CustomerExternalID: "customer",
		ChannelID:          1,
		Until:              time.Unix(0, 0),
	}

	defer gock.Off()
	t.gock().
		Post("messages/read_until").
		Reply(http.StatusOK).
		JSON(MarkMessagesReadUntilResponse{
			IDs: []int64{1},
		})

	resp, st, err := c.ReadUntil(req)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, st)
	t.Assert().Equal([]int64{1}, resp.IDs)
}

func (t *MGClientTest) Test_MessagesHistory() {
	c := t.client()

	snd := SendHistoryMessageRequest{
		Message: SendMessageRequestMessage{
			ExternalID: "external_id",
			Type:       MsgTypeText,
			Text:       "hello!",
		},
		Originator: OriginatorCustomer,
		Customer: &Customer{
			ExternalID: "6",
			Nickname:   "octopus",
			Firstname:  "Joe",
		},
		ChannelID:      1,
		ExternalChatID: "24798237492374",
	}

	defer gock.Off()
	t.gock().
		Post(t.transportURL("messages/history")).
		Reply(http.StatusOK).
		JSON(
			MessagesResponse{
				MessageID: 1,
				Time:      time.Now(),
			},
		)

	data, status, err := c.MessagesHistory(snd)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().NotEmpty(data.Time.String())
	t.Assert().Equal(1, data.MessageID)
}

func (t *MGClientTest) Test_MarkMessageReadAndDelete() {
	c := t.client()

	snd := MarkMessageReadRequest{
		MarkMessageReadRequestMessage{
			ExternalID: "external_1",
		},
		1,
	}

	defer gock.Off()
	t.gock().
		Post(t.transportURL("messages/read")).
		Reply(http.StatusOK).
		JSON(MarkMessageReadResponse{})

	t.gock().
		Delete(t.transportURL("messages")).
		JSON(
			DeleteData{
				Message: Message{
					ExternalID: "deleted",
				},
				Channel: 1,
			},
		).
		Reply(http.StatusOK).
		JSON(
			MessagesResponse{
				MessageID: 2,
				Time:      time.Now(),
			},
		)

	_, status, err := c.MarkMessageRead(snd)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)

	previousChatMessage, status, err := c.DeleteMessage(
		DeleteData{
			Message{
				ExternalID: "deleted",
			},
			1,
		},
	)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().Equal(2, previousChatMessage.MessageID)
}

func (t *MGClientTest) Test_DeactivateTransportChannel() {
	c := t.client()

	defer gock.Off()
	t.gock().
		Delete(t.transportURL("channels/1")).
		Reply(http.StatusOK).
		JSON(
			DeleteResponse{
				ChannelID:     1,
				DeactivatedAt: time.Now(),
			},
		)

	deleteData, status, err := c.DeactivateTransportChannel(1)
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)
	t.Assert().NotEmpty(deleteData.DeactivatedAt.String())
	t.Assert().Equal(uint64(1), deleteData.ChannelID)
}

func (t *MGClientTest) Test_UploadFile() {
	c := t.client()

	// 1x1 png picture
	img := "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEX/TQBcNTh/AAAAAXRSTlPM0jRW/QAAAApJREFUeJxjYgAAAAYAAzY3fKgAAAAASUVORK5CYII="
	binary, err := base64.StdEncoding.DecodeString(img)
	if err != nil {
		t.T().Errorf("cannot convert base64 to binary: %s", err)
	}

	resp := UploadFileResponse{
		ID:        "1",
		Hash:      "1",
		Type:      "image/png",
		MimeType:  "",
		Size:      1024,
		CreatedAt: time.Now(),
	}

	defer gock.Off()
	t.gock().
		Post(t.transportURL("files/upload")).
		Body(bytes.NewReader(binary)).
		Reply(http.StatusOK).
		JSON(resp)

	data, status, err := c.UploadFile(bytes.NewReader(binary))
	t.Require().NoError(err)
	t.Assert().Equal(http.StatusOK, status)

	resp.CreatedAt = data.CreatedAt
	t.Assert().Equal(resp, data)
}

func (t *MGClientTest) Test_SuccessHandleError() {
	client := t.client()
	json := `{"errors": ["Channel not found"]}`

	defer gock.Off()
	t.gock().
		Delete(t.transportURL("channels/123")).
		Reply(http.StatusInternalServerError)

	t.gock().
		Delete(t.transportURL("channels/455")).
		Reply(http.StatusBadRequest).
		JSON(json)

	_, statusCode, err := client.DeactivateTransportChannel(123)

	t.Assert().Equal(http.StatusInternalServerError, statusCode)
	t.Assert().IsType(new(HTTPClientError), err)
	t.Assert().Equal(internalServerError, err.Error())
	var serverErr *HTTPClientError
	if errors.As(err, &serverErr) {
		t.Assert().Nil(serverErr.Response)
	} else {
		t.Fail("Unexpected type of error")
	}

	_, statusCode, err = client.DeactivateTransportChannel(455)
	t.Assert().Equal(http.StatusBadRequest, statusCode)
	t.Assert().IsType(new(HTTPClientError), err)
	t.Assert().Equal("Channel not found", err.Error())
}