go-svc-configure/test/my_config_token_test.go

package main

import (
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"os/exec"
	"testing"

	"git.x2erp.com/qdy/go-base/model/request/configreq"
)

// 建立访问配置中心的token
func TestCreateConfigToken(t *testing.T) {
	// 清除测试缓存
	cmd := exec.Command("go", "clean", "-testcache")
	if err := cmd.Run(); err != nil {
		t.Logf("清除测试缓存失败: %v", err)
		// 继续执行测试
	}

	httpClient := &http.Client{}

	// 首先登录获取有效token
	loginURL := "http://localhost:8080/api/login/user"
	loginData := map[string]string{
		"user_id":  "test-user-001",
		"password": "password123",
	}
	loginJSON, _ := json.Marshal(loginData)

	loginReq, err := http.NewRequest("POST", loginURL, bytes.NewReader(loginJSON))
	if err != nil {
		t.Fatalf("创建登录请求失败: %v", err)
	}
	loginReq.Header.Set("Content-Type", "application/json")

	loginResp, err := httpClient.Do(loginReq)
	if err != nil {
		t.Fatalf("发送登录请求失败: %v", err)
	}
	defer loginResp.Body.Close()

	loginBody, err := io.ReadAll(loginResp.Body)
	if err != nil {
		t.Fatalf("读取登录响应失败: %v", err)
	}

	var loginResult map[string]interface{}
	if err := json.Unmarshal(loginBody, &loginResult); err != nil {
		t.Fatalf("解析登录响应失败: %v", err)
	}

	if !loginResult["success"].(bool) {
		t.Fatalf("登录失败: %v", loginResult)
	}

	authToken := loginResult["data"].(string)
	t.Logf("获取到认证token: %s", authToken[:50]+"...")

	// 使用获取的token创建配置token
	url := "http://localhost:8080/api/create/config/token"
	data := configreq.ConfigTokenRequest{
		ProjectID:   "test-project",
		ServiceName: "test-service",
		ExpiresDays: 30,
	}
	jsonData, err := json.Marshal(data)
	if err != nil {
		t.Fatalf("JSON序列化失败: %v", err)
	}

	req, err := http.NewRequest("POST", url, bytes.NewReader(jsonData))
	if err != nil {
		t.Fatalf("创建请求失败: %v", err)
	}

	// Token Auth 认证
	req.Header.Set("Authorization", "Bearer "+authToken)
	req.Header.Set("Content-Type", "application/json")

	resp, err := httpClient.Do(req)
	if err != nil {
		t.Fatalf("发送请求失败: %v", err)
	}
	defer resp.Body.Close()

	// 读取响应
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		t.Fatalf("读取响应失败: %v", err)
	}

	var result map[string]interface{}
	if err := json.Unmarshal(body, &result); err != nil {
		t.Fatalf("解析响应失败: %v", err)
	}

	t.Logf("状态码: %d", resp.StatusCode)
	t.Logf("响应体: %s", string(body))

	// 验证响应
	if resp.StatusCode != 200 {
		t.Errorf("期望状态码200，实际: %d", resp.StatusCode)
	}

	if !result["success"].(bool) {
		t.Errorf("创建配置token失败: %v", result)
	}

	if data, ok := result["data"].(string); !ok || data == "" {
		t.Errorf("未返回有效的配置token")
	} else {
		t.Logf("成功创建配置token: %s...", data[:50])
	}
}