84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
package yushan
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestGenerateRequestID(t *testing.T) {
|
||
service := &YushanService{
|
||
config: YushanConfig{
|
||
ApiKey: "test_api_key_123",
|
||
},
|
||
}
|
||
|
||
id1 := service.generateRequestID()
|
||
|
||
// 等待一小段时间确保时间戳不同
|
||
time.Sleep(time.Millisecond)
|
||
|
||
id2 := service.generateRequestID()
|
||
|
||
if id1 == "" || id2 == "" {
|
||
t.Error("请求ID生成失败")
|
||
}
|
||
|
||
if id1 == id2 {
|
||
t.Error("不同时间生成的请求ID应该不同")
|
||
}
|
||
|
||
// 验证ID格式
|
||
if len(id1) < 20 { // yushan_ + 8位十六进制 + 其他
|
||
t.Errorf("请求ID长度不足，实际: %s", id1)
|
||
}
|
||
}
|
||
|
||
func TestGenerateRandomString(t *testing.T) {
|
||
service := &YushanService{}
|
||
|
||
str1, err := service.GenerateRandomString()
|
||
if err != nil {
|
||
t.Fatalf("生成随机字符串失败: %v", err)
|
||
}
|
||
|
||
str2, err := service.GenerateRandomString()
|
||
if err != nil {
|
||
t.Fatalf("生成随机字符串失败: %v", err)
|
||
}
|
||
|
||
if str1 == "" || str2 == "" {
|
||
t.Error("随机字符串为空")
|
||
}
|
||
|
||
if str1 == str2 {
|
||
t.Error("两次生成的随机字符串应该不同")
|
||
}
|
||
|
||
// 验证长度（16字节 = 32位十六进制字符）
|
||
if len(str1) != 32 || len(str2) != 32 {
|
||
t.Error("随机字符串长度应该是32位")
|
||
}
|
||
}
|
||
|
||
func TestIsJSON(t *testing.T) {
|
||
testCases := []struct {
|
||
input string
|
||
expected bool
|
||
}{
|
||
{"{}", true},
|
||
{"[]", true},
|
||
{"{\"key\": \"value\"}", true},
|
||
{"[1, 2, 3]", true},
|
||
{"invalid json", false},
|
||
{"", false},
|
||
{"{invalid}", false},
|
||
}
|
||
|
||
for _, tc := range testCases {
|
||
result := IsJSON(tc.input)
|
||
if result != tc.expected {
|
||
t.Errorf("输入: %s, 期望: %v, 实际: %v", tc.input, tc.expected, result)
|
||
}
|
||
}
|
||
}
|