sync: 对齐小程序 v1.0.2 生产代码并脱敏

2026-10-29 15:15:12 +08:00
parent a7abd476f0
commit 5fcf74deb0
423 changed files with 15001 additions and 6382 deletions
@@ -0,0 +1,132 @@
+package app
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"git.5iot.com/iotlcm-public/iotlcm-app/internal/model/entity"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcfg"
+)
+
+func TestAppNormalizeUpstreamSessionSupportsTelecomFields(t *testing.T) {
+	session := &appOneLinkMockSimSessionItem{}
+	raw := json.RawMessage(`{"onlineStatus":"0","accessType":"5G","ipv4Address":"192.0.2.10","ipv6Address":"2001:db8::1","apn":"CMIOT","startTime":"2026-10-08 10:00:00"}`)
+	appNormalizeUpstreamSession(session, raw)
+	if session.Status != "0" || session.Rat != "5G" || session.Ipv4 != "192.0.2.10" || session.Ipv6 != "2001:db8::1" || session.ApnId != "CMIOT" {
+		t.Fatalf("normalized session = %+v", session)
+	}
+}
+
+func TestAppDiagnosisUpstreamUsesProxy(t *testing.T) {
+	proxy := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
+		if request.Method != http.MethodPost || request.URL.Path != "/sim/query/sim-basic-info" {
+			t.Errorf("request = %s %s", request.Method, request.URL.Path)
+		}
+		var input appUpstreamSimQueryInput
+		if err := json.NewDecoder(request.Body).Decode(&input); err != nil {
+			t.Errorf("decode proxy request: %v", err)
+		} else if len(input.SimKeys) != 1 || input.SimKeys[0].ChanKey != "test-channel" || input.SimKeys[0].ICCID != "test-iccid" || input.SimKeys[0].IMSI != "test-imsi" || input.SimKeys[0].MSISDN != "test-msisdn" {
+			t.Errorf("proxy sim key = %+v", input.SimKeys)
+		}
+		writer.Header().Set("Content-Type", "application/json")
+		_, _ = writer.Write([]byte(`{"code":0,"data":[{"code":0,"resData":{"simStatus":"0"}}]}`))
+	}))
+	defer proxy.Close()
+
+	config, err := gcfg.NewAdapterContent(fmt.Sprintf(`{"upstream":{"default":{"clientName":%q,"timeout":"2s"},"gateway":{"clientName":"http://127.0.0.1:1"}}}`, proxy.URL))
+	if err != nil {
+		t.Fatalf("configure upstream: %v", err)
+	}
+	original := g.Cfg().GetAdapter()
+	g.Cfg().SetAdapter(config)
+	t.Cleanup(func() { g.Cfg().SetAdapter(original) })
+
+	items, err := queryAppUpstreamSimWithKey(context.Background(), &appUpstreamSimKey{
+		ChanKey: "test-channel", ICCID: "test-iccid", IMSI: "test-imsi", MSISDN: "test-msisdn",
+	}, "/sim/query/sim-basic-info", nil)
+	if err != nil {
+		t.Fatalf("query upstream through proxy: %v", err)
+	}
+	if len(items) != 1 || items[0].Code != 0 {
+		t.Fatalf("proxy response = %+v", items)
+	}
+}
+
+func TestAppDiagnosisUsesDedicatedProxyRoute(t *testing.T) {
+	proxy := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
+		if request.URL.Path != "/internal/app-diagnosis/sim/query/sim-basic-info" {
+			t.Errorf("unexpected path: %s", request.URL.Path)
+		}
+		if request.Header.Get("X-App-Diagnosis-Token") != "test-internal-token" {
+			t.Errorf("missing internal token")
+		}
+		writer.Header().Set("Content-Type", "application/json")
+		_, _ = writer.Write([]byte(`{"code":0,"data":[{"code":0,"resData":{"simStatus":"0"}}]}`))
+	}))
+	defer proxy.Close()
+	adapter, err := gcfg.NewAdapterContent(fmt.Sprintf(`{"upstream":{"default":{"clientName":%q,"timeout":"2s"},"appDiagnosisToken":"test-internal-token"}}`, proxy.URL))
+	if err != nil {
+		t.Fatal(err)
+	}
+	previous := g.Cfg().GetAdapter()
+	g.Cfg().SetAdapter(adapter)
+	t.Cleanup(func() { g.Cfg().SetAdapter(previous) })
+	ctx := context.WithValue(context.Background(), appDiagnosisProxyContextKey{}, true)
+	items, err := queryAppUpstreamSimWithKey(ctx, &appUpstreamSimKey{ChanKey: "test-channel", ICCID: "test-iccid"}, "/sim/query/sim-basic-info", nil)
+	if err != nil || len(items) != 1 || items[0].Code != 0 {
+		t.Fatalf("dedicated proxy request: items=%+v err=%v", items, err)
+	}
+}
+
+func TestAppDiagnosisDoesNotFallbackToMockSession(t *testing.T) {
+	proxy := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
+		writer.Header().Set("Content-Type", "application/json")
+		_, _ = writer.Write([]byte(`{"code":500,"message":"upstream unavailable"}`))
+	}))
+	defer proxy.Close()
+	adapter, err := gcfg.NewAdapterContent(fmt.Sprintf(`{"upstream":{"default":{"clientName":%q,"timeout":"2s"},"appDiagnosisToken":"test-token","onelinkMock":{"enabled":true}}}`, proxy.URL))
+	if err != nil {
+		t.Fatal(err)
+	}
+	previous := g.Cfg().GetAdapter()
+	g.Cfg().SetAdapter(adapter)
+	t.Cleanup(func() { g.Cfg().SetAdapter(previous) })
+	ctx := context.WithValue(context.Background(), appDiagnosisProxyContextKey{}, true)
+	session, source, err := queryAppCardRealtimeSessionWithKey(ctx, &appCardQueryRow{}, &appUpstreamSimKey{ChanKey: "test-channel", ICCID: "test-iccid"})
+	if err == nil || session != nil || source != "upstream" {
+		t.Fatalf("session = %+v, source = %q, err = %v; want upstream failure without mock fallback", session, source, err)
+	}
+}
+
+func TestAppDiagnosisTreatsMissingProxyChannelAsNotIncluded(t *testing.T) {
+	proxy := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
+		writer.WriteHeader(http.StatusNotFound)
+		_, _ = writer.Write([]byte(`Not Found{"code":404,"message":"诊断通道不可用"}`))
+	}))
+	defer proxy.Close()
+	adapter, err := gcfg.NewAdapterContent(fmt.Sprintf(`{"upstream":{"default":{"clientName":%q,"timeout":"2s"},"appDiagnosisToken":"test-internal-token"}}`, proxy.URL))
+	if err != nil {
+		t.Fatal(err)
+	}
+	previous := g.Cfg().GetAdapter()
+	g.Cfg().SetAdapter(adapter)
+	t.Cleanup(func() { g.Cfg().SetAdapter(previous) })
+	ctx := context.WithValue(context.Background(), appDiagnosisProxyContextKey{}, true)
+	_, err = queryAppUpstreamSimWithKey(ctx, &appUpstreamSimKey{ChanKey: "missing-channel", ICCID: "test-iccid"}, "/sim/query/sim-basic-info", nil)
+	if err == nil || err.Error() != appDiagnosisNotIncludedMessage {
+		t.Fatalf("missing proxy channel error = %v", err)
+	}
+	row := &appCardQueryRow{TSimCardInfo: entity.TSimCardInfo{
+		Iccid: "test-iccid", Imsi: "test-imsi", Msisdn: "test-msisdn", ChanKey: "missing-channel",
+	}}
+	_, err = (&sAppDiagnosis{}).prepareDiagnosisSimKey(ctx, row)
+	if err == nil || err.Error() != appDiagnosisNotIncludedMessage {
+		t.Fatalf("legacy diagnosis preflight error = %v", err)
+	}
+}