package integration import ( "bytes" "context" "encoding/json" "fmt" "mime/multipart" "testing" "time" "github.com/break/junhong_cmp_fiber/internal/model" pkggorm "github.com/break/junhong_cmp_fiber/pkg/gorm" "github.com/break/junhong_cmp_fiber/pkg/response" "github.com/break/junhong_cmp_fiber/tests/testutils/integ" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestIotCard_ListStandalone(t *testing.T) { env := integ.NewIntegrationTestEnv(t) cards := []*model.IotCard{ {ICCID: "TEST0012345678901001", CardType: "data_card", CarrierID: 1, Status: 1}, {ICCID: "TEST0012345678901002", CardType: "data_card", CarrierID: 1, Status: 1}, {ICCID: "TEST0012345678901003", CardType: "data_card", CarrierID: 2, Status: 2}, } for _, card := range cards { require.NoError(t, env.TX.Create(card).Error) } t.Run("获取单卡列表-无过滤", func(t *testing.T) { resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/iot-cards/standalone?page=1&page_size=20", nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) }) t.Run("获取单卡列表-按运营商过滤", func(t *testing.T) { resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/iot-cards/standalone?carrier_id=1", nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) }) t.Run("获取单卡列表-按ICCID模糊查询", func(t *testing.T) { resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/iot-cards/standalone?iccid=901001", nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) }) t.Run("未认证请求应返回错误", func(t *testing.T) { resp, err := env.ClearAuth().Request("GET", "/api/admin/iot-cards/standalone", nil) require.NoError(t, err) defer resp.Body.Close() var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.NotEqual(t, 0, result.Code, "未认证请求应返回错误码") }) } func TestIotCard_Import(t *testing.T) { t.Skip("E2E测试:需要 Worker 服务运行处理异步导入任务") env := integ.NewIntegrationTestEnv(t) t.Run("导入CSV文件", func(t *testing.T) { body := &bytes.Buffer{} writer := multipart.NewWriter(body) part, err := writer.CreateFormFile("file", "test.csv") require.NoError(t, err) csvContent := "iccid\nTEST0012345678902001\nTEST0012345678902002\nTEST0012345678902003" _, err = part.Write([]byte(csvContent)) require.NoError(t, err) _ = writer.WriteField("carrier_id", "1") _ = writer.WriteField("carrier_type", "CMCC") _ = writer.WriteField("batch_no", "TEST_BATCH_001") writer.Close() resp, err := env.AsSuperAdmin().RequestWithHeaders("POST", "/api/admin/iot-cards/import", body.Bytes(), map[string]string{ "Content-Type": writer.FormDataContentType(), }) require.NoError(t, err) defer resp.Body.Close() var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) t.Logf("Import response: code=%d, message=%s", result.Code, result.Message) assert.Equal(t, 200, resp.StatusCode) assert.Equal(t, 0, result.Code) }) t.Run("导入无文件应返回错误", func(t *testing.T) { body := &bytes.Buffer{} writer := multipart.NewWriter(body) _ = writer.WriteField("carrier_id", "1") _ = writer.WriteField("carrier_type", "CMCC") writer.Close() resp, err := env.AsSuperAdmin().RequestWithHeaders("POST", "/api/admin/iot-cards/import", body.Bytes(), map[string]string{ "Content-Type": writer.FormDataContentType(), }) require.NoError(t, err) defer resp.Body.Close() var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) t.Logf("No file response: code=%d, message=%s, data=%v", result.Code, result.Message, result.Data) assert.NotEqual(t, 0, result.Code, "无文件时应返回错误码") }) } func TestIotCard_ImportTaskList(t *testing.T) { env := integ.NewIntegrationTestEnv(t) task := &model.IotCardImportTask{ TaskNo: "TEST20260123001", Status: model.ImportTaskStatusCompleted, CarrierID: 1, CarrierType: "CMCC", CarrierName: "中国移动", TotalCount: 100, } require.NoError(t, env.TX.Create(task).Error) t.Run("获取导入任务列表", func(t *testing.T) { resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/iot-cards/import-tasks?page=1&page_size=20", nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) }) t.Run("获取导入任务详情-应包含冗余字段", func(t *testing.T) { url := fmt.Sprintf("/api/admin/iot-cards/import-tasks/%d", task.ID) resp, err := env.AsSuperAdmin().Request("GET", url, nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) dataMap := result.Data.(map[string]interface{}) assert.Equal(t, "CMCC", dataMap["carrier_type"], "任务详情应返回冗余的运营商类型") assert.Equal(t, "中国移动", dataMap["carrier_name"], "任务详情应返回冗余的运营商名称") }) } func TestIotCard_ImportE2E(t *testing.T) { t.Skip("E2E测试:需要 Worker 服务运行处理异步导入任务") env := integ.NewIntegrationTestEnv(t) // 准备测试用的 ICCID(20位,满足 CMCC 要求) testICCIDPrefix := "E2ETEST" testBatchNo1 := fmt.Sprintf("E2E_BATCH_%d_001", time.Now().UnixNano()) testBatchNo2 := fmt.Sprintf("E2E_BATCH_%d_002", time.Now().UnixNano()) testICCIDs := []string{ testICCIDPrefix + "1234567890123", testICCIDPrefix + "1234567890124", testICCIDPrefix + "1234567890125", } t.Run("完整导入流程验证", func(t *testing.T) { // Step 1: 通过 API 提交导入任务 body := &bytes.Buffer{} writer := multipart.NewWriter(body) part, err := writer.CreateFormFile("file", "e2e_test.csv") require.NoError(t, err) csvContent := "iccid\n" + testICCIDs[0] + "\n" + testICCIDs[1] + "\n" + testICCIDs[2] _, err = part.Write([]byte(csvContent)) require.NoError(t, err) _ = writer.WriteField("carrier_id", "1") _ = writer.WriteField("carrier_type", "CMCC") _ = writer.WriteField("batch_no", testBatchNo1) writer.Close() resp, err := env.AsSuperAdmin().RequestWithHeaders("POST", "/api/admin/iot-cards/import", body.Bytes(), map[string]string{ "Content-Type": writer.FormDataContentType(), }) require.NoError(t, err) defer resp.Body.Close() var apiResult response.Response err = json.NewDecoder(resp.Body).Decode(&apiResult) require.NoError(t, err) require.Equal(t, 0, apiResult.Code, "API 应返回成功: %s", apiResult.Message) // 从响应中提取 task_id dataMap, ok := apiResult.Data.(map[string]interface{}) require.True(t, ok, "响应数据应为 map") taskIDFloat, ok := dataMap["task_id"].(float64) require.True(t, ok, "task_id 应存在") taskID := uint(taskIDFloat) t.Logf("创建的导入任务 ID: %d", taskID) // Step 2: 等待 Worker 处理完成(轮询检查任务状态) var importTask model.IotCardImportTask maxWaitTime := 30 * time.Second pollInterval := 500 * time.Millisecond startTime := time.Now() ctx := context.Background() skipCtx := pkggorm.SkipDataPermission(ctx) for { if time.Since(startTime) > maxWaitTime { t.Fatalf("等待超时:任务 %d 未在 %v 内完成", taskID, maxWaitTime) } err = env.RawDB().WithContext(skipCtx).First(&importTask, taskID).Error require.NoError(t, err) t.Logf("任务状态: %d (1=pending, 2=processing, 3=completed, 4=failed)", importTask.Status) if importTask.Status == model.ImportTaskStatusCompleted || importTask.Status == model.ImportTaskStatusFailed { break } time.Sleep(pollInterval) } // Step 3: 验证任务完成状态 assert.Equal(t, model.ImportTaskStatusCompleted, importTask.Status, "任务应完成") assert.Equal(t, 3, importTask.TotalCount, "总数应为3") assert.Equal(t, 3, importTask.SuccessCount, "成功数应为3") assert.Equal(t, 0, importTask.SkipCount, "跳过数应为0") assert.Equal(t, 0, importTask.FailCount, "失败数应为0") t.Logf("任务完成: total=%d, success=%d, skip=%d, fail=%d", importTask.TotalCount, importTask.SuccessCount, importTask.SkipCount, importTask.FailCount) // Step 4: 验证 IoT 卡已入库 var cards []model.IotCard err = env.RawDB().WithContext(skipCtx).Where("iccid IN ?", testICCIDs).Find(&cards).Error require.NoError(t, err) assert.Len(t, cards, 3, "应创建3张 IoT 卡") for _, card := range cards { assert.Equal(t, uint(1), card.CarrierID, "运营商ID应为1") assert.Equal(t, testBatchNo1, card.BatchNo, "批次号应匹配") assert.Equal(t, 1, card.Status, "状态应为在库(1)") t.Logf("已创建 IoT 卡: ICCID=%s, ID=%d", card.ICCID, card.ID) } }) t.Run("重复导入应跳过已存在的ICCID", func(t *testing.T) { // 再次导入相同的 ICCID,应该全部跳过 body := &bytes.Buffer{} writer := multipart.NewWriter(body) part, err := writer.CreateFormFile("file", "e2e_test_dup.csv") require.NoError(t, err) csvContent := "iccid\n" + testICCIDs[0] + "\n" + testICCIDs[1] _, err = part.Write([]byte(csvContent)) require.NoError(t, err) _ = writer.WriteField("carrier_id", "1") _ = writer.WriteField("carrier_type", "CMCC") _ = writer.WriteField("batch_no", testBatchNo2) writer.Close() resp, err := env.AsSuperAdmin().RequestWithHeaders("POST", "/api/admin/iot-cards/import", body.Bytes(), map[string]string{ "Content-Type": writer.FormDataContentType(), }) require.NoError(t, err) defer resp.Body.Close() var apiResult response.Response err = json.NewDecoder(resp.Body).Decode(&apiResult) require.NoError(t, err) require.Equal(t, 0, apiResult.Code) dataMap := apiResult.Data.(map[string]interface{}) taskID := uint(dataMap["task_id"].(float64)) // 等待处理完成 var importTask model.IotCardImportTask maxWaitTime := 30 * time.Second startTime := time.Now() ctx := context.Background() skipCtx := pkggorm.SkipDataPermission(ctx) for { if time.Since(startTime) > maxWaitTime { t.Fatalf("等待超时") } env.RawDB().WithContext(skipCtx).First(&importTask, taskID) if importTask.Status == model.ImportTaskStatusCompleted || importTask.Status == model.ImportTaskStatusFailed { break } time.Sleep(500 * time.Millisecond) } // 验证:2条应该全部跳过 assert.Equal(t, model.ImportTaskStatusCompleted, importTask.Status) assert.Equal(t, 2, importTask.TotalCount) assert.Equal(t, 0, importTask.SuccessCount, "成功数应为0(全部跳过)") assert.Equal(t, 2, importTask.SkipCount, "跳过数应为2") t.Logf("重复导入结果: success=%d, skip=%d", importTask.SuccessCount, importTask.SkipCount) }) } func TestIotCard_CarrierRedundantFields(t *testing.T) { env := integ.NewIntegrationTestEnv(t) carrierCode := fmt.Sprintf("REDUND_%d", time.Now().UnixNano()) carrier := &model.Carrier{ CarrierCode: carrierCode, CarrierName: "冗余字段测试运营商", CarrierType: "CUCC", Status: 1, } require.NoError(t, env.TX.Create(carrier).Error) testICCID := fmt.Sprintf("8986%016d", time.Now().UnixNano()%10000000000000000) card := &model.IotCard{ ICCID: testICCID, CarrierID: carrier.ID, CarrierType: carrier.CarrierType, CarrierName: carrier.CarrierName, CardType: "data_card", Status: 1, } require.NoError(t, env.TX.Create(card).Error) t.Run("单卡列表应返回冗余字段", func(t *testing.T) { resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/iot-cards/standalone?iccid="+testICCID, nil) require.NoError(t, err) defer resp.Body.Close() require.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) require.Equal(t, 0, result.Code, "API应返回成功,实际: %v", result.Message) dataMap, ok := result.Data.(map[string]interface{}) require.True(t, ok, "Data应为map类型,实际: %T", result.Data) items, ok := dataMap["items"].([]interface{}) require.True(t, ok, "items字段应存在且为数组,dataMap: %+v", dataMap) require.GreaterOrEqual(t, len(items), 1, "列表应至少有1条记录,ICCID: %s", testICCID) cardData := items[0].(map[string]interface{}) assert.Equal(t, "CUCC", cardData["carrier_type"], "列表应返回冗余的运营商类型") assert.Equal(t, "冗余字段测试运营商", cardData["carrier_name"], "列表应返回冗余的运营商名称") }) t.Run("单卡详情应返回冗余字段", func(t *testing.T) { url := fmt.Sprintf("/api/admin/iot-cards/by-iccid/%s", card.ICCID) resp, err := env.AsSuperAdmin().Request("GET", url, nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) dataMap := result.Data.(map[string]interface{}) assert.Equal(t, "CUCC", dataMap["carrier_type"], "详情应返回冗余的运营商类型") assert.Equal(t, "冗余字段测试运营商", dataMap["carrier_name"], "详情应返回冗余的运营商名称") }) } func TestIotCard_GetByICCID(t *testing.T) { env := integ.NewIntegrationTestEnv(t) carrierCode := fmt.Sprintf("ICCID_%d", time.Now().UnixNano()) carrier := &model.Carrier{ CarrierCode: carrierCode, CarrierName: "测试运营商", CarrierType: "CMCC", Status: 1, } require.NoError(t, env.TX.Create(carrier).Error) testICCID := fmt.Sprintf("8986%016d", time.Now().UnixNano()%10000000000000000) card := &model.IotCard{ ICCID: testICCID, CarrierID: carrier.ID, CarrierType: carrier.CarrierType, CarrierName: carrier.CarrierName, MSISDN: "13800000001", CardType: "physical", CardCategory: "normal", CostPrice: 1000, DistributePrice: 1500, Status: 1, } require.NoError(t, env.TX.Create(card).Error) t.Run("通过ICCID查询单卡详情-成功", func(t *testing.T) { url := fmt.Sprintf("/api/admin/iot-cards/by-iccid/%s", card.ICCID) resp, err := env.AsSuperAdmin().Request("GET", url, nil) require.NoError(t, err) defer resp.Body.Close() assert.Equal(t, 200, resp.StatusCode) var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.Equal(t, 0, result.Code) dataMap, ok := result.Data.(map[string]interface{}) require.True(t, ok) assert.Equal(t, testICCID, dataMap["iccid"]) assert.Equal(t, "13800000001", dataMap["msisdn"]) assert.Equal(t, "CMCC", dataMap["carrier_type"], "应返回冗余的运营商类型") assert.Equal(t, "测试运营商", dataMap["carrier_name"], "应返回冗余的运营商名称") }) t.Run("通过不存在的ICCID查询-应返回错误", func(t *testing.T) { resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/iot-cards/by-iccid/NONEXISTENT_ICCID", nil) require.NoError(t, err) defer resp.Body.Close() var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.NotEqual(t, 0, result.Code, "不存在的ICCID应返回错误码") }) t.Run("未认证请求-应返回错误", func(t *testing.T) { url := fmt.Sprintf("/api/admin/iot-cards/by-iccid/%s", card.ICCID) resp, err := env.ClearAuth().Request("GET", url, nil) require.NoError(t, err) defer resp.Body.Close() var result response.Response err = json.NewDecoder(resp.Body).Decode(&result) require.NoError(t, err) assert.NotEqual(t, 0, result.Code, "未认证请求应返回错误码") }) }