复制项目

test/e2e/api/token/token.go

@@ -0,0 +1,149 @@
+// Copyright © 2023 OpenIM. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package token
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+)
+
+// API endpoints and other constants.
+const (
+	APIHost         = "http://127.0.0.1:10002"
+	UserTokenURL    = APIHost + "/auth/user_token"
+	UserRegisterURL = APIHost + "/user/user_register"
+	SecretKey       = "openIM123"
+	OperationID     = "1646445464564"
+)
+
+// UserTokenRequest represents a request to get a user token.
+type UserTokenRequest struct {
+	Secret     string `json:"secret"`
+	PlatformID int    `json:"platformID"`
+	UserID     string `json:"userID"`
+}
+
+// UserTokenResponse represents a response containing a user token.
+type UserTokenResponse struct {
+	Token   string `json:"token"`
+	ErrCode int    `json:"errCode"`
+}
+
+// User represents user data for registration.
+type User struct {
+	UserID   string `json:"userID"`
+	Nickname string `json:"nickname"`
+	FaceURL  string `json:"faceURL"`
+}
+
+// UserRegisterRequest represents a request to register a user.
+type UserRegisterRequest struct {
+	Users []User `json:"users"`
+}
+
+/* func main() {
+	// Example usage of functions
+	token, err := GetUserToken("openIM123456")
+	if err != nil {
+		log.Fatalf("Error getting user token: %v", err)
+	}
+	fmt.Println("Token:", token)
+
+	err = RegisterUser(token, "testUserID", "TestNickname", "https://example.com/image.jpg")
+	if err != nil {
+		log.Fatalf("Error registering user: %v", err)
+	}
+} */
+
+// GetUserToken requests a user token from the API.
+func GetUserToken(userID string) (string, error) {
+	reqBody := UserTokenRequest{
+		Secret:     SecretKey,
+		PlatformID: 1,
+		UserID:     userID,
+	}
+	reqBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return "", err
+	}
+
+	resp, err := http.Post(UserTokenURL, "application/json", bytes.NewBuffer(reqBytes))
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	var tokenResp UserTokenResponse
+	if err := json.NewDecoder(resp.Body).Decode(&tokenResp); err != nil {
+		return "", err
+	}
+
+	if tokenResp.ErrCode != 0 {
+		return "", fmt.Errorf("error in token response: %v", tokenResp.ErrCode)
+	}
+
+	return tokenResp.Token, nil
+}
+
+// RegisterUser registers a new user using the API.
+func RegisterUser(token, userID, nickname, faceURL string) error {
+	user := User{
+		UserID:   userID,
+		Nickname: nickname,
+		FaceURL:  faceURL,
+	}
+	reqBody := UserRegisterRequest{
+		Users: []User{user},
+	}
+	reqBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return err
+	}
+
+	client := &http.Client{}
+	req, err := http.NewRequest("POST", UserRegisterURL, bytes.NewBuffer(reqBytes))
+	if err != nil {
+		return err
+	}
+
+	req.Header.Add("Content-Type", "application/json")
+	req.Header.Add("operationID", OperationID)
+	req.Header.Add("token", token)
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	respBody, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+
+	var respData map[string]any
+	if err := json.Unmarshal(respBody, &respData); err != nil {
+		return err
+	}
+
+	if errCode, ok := respData["errCode"].(float64); ok && errCode != 0 {
+		return fmt.Errorf("error in user registration response: %v", respData)
+	}
+
+	return nil
+}