mirror of
https://github.com/usememos/memos.git
synced 2025-06-05 22:09:59 +02:00
58
test/server/system_test.go
Normal file
58
test/server/system_test.go
Normal file
@ -0,0 +1,58 @@
|
||||
package testserver
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"testing"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/usememos/memos/api"
|
||||
)
|
||||
|
||||
func TestSystemServer(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
s, err := NewTestingServer(ctx, t)
|
||||
require.NoError(t, err)
|
||||
defer s.Shutdown(ctx)
|
||||
|
||||
status, err := s.getSystemStatus()
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, (*api.User)(nil), status.Host)
|
||||
|
||||
signup := &api.SignUp{
|
||||
Username: "testuser",
|
||||
Password: "testpassword",
|
||||
}
|
||||
user, err := s.postAuthSignup(signup)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, signup.Username, user.Username)
|
||||
|
||||
status, err = s.getSystemStatus()
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, user.ID, status.Host.ID)
|
||||
require.Equal(t, user.Username, status.Host.Username)
|
||||
}
|
||||
|
||||
func (s *TestingServer) getSystemStatus() (*api.SystemStatus, error) {
|
||||
body, err := s.get("/api/status", nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
buf := &bytes.Buffer{}
|
||||
_, err = buf.ReadFrom(body)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "fail to read response body")
|
||||
}
|
||||
|
||||
type SystemStatusResponse struct {
|
||||
Data *api.SystemStatus `json:"data"`
|
||||
}
|
||||
res := new(SystemStatusResponse)
|
||||
if err = json.Unmarshal(buf.Bytes(), res); err != nil {
|
||||
return nil, errors.Wrap(err, "fail to unmarshal get system status response")
|
||||
}
|
||||
return res.Data, nil
|
||||
}
|
Reference in New Issue
Block a user