niki/delivery/http_server/end2end/benefactor_kindboxreqs_test.go

//go:build end2end

package end2end

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"git.gocasts.ir/ebhomengo/niki/delivery/http_server/end2end/setup"
	"git.gocasts.ir/ebhomengo/niki/entity"
	benefactoreparam "git.gocasts.ir/ebhomengo/niki/param/benefactor/benefactor"
	benefactorkindboxreqparam "git.gocasts.ir/ebhomengo/niki/param/benefactor/kind_box_req"
	"github.com/labstack/echo/v4"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/suite"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"
)

// Define a suite struct that embeds suite.Suite
type BenefactorKindBoxReqsTestSuite struct {
	suite.Suite
	benefactorPhone string
	benefactorID    uint
	kindBoxReqID    uint
	getAllExpected  map[string]interface{}
	getExpected     map[string]interface{}
	createData      benefactorkindboxreqparam.KindBoxReqAddRequest
	updateData      benefactorkindboxreqparam.KindBoxReqUpdateRequest
}

// SetupTest will run before each test in the suite
func (suite *BenefactorKindBoxReqsTestSuite) SetupTest() {
	teardown := setup.SeedMariaDB(testContainer.GetMariaDBConfig())
	suite.T().Cleanup(teardown)
	suite.benefactorPhone = "09384664404"
	suite.benefactorID = uint(1)
	suite.kindBoxReqID = uint(1)
	suite.getAllExpected = map[string]interface{}{
		"count": 5,
	}
	suite.getExpected = map[string]interface{}{
		"kind_box_type":      entity.KindBoxOnTable,
		"deliver_address":    uint(1),
		"deliver_refer_date": time.Now().AddDate(0, 0, 7).UTC(),
		"deliver_refer_time": uint(1),
	}
	suite.createData = benefactorkindboxreqparam.KindBoxReqAddRequest{
		KindBoxType:        entity.KindBoxCylindrical,
		DeliverAddressID:   uint(1),
		DeliverReferDate:   time.Now().AddDate(0, 0, 7).UTC(),
		DeliverReferTimeID: uint(1),
		CountRequested:     uint(5),
	}
	suite.updateData = benefactorkindboxreqparam.KindBoxReqUpdateRequest{
		KindBoxType:        entity.KindBoxCylindrical,
		CountRequested:     uint(10),
		Description:        "updated description",
		DeliverReferTimeID: uint(1),
		DeliverReferDate:   time.Now().AddDate(0, 0, 7).UTC(),
		DeliverAddressID:   uint(1),
	}
}

// TODO: find better place
// loginBenefactor utility function
func (suite *BenefactorKindBoxReqsTestSuite) loginBenefactor() string {
	sendOTPRes, err := services.BenefactorSvc.SendOtp(context.Background(), benefactoreparam.SendOtpRequest{
		PhoneNumber: suite.benefactorPhone,
	})
	if err != nil {
		suite.T().Fatalf("failed to send OTP: %s", err)
	}
	registerRes, err := services.BenefactorSvc.LoginOrRegister(context.Background(), benefactoreparam.LoginOrRegisterRequest{
		PhoneNumber:      suite.benefactorPhone,
		VerificationCode: sendOTPRes.Code,
	})
	if err != nil {
		suite.T().Fatalf("failed to login or register: %s", err)
	}
	return registerRes.Tokens.AccessToken
}

// Utility function to create and send HTTP requests
func (suite *BenefactorKindBoxReqsTestSuite) createRequest(method, url string, body interface{}) *httptest.ResponseRecorder {
	var buf bytes.Buffer
	if body != nil {
		if err := json.NewEncoder(&buf).Encode(body); err != nil {
			suite.T().Fatalf("could not encode body: %s", err)
		}
	}
	token := suite.loginBenefactor()
	req := httptest.NewRequest(method, url, &buf)
	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
	req.Header.Set(echo.HeaderAuthorization, fmt.Sprintf("Bearer %s", token))
	rec := httptest.NewRecorder()
	testServer.Serve(rec, req)
	return rec
}

// Test to get all kind box requests
func (suite *BenefactorKindBoxReqsTestSuite) TestBenefactorKindBoxReqs_GetAll_Success() {
	rec := suite.createRequest(http.MethodGet, "/benefactor/kindboxreqs/", nil)

	assert.Equal(suite.T(), http.StatusOK, rec.Code)

	var response benefactorkindboxreqparam.GetAllResponse
	err := json.NewDecoder(rec.Body).Decode(&response)
	if err != nil {
		suite.T().Fatalf("failed to decode response body: %s", err)
	}
	assert.Equal(suite.T(), suite.getAllExpected["count"], len(response.AllKindBoxReq))

}

