asg-server/test/e2e/mobile_test.go

package e2e

import (
	"encoding/json"
	"testing"
	"time"

	"github.com/go-resty/resty/v2"
)

var (
	publicClient = resty.New().
			SetBaseURL(PublicURL + "/mobile/v1").
			SetTimeout(300 * time.Millisecond)
	privateClient = resty.New().
			SetBaseURL(PrivateURL + "/mobile/v1").
			SetTimeout(300 * time.Millisecond)
)

type mobileRegisterResponse struct {
	Token    string `json:"token"`
	Password string `json:"password"`
}

func mobileDeviceRegister(t *testing.T, client *resty.Client) mobileRegisterResponse {
	res, err := client.R().
		SetHeader("Content-Type", "application/json").
		SetBody(`{"name": "Public Device Name", "pushToken": "token"}`).
		Post("device")
	if err != nil {
		t.Fatal(err)
	}

	if !res.IsSuccess() {
		t.Fatal(res.StatusCode(), res.String())
	}

	var resp mobileRegisterResponse
	if err := json.Unmarshal(res.Body(), &resp); err != nil {
		t.Fatal(err)
	}

	return resp
}

func TestPublicDeviceRegister(t *testing.T) {
	cases := []struct {
		name               string
		headers            map[string]string
		expectedStatusCode int
	}{
		{
			name: "with valid token",
			headers: map[string]string{
				"Authorization": "Bearer 123456789",
			},
			expectedStatusCode: 201,
		},
		{
			name:               "without token",
			headers:            map[string]string{},
			expectedStatusCode: 201,
		},
		{
			name: "with invalid token",
			headers: map[string]string{
				"Authorization": "Bearer 987654321",
			},
			expectedStatusCode: 201,
		},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			res, err := publicClient.R().
				SetHeader("Content-Type", "application/json").
				SetBody(`{"name": "Public Device Name", "pushToken": "token"}`).
				SetHeaders(c.headers).
				Post("device")
			if err != nil {
				t.Fatal(err)
			}

			if res.StatusCode() != c.expectedStatusCode {
				t.Fatal(res.StatusCode(), res.String())
			}
		})
	}
}

func TestPrivateDeviceRegister(t *testing.T) {
	cases := []struct {
		name               string
		headers            map[string]string
		expectedStatusCode int
	}{
		{
			name: "with valid token",
			headers: map[string]string{
				"Authorization": "Bearer 123456789",
			},
			expectedStatusCode: 201,
		},
		{
			name:               "without token",
			headers:            map[string]string{},
			expectedStatusCode: 401,
		},
		{
			name: "with invalid token",
			headers: map[string]string{
				"Authorization": "Bearer 987654321",
			},
			expectedStatusCode: 401,
		},
	}

	client := privateClient

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			res, err := client.R().
				SetHeader("Content-Type", "application/json").
				SetBody(`{"name": "Private Device Name", "pushToken": "token"}`).
				SetHeaders(c.headers).
				Post("device")
			if err != nil {
				t.Fatal(err)
			}

			if res.StatusCode() != c.expectedStatusCode {
				t.Fatal(res.StatusCode(), res.String())
			}
		})
	}
}

func TestPublicDevicePasswordChange(t *testing.T) {
	device := mobileDeviceRegister(t, publicClient)

	cases := []struct {
		name               string
		headers            map[string]string
		body               string
		expectedStatusCode int
	}{
		{
			name: "with invalid token",
			headers: map[string]string{
				"Authorization": "Bearer 123456789",
			},
			body:               `{"currentPassword": "123456789", "newPassword": "123456789"}`,
			expectedStatusCode: 401,
		},
		{
			name: "with invalid password",
			headers: map[string]string{
				"Authorization": "Bearer " + device.Token,
			},
			body:               `{"currentPassword": "123456789", "newPassword": "changemeonemoretime"}`,
			expectedStatusCode: 401,
		},
		{
			name: "short password",
			headers: map[string]string{
				"Authorization": "Bearer " + device.Token,
			},
			body:               `{"currentPassword": "` + device.Password + `", "newPassword": "changeme"}`,
			expectedStatusCode: 400,
		},
		{
			name: "success",
			headers: map[string]string{
				"Authorization": "Bearer " + device.Token,
			},
			body:               `{"currentPassword": "` + device.Password + `", "newPassword": "changemeonemoretime"}`,
			expectedStatusCode: 204,
		},
		{
			name: "success with new password",
			headers: map[string]string{
				"Authorization": "Bearer " + device.Token,
			},
			body:               `{"currentPassword": "changemeonemoretime", "newPassword": "` + device.Password + `"}`,
			expectedStatusCode: 204,
		},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			res, err := publicClient.R().
				SetHeader("Content-Type", "application/json").
				SetBody(c.body).
				SetHeaders(c.headers).
				Patch("user/password")
			if err != nil {
				t.Fatal(err)
			}

			if res.StatusCode() != c.expectedStatusCode {
				t.Fatal(res.StatusCode(), res.String())
			}
		})
	}
}