chore: 并入 iotlcm-admin-svr 历史到 admin-svr/（原提交 49818c6）

2026-11-02 13:27:59 +08:00
parent 976980a3f5 49818c6dbe
commit 193f7393c0
1250 changed files with 154111 additions and 0 deletions
@@ -0,0 +1,83 @@
+package system
+
+import (
+	"context"
+	"io"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
+	"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,
+			}, "xxx@example.com", "123456")
+			if (err != nil) != (scenario != "success") || time.Since(started) > time.Second {
+				t.Fatalf("unexpected sender result: %v", err)
+			}
+		})
+	}
+}