feat: test

This commit is contained in:
georgehao
2024-04-20 22:16:51 +08:00
parent 59ba4b90d7
commit 8d3221900d
2 changed files with 24 additions and 24 deletions

View File

@@ -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 {

View File

@@ -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 (