chore: 并入 iotlcm-upstream-chan 历史到 upstream-chan/（原提交 1e74497）
This commit is contained in:
MiFaZhan
@@ -0,0 +1,159 @@
|
||||
package mingqian
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"git.5iot.com/iotlcm-public/iotlcm-upstream-chan/internal/model"
|
||||
)
|
||||
|
||||
func TestPackageFlowFailuresPreserveMonthlyUsage(t *testing.T) {
|
||||
cases := []struct {
|
||||
name string
|
||||
body string
|
||||
}{
|
||||
{"business failure", `{"code":500,"msg":"private upstream detail","data":[]}`},
|
||||
{"missing card", `{"code":200,"data":[]}`},
|
||||
{"duplicate card", `{"code":200,"data":[{"iccid":"A","totalpackage":10,"targetpackageusage":2},{"iccid":"A","totalpackage":10,"targetpackageusage":2}]}`},
|
||||
{"missing total", `{"code":200,"data":[{"iccid":"A","targetpackageusage":2}]}`},
|
||||
{"missing usage", `{"code":200,"data":[{"iccid":"A","totalpackage":10}]}`},
|
||||
{"null total", `{"code":200,"data":[{"iccid":"A","totalpackage":null,"targetpackageusage":2}]}`},
|
||||
{"negative total", `{"code":200,"data":[{"iccid":"A","totalpackage":-1,"targetpackageusage":2}]}`},
|
||||
{"negative usage", `{"code":200,"data":[{"iccid":"A","totalpackage":10,"targetpackageusage":-1}]}`},
|
||||
{"nonfinite total", `{"code":200,"data":[{"iccid":"A","totalpackage":"NaN","targetpackageusage":2}]}`},
|
||||
{"invalid usage", `{"code":200,"data":[{"iccid":"A","totalpackage":10,"targetpackageusage":""}]}`},
|
||||
}
|
||||
for _, test := range cases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
adapter := testAdapter(t, func(writer http.ResponseWriter, request *http.Request) {
|
||||
if request.URL.Path == usagePath {
|
||||
fmt.Fprint(writer, `{"code":200,"data":[{"iccid":"A","use":1024}]}`)
|
||||
return
|
||||
}
|
||||
fmt.Fprint(writer, test.body)
|
||||
})
|
||||
output, err := adapter.QuerySimDataUsage(context.Background(), &model.SimDataUsageInput{SimKeys: []*model.SimKey{{ICCID: "A"}}})
|
||||
if err != nil || len(output) != 1 {
|
||||
t.Fatalf("output=%+v err=%v", output, err)
|
||||
}
|
||||
item := output[0]
|
||||
if item.Code != 0 || item.ResData.DataUsage != 1 || item.ResData.PackageFlow != nil || item.ResData.PackageFlowError == "" || item.Message == "" {
|
||||
t.Fatalf("must retain monthly usage with an explicit package failure: %+v", item)
|
||||
}
|
||||
if strings.Contains(item.ResData.PackageFlowError, "private upstream detail") {
|
||||
t.Fatal("raw upstream errors must not leak")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestPackageFlowNeverSubstitutesForMonthlyUsage(t *testing.T) {
|
||||
adapter := testAdapter(t, func(writer http.ResponseWriter, request *http.Request) {
|
||||
if request.URL.Path == usagePath {
|
||||
writer.WriteHeader(http.StatusBadGateway)
|
||||
return
|
||||
}
|
||||
fmt.Fprint(writer, `{"code":200,"data":[{"iccid":"A","totalpackage":10,"targetpackageusage":8}]}`)
|
||||
})
|
||||
output, err := adapter.QuerySimDataUsage(context.Background(), &model.SimDataUsageInput{SimKeys: []*model.SimKey{{ICCID: "A"}}})
|
||||
if err != nil || len(output) != 1 || output[0].Code == 0 || !strings.Contains(output[0].Message, "502") {
|
||||
t.Fatalf("monthly failure must remain a failure: %+v, %v", output, err)
|
||||
}
|
||||
if output[0].ResData.DataUsage != 0 || output[0].ResData.PackageFlow == nil || output[0].ResData.PackageFlow.DataUsage != 8 {
|
||||
t.Fatal("package usage must not become monthly usage")
|
||||
}
|
||||
}
|
||||
|
||||
func TestPackageFlowPerCardMatchingAndZero(t *testing.T) {
|
||||
adapter := testAdapter(t, func(writer http.ResponseWriter, request *http.Request) {
|
||||
if request.URL.Path == usagePath {
|
||||
fmt.Fprint(writer, `{"code":200,"data":[{"iccid":"A","use":0},{"msisdn":"144000","use":512},{"iccid":"missing-package","use":2048}]}`)
|
||||
return
|
||||
}
|
||||
fmt.Fprint(writer, `{"code":200,"data":[{"msisdn":"144000","totalpackage":"1.5","targetpackageusage":"2.25"},{"iccid":"A","totalpackage":0,"targetpackageusage":0}]}`)
|
||||
})
|
||||
keys := []*model.SimKey{{ICCID: " A "}, {MSISDN: "144000"}, {ICCID: "missing-package"}, {ICCID: "A"}}
|
||||
output, err := adapter.QuerySimDataUsage(context.Background(), &model.SimDataUsageInput{SimKeys: keys})
|
||||
if err != nil || len(output) != len(keys) {
|
||||
t.Fatalf("output=%+v err=%v", output, err)
|
||||
}
|
||||
for index, item := range output {
|
||||
if item.Code != 0 || item.SimKey != keys[index] {
|
||||
t.Fatalf("identity or monthly status changed: %+v", item)
|
||||
}
|
||||
}
|
||||
if output[0].ResData.PackageFlow == nil || *output[0].ResData.PackageFlow != (model.SimPackageFlowItem{}) || output[3].ResData.PackageFlow == nil {
|
||||
t.Fatal("zero flow is valid and duplicate request identifiers must retain their results")
|
||||
}
|
||||
flow := output[1].ResData.PackageFlow
|
||||
if flow == nil || flow.TotalFlow != 1.5 || flow.DataUsage != 2.25 || flow.RemainingFlow != 0 || output[1].ResData.DataUsage != 0.5 {
|
||||
t.Fatalf("incorrect overage or units: %+v", flow)
|
||||
}
|
||||
if output[2].ResData.PackageFlow != nil || output[2].ResData.PackageFlowError == "" {
|
||||
t.Fatal("missing package card must not be reported as zero")
|
||||
}
|
||||
}
|
||||
|
||||
func TestFlowBothEndpointsBatchAtFifty(t *testing.T) {
|
||||
calls := map[string]int{}
|
||||
adapter := testAdapter(t, func(writer http.ResponseWriter, request *http.Request) {
|
||||
calls[request.URL.Path]++
|
||||
var payload struct {
|
||||
Data string `json:"data"`
|
||||
}
|
||||
var data struct {
|
||||
Numbers []string `json:"numberArray"`
|
||||
}
|
||||
if err := json.NewDecoder(request.Body).Decode(&payload); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if err := json.Unmarshal([]byte(payload.Data), &data); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if len(data.Numbers) == 0 || len(data.Numbers) > 50 {
|
||||
t.Errorf("wrong batch size: %d", len(data.Numbers))
|
||||
}
|
||||
rows := []map[string]any{}
|
||||
for _, identifier := range data.Numbers {
|
||||
rows = append(rows, map[string]any{"iccid": identifier, "use": 1024, "totalpackage": 10, "targetpackageusage": 8})
|
||||
}
|
||||
json.NewEncoder(writer).Encode(map[string]any{"code": 200, "data": rows})
|
||||
})
|
||||
keys := make([]*model.SimKey, 51)
|
||||
for index := range keys {
|
||||
keys[index] = &model.SimKey{ICCID: strconv.Itoa(index)}
|
||||
}
|
||||
output, err := adapter.QuerySimDataUsage(context.Background(), &model.SimDataUsageInput{SimKeys: keys})
|
||||
if err != nil || len(output) != 51 || calls[usagePath] != 2 || calls[packageUsagePath] != 2 {
|
||||
t.Fatalf("calls=%v count=%d err=%v", calls, len(output), err)
|
||||
}
|
||||
for _, item := range output {
|
||||
if item.Code != 0 || item.ResData.DataUsage != 1 || item.ResData.PackageFlow == nil || item.ResData.PackageFlow.RemainingFlow != 2 {
|
||||
t.Fatalf("incomplete mapping: %+v", item)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestFlowEmptyInputAndUnsupportedQueries(t *testing.T) {
|
||||
calls := 0
|
||||
adapter := testAdapter(t, func(writer http.ResponseWriter, request *http.Request) { calls++ })
|
||||
if _, err := adapter.QuerySimDataUsage(context.Background(), nil); err == nil {
|
||||
t.Fatal("nil input must fail")
|
||||
}
|
||||
output, err := adapter.QuerySimDataUsage(context.Background(), &model.SimDataUsageInput{})
|
||||
if err != nil || len(output) != 0 {
|
||||
t.Fatalf("empty input: %+v %v", output, err)
|
||||
}
|
||||
keys := []*model.SimKey{{ICCID: "A"}}
|
||||
session, _ := adapter.QuerySimSession(context.Background(), &model.SimSessionInput{SimKeys: keys})
|
||||
monthly, _ := adapter.QuerySimDataUsageMonthly(context.Background(), &model.SimDataUsageMonthlyInput{SimKeys: keys})
|
||||
daily, _ := adapter.QuerySimDataUsageDaily(context.Background(), &model.SimDataUsageDailyInput{SimKeys: keys})
|
||||
if session[0].Code == 0 || monthly[0].Code == 0 || daily[0].Code == 0 || calls != 0 {
|
||||
t.Fatal("unsupported queries must not invent data or contact upstream")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user