// Test to get a specific kind box request by ID
func (suite *BenefactorKindBoxReqsTestSuite) TestBenefactorKindBoxReqs_Get_Success() {

	rec := suite.createRequest(http.MethodGet, fmt.Sprintf("/benefactor/kindboxreqs/%d", suite.benefactorID), nil)

	assert.Equal(suite.T(), http.StatusOK, rec.Code)

	var response benefactorkindboxreqparam.KindBoxReqGetResponse
	err := json.NewDecoder(rec.Body).Decode(&response)
	if err != nil {
		suite.T().Fatalf("failed to decode response body: %s", err)
	}

	assert.Equal(suite.T(), suite.benefactorID, response.KindBoxReq.ID)
	assert.Equal(suite.T(), suite.getExpected["kind_box_type"], response.KindBoxReq.KindBoxType)
	assert.Equal(suite.T(), suite.getExpected["deliver_address"], response.KindBoxReq.DeliverAddressID)
	assert.Equal(
		suite.T(),
		suite.getExpected["deliver_refer_date"].(time.Time).Format("2006-01-02"),
		response.KindBoxReq.DeliverReferDate.Format("2006-01-02"),
	)
	assert.Equal(suite.T(), suite.getExpected["deliver_refer_time"], response.KindBoxReq.DeliverReferTimeID)
}

// Test to create a kind box request
func (suite *BenefactorKindBoxReqsTestSuite) TestBenefactorKindBoxReqs_Create_Success() {

	rec := suite.createRequest(http.MethodPost, "/benefactor/kindboxreqs/", suite.createData)
	assert.Equal(suite.T(), http.StatusCreated, rec.Code)

	var response benefactorkindboxreqparam.KindBoxReqAddResponse
	err := json.NewDecoder(rec.Body).Decode(&response)
	if err != nil {
		suite.T().Fatalf("failed to decode response body: %s", err)
	}
	assert.Equal(suite.T(), suite.createData.KindBoxType, response.KindBoxReq.KindBoxType)
	assert.Equal(suite.T(), suite.createData.DeliverAddressID, response.KindBoxReq.DeliverAddressID)
	assert.Equal(suite.T(), suite.createData.DeliverReferDate, response.KindBoxReq.DeliverReferDate)
	assert.Equal(
		suite.T(),
		suite.getExpected["deliver_refer_date"].(time.Time).Format("2006-01-02"),
		response.KindBoxReq.DeliverReferDate.Format("2006-01-02"),
	)
	assert.Equal(suite.T(), suite.createData.CountRequested, response.KindBoxReq.CountRequested)
}

// Test to update a kind box request
func (suite *BenefactorKindBoxReqsTestSuite) TestBenefactorKindBoxReqs_Update_Success() {

	rec := suite.createRequest(
		http.MethodPut, fmt.Sprintf("/benefactor/kindboxreqs/%d", suite.kindBoxReqID), suite.updateData,
	)
	assert.Equal(suite.T(), http.StatusNoContent, rec.Code)

	kindBoxReq, err := services.BenefactorKindBoxReqSvc.Get(context.Background(),
		benefactorkindboxreqparam.KindBoxReqGetRequest{
			BenefactorID: suite.benefactorID,
			KindBoxReqID: suite.kindBoxReqID,
		},
	)
	if err != nil {
		suite.T().Fatalf("failed to get kind box request: %s", err)
	}

	assert.Equal(suite.T(), suite.updateData.KindBoxType, kindBoxReq.KindBoxType)
	assert.Equal(suite.T(), suite.updateData.CountRequested, kindBoxReq.CountRequested)
	assert.Equal(suite.T(), suite.updateData.Description, kindBoxReq.Description)
	assert.Equal(suite.T(), suite.updateData.DeliverReferTimeID, kindBoxReq.DeliverReferTimeID)
	assert.Equal(suite.T(), suite.updateData.DeliverReferDate, kindBoxReq.DeliverReferDate)
	assert.Equal(suite.T(), suite.updateData.DeliverAddressID, kindBoxReq.DeliverAddressID)

}

func (suite *BenefactorKindBoxReqsTestSuite) TestBenefactorKindBoxReqs_Delete_Success() {
	rec := suite.createRequest(
		http.MethodDelete, fmt.Sprintf("/benefactor/kindboxreqs/%d", suite.kindBoxReqID), nil,
	)
	assert.Equal(suite.T(), http.StatusOK, rec.Code)

	_, err := services.BenefactorKindBoxReqSvc.Get(context.Background(),
		benefactorkindboxreqparam.KindBoxReqGetRequest{
			BenefactorID: suite.benefactorID,
			KindBoxReqID: suite.kindBoxReqID,
		},
	)
	// TODO: Fix to assert equal to errmsg.ErrorMsgNotFound
	assert.Error(suite.T(), err)
}

// Entry point for the test suite
func TestBenefactorKindBoxReqsTestSuite(t *testing.T) {
	suite.Run(t, new(BenefactorKindBoxReqsTestSuite))
}