NdoleStudio / mtnmomo-go

import (

	"context"

	"net/http"

	"strings"

	"testing"

	"github.com/google/uuid"

	"github.com/NdoleStudio/mtnmomo-go/internal/helpers"

	"github.com/NdoleStudio/mtnmomo-go/internal/stubs"

	"github.com/stretchr/testify/assert"

)

func TestDisbursementsService_Token(t *testing.T) {

	// Setup

	t.Parallel()

	// Arrange

	requests := make([]*http.Request, 0)

	server := helpers.MakeRequestCapturingTestServer(http.StatusOK, [][]byte{stubs.DisbursementToken()}, &requests)

	client := New(

		WithBaseURL(server.URL),

		WithSubscriptionKey(testSubscriptionKey),

		WithDisbursementAccount(testAPIUser, testAPIKey),

	)

	// Act

	authToken, response, err := client.Disbursement.Token(context.Background())

	// Assert

	assert.Nil(t, err)

	assert.GreaterOrEqual(t, 1, len(requests))

	request := requests[0]

	username, password, _ := request.BasicAuth()

	assert.Equal(t, testAPIUser, username)

	assert.Equal(t, testAPIKey, password)

	assert.Equal(t, testSubscriptionKey, request.Header.Get(headerKeySubscriptionKey))

	assert.Equal(t, http.StatusOK, response.HTTPResponse.StatusCode)

	assert.Equal(t, int64(3600), authToken.ExpiresIn)

	assert.Equal(t, "access_token", authToken.TokenType)

	assert.Equal(t, "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c", authToken.AccessToken)

	// Teardown

	server.Close()

}

func TestDisbursementsService_Transfer(t *testing.T) {

	// Setup

	t.Parallel()

	// Arrange

	requests := make([]*http.Request, 0)

	server := helpers.MakeRequestCapturingTestServer(http.StatusOK, [][]byte{stubs.CollectionToken(), nil}, &requests)

	client := New(

		WithBaseURL(server.URL),

		WithSubscriptionKey(testSubscriptionKey),

		WithDisbursementAccount(testAPIUser, testAPIKey),

	)

	// Act

	response, err := client.Disbursement.Transfer(

		context.Background(),

		uuid.NewString(),

		&TransferParams{

			Amount:     "100",

			Currency:   "EUR",

			ExternalID: uuid.NewString(),

			Payee: &AccountHolder{

				PartyIDType: "MSISDN",

				PartyID:     "46733123453",

			},

			PayerMessage: "Test Payer Message",

			PayeeNote:    "Test Payee Note",

		},

		nil,

	)

	// Assert

	assert.Nil(t, err)

	assert.GreaterOrEqual(t, len(requests), 1)

	request := requests[len(requests)-1]

	assert.Equal(t, "/disbursement/v1_0/transfer", request.URL.Path)

	assert.True(t, strings.HasPrefix(request.Header.Get("Authorization"), "Bearer"))

	assert.Equal(t, testSubscriptionKey, request.Header.Get(headerKeySubscriptionKey))

	assert.Equal(t, http.StatusOK, response.HTTPResponse.StatusCode)

	// Teardown

	server.Close()

}

func TestDisbursementsService_GetTransferStatus(t *testing.T) {

	// Setup

	t.Parallel()

	// Arrange

	requests := make([]*http.Request, 0)

	responses := [][]byte{stubs.CollectionToken(), stubs.DisbursementTransferStatus()}

	server := helpers.MakeRequestCapturingTestServer(http.StatusOK, responses, &requests)

	client := New(

		WithBaseURL(server.URL),

		WithSubscriptionKey(testSubscriptionKey),

		WithDisbursementAccount(testAPIUser, testAPIKey),

	)

	referenceID := uuid.NewString()

	// Act

	status, response, err := client.Disbursement.GetTransferStatus(context.Background(), referenceID)

	// Assert

	assert.Nil(t, err)

	assert.GreaterOrEqual(t, len(requests), 1)

	request := requests[len(requests)-1]

	assert.Equal(t, "/disbursement/v1_0/transfer/"+referenceID, request.URL.Path)

	assert.True(t, strings.HasPrefix(request.Header.Get("Authorization"), "Bearer"))

	assert.Equal(t, testSubscriptionKey, request.Header.Get(headerKeySubscriptionKey))

	assert.Equal(t, http.StatusOK, response.HTTPResponse.StatusCode)

	financialTransactionID := "23503452"

	assert.Equal(t, &DisbursementTransactionStatus{

		Amount:                 "100",

		Currency:               "UGX",

		FinancialTransactionID: &financialTransactionID,

		ExternalID:             "947354",

		ReferenceID:            referenceID,

		Payee: &AccountHolder{

			PartyIDType: "MSISDN",

			PartyID:     "4656473839",

		},

		Status: "SUCCESSFUL",

	}, status)

	// Teardown

	server.Close()

}