memos/test/server/auth_test.go

package testserver

import (
	"bytes"
	"context"
	"encoding/json"
	"testing"

	"github.com/pkg/errors"
	"github.com/stretchr/testify/require"
	apiv1 "github.com/usememos/memos/api/v1"
)

func TestAuthServer(t *testing.T) {
	ctx := context.Background()
	s, err := NewTestingServer(ctx, t)
	require.NoError(t, err)
	defer s.Shutdown(ctx)

	signup := &apiv1.SignUp{
		Username: "testuser",
		Password: "testpassword",
	}
	user, err := s.postAuthSignUp(signup)
	require.NoError(t, err)
	require.Equal(t, signup.Username, user.Username)

	signin := &apiv1.SignIn{
		Username: "testuser",
		Password: "testpassword",
	}
	user, err = s.postAuthSignIn(signin)
	require.NoError(t, err)
	require.Equal(t, signup.Username, user.Username)
	err = s.postSignOut()
	require.NoError(t, err)
	_, err = s.getCurrentUser()
	require.Error(t, err)
}

func (s *TestingServer) postAuthSignUp(signup *apiv1.SignUp) (*apiv1.User, error) {
	rawData, err := json.Marshal(&signup)
	if err != nil {
		return nil, errors.Wrap(err, "failed to marshal signup")
	}
	reader := bytes.NewReader(rawData)
	body, err := s.post("/api/v1/auth/signup", reader, nil)
	if err != nil {
		return nil, errors.Wrap(err, "fail to post request")
	}

	buf := &bytes.Buffer{}
	_, err = buf.ReadFrom(body)
	if err != nil {
		return nil, errors.Wrap(err, "fail to read response body")
	}

	user := &apiv1.User{}
	if err = json.Unmarshal(buf.Bytes(), user); err != nil {
		return nil, errors.Wrap(err, "fail to unmarshal post signup response")
	}
	return user, nil
}

func (s *TestingServer) postAuthSignIn(signip *apiv1.SignIn) (*apiv1.User, error) {
	rawData, err := json.Marshal(&signip)
	if err != nil {
		return nil, errors.Wrap(err, "failed to marshal signin")
	}
	reader := bytes.NewReader(rawData)
	body, err := s.post("/api/v1/auth/signin", reader, nil)
	if err != nil {
		return nil, errors.Wrap(err, "fail to post request")
	}

	buf := &bytes.Buffer{}
	_, err = buf.ReadFrom(body)
	if err != nil {
		return nil, errors.Wrap(err, "fail to read response body")
	}

	user := &apiv1.User{}
	if err = json.Unmarshal(buf.Bytes(), user); err != nil {
		return nil, errors.Wrap(err, "fail to unmarshal post signin response")
	}
	return user, nil
}

func (s *TestingServer) postSignOut() error {
	_, err := s.post("/api/v1/auth/signout", nil, nil)
	if err != nil {
		return errors.Wrap(err, "fail to post request")
	}
	return nil
}