forked from ebhomengo/niki
126 lines
3.5 KiB
Go
126 lines
3.5 KiB
Go
//go:build end2end
|
|
// +build end2end
|
|
|
|
package end2end
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"git.gocasts.ir/ebhomengo/niki/delivery/http_server/end2end/setup"
|
|
"git.gocasts.ir/ebhomengo/niki/entity"
|
|
adminserviceparam "git.gocasts.ir/ebhomengo/niki/param/admin/admin"
|
|
adminParam "git.gocasts.ir/ebhomengo/niki/param/admin/kind_box"
|
|
agentParam "git.gocasts.ir/ebhomengo/niki/param/agent/kind_box"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const (
|
|
agentPhoneNumber = "09384664403"
|
|
agentPassword = "Abc123456"
|
|
agentID uint = 4
|
|
)
|
|
|
|
func TestAgent_KindBox_Get(t *testing.T) {
|
|
kindBox := entity.KindBox{
|
|
ID: 2,
|
|
SerialNumber: "serial-2",
|
|
ReceiverAgentID: agentID,
|
|
Status: entity.KindBoxAssignedReceiverAgentStatus,
|
|
}
|
|
url := fmt.Sprintf("/agents/kindboxes/%d", kindBox.ID)
|
|
|
|
teardown := setup.SeedMariaDB(testContainer.GetMariaDBConfig())
|
|
t.Cleanup(teardown)
|
|
|
|
token, err := getToken()
|
|
if err != nil {
|
|
t.Fatalf("could not login: %s", err)
|
|
}
|
|
|
|
resRecord := createRequest(http.MethodGet, url, token, nil)
|
|
var res agentParam.GetKindBoxResponse
|
|
err = json.NewDecoder(resRecord.Body).Decode(&res)
|
|
assert.NoError(t, err, "could not decode response body")
|
|
|
|
assert.Equal(t, http.StatusOK, resRecord.Code)
|
|
assert.Equal(t, kindBox.ID, res.ID)
|
|
assert.Equal(t, kindBox.ReceiverAgentID, res.ReceiverAgentID)
|
|
assert.Equal(t, kindBox.SerialNumber, res.SerialNumber)
|
|
assert.Equal(t, kindBox.Status, res.Status)
|
|
}
|
|
|
|
// TODO: we can add more tests for return.
|
|
// we can get check if the data is updated in db as expects
|
|
// or api should return error if the kindbox is not for this agent
|
|
// CHECK: is the flow correct? agent can change kindbox status to return no matter what was the previous status?
|
|
func TestAgent_KindBox_Return(t *testing.T) {
|
|
kindBoxReqBody := agentParam.ReturnKindBoxRequest{SerialNumber: "new-serial"}
|
|
var kindBoxID uint = 2
|
|
url := fmt.Sprintf("/agents/kindboxes/return/%d", kindBoxID)
|
|
|
|
teardown := setup.SeedMariaDB(testContainer.GetMariaDBConfig())
|
|
t.Cleanup(teardown)
|
|
|
|
token, err := getToken()
|
|
if err != nil {
|
|
t.Fatalf("could not login: %s", err)
|
|
}
|
|
|
|
resRecord := createRequest(http.MethodPatch, url, token, kindBoxReqBody)
|
|
updatedKindBox, err := services.AdminKindBoxSvc.Get(
|
|
context.Background(),
|
|
adminParam.KindBoxGetRequest{KindBoxID: kindBoxID},
|
|
)
|
|
|
|
assert.Equal(t, http.StatusNoContent, resRecord.Code)
|
|
assert.Equal(t, kindBoxReqBody.SerialNumber, updatedKindBox.SerialNumber)
|
|
assert.Equal(t, agentID, updatedKindBox.ReceiverAgentID)
|
|
assert.Equal(t, entity.KindBoxReturnedStatus, updatedKindBox.Status)
|
|
}
|
|
|
|
func getToken() (string, error) {
|
|
res, err := services.AdminSvc.LoginWithPhoneNumber(
|
|
context.Background(),
|
|
adminserviceparam.LoginWithPhoneNumberRequest{
|
|
PhoneNumber: agentPhoneNumber,
|
|
Password: agentPassword,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return res.Tokens.AccessToken, nil
|
|
}
|
|
|
|
func createRequest(
|
|
method string,
|
|
url string,
|
|
token string,
|
|
body any,
|
|
) *httptest.ResponseRecorder {
|
|
var buf bytes.Buffer
|
|
if body != nil {
|
|
err := json.NewEncoder(&buf).Encode(body)
|
|
if err != nil {
|
|
log.Fatalf("could not encode body: %v", err)
|
|
}
|
|
}
|
|
|
|
req := httptest.NewRequest(method, url, &buf)
|
|
req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
|
req.Header.Set(echo.HeaderAuthorization, fmt.Sprintf("Bearer %s", token))
|
|
res := httptest.NewRecorder()
|
|
|
|
// Serve the request
|
|
testServer.Serve(res, req)
|
|
return res
|
|
}
|