瀏覽代碼

wip:提交

fuwencai 2 年之前
父節點
當前提交
e89f6a1aae
共有 1 個文件被更改,包括 10 次插入9 次删除
  1. 10 9
      service/claim_test.go

+ 10 - 9
service/claim_test.go

@@ -26,7 +26,7 @@ func TestEntClaimService_Claim(t *testing.T) {
 		}, true, false,
 		},
 		{"认领医疗机构-重复认领", args{
-			data: &entity.EntClaim{UserId: -1, AppId: "10000", EntId: "25caa43ce64036dfd1f55635e06394a6", EntName: "汉川市人民医院", Status: entity.StatusClaimed, Type: entity.TypeInstitution, CreateTime: time.Now().Format("2006-01-02 15:04:05")},
+			data: &entity.EntClaim{UserId: -1, AppId: "10000", EntId: "5a8f3bde5e75b16a98715041ba687705", EntName: "汉川市人民医院", Status: entity.StatusClaimed, Type: entity.TypeInstitution, CreateTime: time.Now().Format("2006-01-02 15:04:05")},
 		}, false, true,
 		},
 		{"认领经销商-新增 ", args{
@@ -34,7 +34,7 @@ func TestEntClaimService_Claim(t *testing.T) {
 		}, true, false,
 		},
 		{"认领经销商-重复认领", args{
-			data: &entity.EntClaim{UserId: -1, AppId: "10000", EntId: "d8f48ada3c357d0812df188cc14ed527", EntName: "邢台太行医用材料有限公司", Status: entity.StatusClaimed, Type: entity.TypeDistributor, CreateTime: time.Now().Format("2006-01-02 15:04:05")},
+			data: &entity.EntClaim{UserId: -1, AppId: "10000", EntId: "b4eb554e9e7368c17761d4f722e7cf5e", EntName: "邢台太行医用材料有限公司", Status: entity.StatusClaimed, Type: entity.TypeDistributor, CreateTime: time.Now().Format("2006-01-02 15:04:05")},
 		}, false, true,
 		},
 	}
@@ -248,10 +248,10 @@ func TestEntClaimService_UnclaimedByEnt(t *testing.T) {
 }
 
 // 数据准备
-func setupClaim() {
+func setupClaim(conn *entity.Conn) {
 	fmt.Println("setup ...")
 	// 删除测试用例产生的数据
-	EntClaimSrv.BaseMysql.Delete(entity.TableDomainEntClaim, map[string]interface{}{
+	conn.BaseMysql.Delete(entity.TableDomainEntClaim, map[string]interface{}{
 		"user_id": "-1",
 	})
 	fmt.Println("clean over...")
@@ -281,15 +281,15 @@ func setupClaim() {
 		"10000", -1, "be23b68d98c14847d341f84f6c5da4f3", "洪泽县晨光医疗器械有限公司", entity.StatusClaimed, entity.TypeDistributor, time.Now().Format("2006-01-02 15:04:05"),
 		"10000", -1, "b4eb554e9e7368c17761d4f722e7cf5e", "深圳市华晨阳科技有限公司", entity.StatusClaimed, entity.TypeDistributor, time.Now().Format("2006-01-02 15:04:05"),
 	}
-	EntClaimSrv.BaseMysql.InsertBatch(entity.TableDomainEntClaim, field, values)
+	conn.BaseMysql.InsertBatch(entity.TableDomainEntClaim, field, values)
 	fmt.Println("setup ok...")
 }
 
 // 清理产生的测试数据
-func teardownClaim() {
+func teardownClaim(conn *entity.Conn) {
 	fmt.Println("teardown...")
 	// 删除测试用例产生的数据
-	EntClaimSrv.BaseMysql.Delete(entity.TableDomainEntClaim, map[string]interface{}{
+	conn.BaseMysql.Delete(entity.TableDomainEntClaim, map[string]interface{}{
 		"user_id": "-1",
 	})
 	fmt.Println("teardown  ok...")
@@ -298,9 +298,10 @@ func teardownClaim() {
 // 通过testMain 来执行测试用例  可以做全局的准备和清理工作
 func TestMain(m *testing.M) {
 	// setup code...
-	setupClaim()
+	conn := InitDb() // 显式初始化数据库
+	setupClaim(conn)
 	fmt.Println("start init")
 	code := m.Run()
-	teardownClaim()
+	teardownClaim(conn)
 	os.Exit(code)
 }