MiFaZhan
59eb34976a
iotlcm-admin-svr：管理后台服务。 - Go module：iotlcm → git.5iot.com/iotlcm-public/iotlcm-admin-svr（729 文件） - 凭据与地址脱敏：JWT signingKey（含代码里的 MustGet 默认值）、 MySQL DSN、Redis 口令、告警邮件发件人邮箱全部改为占位； 私网 IP → 文档地址段；库名 xiujie_admin → iotlcm - 真实手机号/邮箱 → 合法 11 位编造号 / example.com - 移除内部资产：CI 触发器、过时 DDL（sql/db_iotlcm.sql）、 私有依赖 go get 行、明文 registry 登录口令 - 启动 banner 统一 IOTLCM 方块字 + 社区惯例署名 - Dockerfile 改用公共基础镜像 - 拼错的 RADEME.md → README.md，README 重写 - 需要真实数据库的用例改为可用 IOTLCM_TEST_WITH_DB=1 显式启用 - 补 Apache-2.0 LICENSE，LF 行尾 + .gitattributes 数据库 schema 不随本仓发布（生产 DDL 已过时），README 已说明。
84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestQrcodeAPISuccessAndBodyTimeout(t *testing.T) {
|
|
for _, scenario := range []string{"success", "header-timeout", "body-timeout", "partial-body-timeout"} {
|
|
t.Run(scenario, func(t *testing.T) {
|
|
release := make(chan struct{})
|
|
server := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
|
|
_, _ = io.Copy(io.Discard, request.Body)
|
|
if scenario != "success" {
|
|
if scenario != "header-timeout" {
|
|
writer.Header().Set("Content-Length", "1024")
|
|
writer.WriteHeader(http.StatusOK)
|
|
if scenario == "partial-body-timeout" {
|
|
_, _ = writer.Write([]byte(`{"code":0,"data":{"ticket":"partial"}}`))
|
|
}
|
|
writer.(http.Flusher).Flush()
|
|
}
|
|
select {
|
|
case <-request.Context().Done():
|
|
case <-release:
|
|
}
|
|
return
|
|
}
|
|
_, _ = writer.Write([]byte(`{"code":0,"data":{"ticket":"test-ticket","expireTime":300,"qrUrl":"test-url"}}`))
|
|
}))
|
|
defer server.Close()
|
|
defer close(release)
|
|
started := time.Now()
|
|
result, err := callQrcodeApi(context.Background(), &SysQrcodeConfig{
|
|
ServerApi: server.URL, Timeout: "70ms", Token: "test-token", SceneIdLogin: 1,
|
|
}, true)
|
|
if scenario != "success" {
|
|
if err == nil || result != nil || time.Since(started) > time.Second {
|
|
t.Fatalf("stalled response was not rejected: result=%v error=%v", result, err)
|
|
}
|
|
} else if err != nil || result == nil || result.Ticket != "test-ticket" {
|
|
t.Fatalf("successful response changed: result=%v error=%v", result, err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUserCodeHTTPTimeoutAndStatus(t *testing.T) {
|
|
for _, scenario := range []string{"success", "rejected", "timeout"} {
|
|
t.Run(scenario, func(t *testing.T) {
|
|
release := make(chan struct{})
|
|
server := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
|
|
_, _ = io.Copy(io.Discard, request.Body)
|
|
if scenario == "timeout" {
|
|
select {
|
|
case <-request.Context().Done():
|
|
case <-release:
|
|
}
|
|
return
|
|
}
|
|
if scenario == "rejected" {
|
|
writer.WriteHeader(http.StatusServiceUnavailable)
|
|
return
|
|
}
|
|
writer.WriteHeader(http.StatusAccepted)
|
|
}))
|
|
defer server.Close()
|
|
defer close(release)
|
|
started := time.Now()
|
|
err := callUserCodeHTTP(context.Background(), model.UserCodeHttpSenderConfig{
|
|
Url: server.URL, Timeout: "70ms", SuccessStatus: http.StatusAccepted,
|
|
}, "user@example.com", "123456")
|
|
if (err != nil) != (scenario != "success") || time.Since(started) > time.Second {
|
|
t.Fatalf("unexpected sender result: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|