mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-06-21 02:23:12 +08:00
6b2427f1c2
Switched all file operations from ioutil package to io and os packages due to the deprecation of ioutil in Go 1.16. This change contributes to the maintainability of the codebase, ensuring it's up-to-date with the current standard library. Additionally, implemented usage of filepath.Join for platform-independent file paths.
127 lines
2.7 KiB
Go
127 lines
2.7 KiB
Go
package tests
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.signoz.io/signoz/pkg/query-service/auth"
|
|
"go.signoz.io/signoz/pkg/query-service/model"
|
|
)
|
|
|
|
func invite(t *testing.T, email string) *model.InviteResponse {
|
|
q := endpoint + fmt.Sprintf("/api/v1/invite?email=%s", email)
|
|
resp, err := client.Get(q)
|
|
require.NoError(t, err)
|
|
|
|
defer resp.Body.Close()
|
|
b, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
|
|
var inviteResp model.InviteResponse
|
|
err = json.Unmarshal(b, &inviteResp)
|
|
require.NoError(t, err)
|
|
|
|
return &inviteResp
|
|
}
|
|
|
|
func register(email, password, token string) (string, error) {
|
|
q := endpoint + fmt.Sprintf("/api/v1/register")
|
|
|
|
req := auth.RegisterRequest{
|
|
Email: email,
|
|
Password: password,
|
|
InviteToken: token,
|
|
}
|
|
|
|
b, err := json.Marshal(req)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
resp, err := client.Post(q, "application/json", bytes.NewBuffer(b))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
b, err = io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return string(b), nil
|
|
}
|
|
|
|
func login(email, password, refreshToken string) (*model.LoginResponse, error) {
|
|
q := endpoint + fmt.Sprintf("/api/v1/login")
|
|
|
|
req := model.LoginRequest{
|
|
Email: email,
|
|
Password: password,
|
|
RefreshToken: refreshToken,
|
|
}
|
|
|
|
b, err := json.Marshal(req)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to marshal")
|
|
}
|
|
resp, err := client.Post(q, "application/json", bytes.NewBuffer(b))
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to post")
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
b, err = io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to read body")
|
|
}
|
|
|
|
loginResp := &model.LoginResponse{}
|
|
err = json.Unmarshal(b, loginResp)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal")
|
|
}
|
|
|
|
return loginResp, nil
|
|
}
|
|
|
|
func TestAuthInviteAPI(t *testing.T) {
|
|
t.Skip()
|
|
email := "abc@signoz.io"
|
|
resp := invite(t, email)
|
|
require.Equal(t, email, resp.Email)
|
|
require.NotNil(t, resp.InviteToken)
|
|
}
|
|
|
|
func TestAuthRegisterAPI(t *testing.T) {
|
|
email := "alice@signoz.io"
|
|
resp, err := register(email, "Password@123", "")
|
|
require.NoError(t, err)
|
|
require.Contains(t, resp, "user registered successfully")
|
|
|
|
}
|
|
|
|
func TestAuthLoginAPI(t *testing.T) {
|
|
t.Skip()
|
|
email := "abc-login@signoz.io"
|
|
password := "Password@123"
|
|
inv := invite(t, email)
|
|
|
|
resp, err := register(email, password, inv.InviteToken)
|
|
require.NoError(t, err)
|
|
require.Contains(t, resp, "user registered successfully")
|
|
|
|
loginResp, err := login(email, password, "")
|
|
require.NoError(t, err)
|
|
|
|
loginResp2, err := login("", "", loginResp.RefreshJwt)
|
|
require.NoError(t, err)
|
|
|
|
require.NotNil(t, loginResp2.AccessJwt)
|
|
}
|