From 8d3221900ddae874f9fc408a89e3b8935c29bcde Mon Sep 17 00:00:00 2001
From: georgehao <haohongfan@gmail.com>
Date: Sat, 20 Apr 2024 22:16:51 +0800
Subject: [PATCH] feat: test

---
 .../logic/provertask/chunk_prover_task.go     | 10 ++---
 coordinator/test/api_test.go                  | 38 +++++++++----------
 2 files changed, 24 insertions(+), 24 deletions(-)

diff --git a/coordinator/internal/logic/provertask/chunk_prover_task.go b/coordinator/internal/logic/provertask/chunk_prover_task.go
index fe079153f..e0124ef1f 100644
--- a/coordinator/internal/logic/provertask/chunk_prover_task.go
+++ b/coordinator/internal/logic/provertask/chunk_prover_task.go
@@ -140,11 +140,11 @@ func (cp *ChunkProverTask) Assign(ctx *gin.Context, getTaskParameter *coordinato
 		AssignedAt: utils.NowUTC(),
 	}
 
-	if err = cp.proverTaskOrm.InsertProverTask(ctx, &proverTask); err != nil {
-		cp.recoverActiveAttempts(ctx, chunkTask)
-		log.Error("insert chunk prover task fail", "taskID", chunkTask.Hash, "publicKey", taskCtx.PublicKey, "err", err)
-		return nil, ErrCoordinatorInternalFailure
-	}
+	//if err = cp.proverTaskOrm.InsertProverTask(ctx, &proverTask); err != nil {
+	//	cp.recoverActiveAttempts(ctx, chunkTask)
+	//	log.Error("insert chunk prover task fail", "taskID", chunkTask.Hash, "publicKey", taskCtx.PublicKey, "err", err)
+	//	return nil, ErrCoordinatorInternalFailure
+	//}
 
 	taskMsg, err := cp.formatProverTask(ctx, &proverTask)
 	if err != nil {
diff --git a/coordinator/test/api_test.go b/coordinator/test/api_test.go
index 24aa9702a..b766219ff 100644
--- a/coordinator/test/api_test.go
+++ b/coordinator/test/api_test.go
@@ -564,25 +564,25 @@ func testValidProof(t *testing.T) {
 	err = chunkOrm.UpdateBatchHashInRange(context.Background(), 0, 0, batch.Hash)
 	assert.NoError(t, err)
 
-	// create mock provers.
-	//provers := make([]*mockProver, 1)
-	//for i := 0; i < len(provers); i++ {
-	//	var proofType message.ProofType
-	//	if i%2 == 0 {
-	//		proofType = message.ProofTypeChunk
-	//	} else {
-	//		proofType = message.ProofTypeBatch
-	//	}
-	//
-	//	provers[i] = newMockProver(t, "prover_test"+strconv.Itoa(i), coordinatorURL, proofType, version.Version)
-	//
-	//	proofStatus := verifiedSuccess
-	//	proverTask, errCode, errMsg := provers[i].getProverTask(t, proofType, "istanbul")
-	//	assert.Equal(t, errCode, types.Success)
-	//	assert.Equal(t, errMsg, "")
-	//	assert.NotNil(t, proverTask)
-	//	provers[i].submitProof(t, proverTask, proofStatus, types.Success, "istanbul")
-	//}
+	//create mock provers.
+	provers := make([]*mockProver, 1)
+	for i := 0; i < len(provers); i++ {
+		var proofType message.ProofType
+		if i%2 == 0 {
+			proofType = message.ProofTypeChunk
+		} else {
+			proofType = message.ProofTypeBatch
+		}
+
+		provers[i] = newMockProver(t, "prover_test"+strconv.Itoa(i), coordinatorURL, proofType, version.Version)
+
+		proofStatus := verifiedSuccess
+		proverTask, errCode, errMsg := provers[i].getProverTask(t, proofType, "istanbul")
+		assert.Equal(t, errCode, types.Success)
+		assert.Equal(t, errMsg, "")
+		assert.NotNil(t, proverTask)
+		provers[i].submitProof(t, proverTask, proofStatus, types.Success, "istanbul")
+	}
 
 	//// verify proof status
 	//var (