chore: fix systests

spacemeshos · Jun 12, 2024 · db361d1 · db361d1
1 parent a2253a2
commit db361d1
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 8 deletions.
diff --git a/activation/activation_test.go b/activation/activation_test.go
@@ -73,7 +73,7 @@ func newTestBuilder(tb testing.TB, numSigners int, opts ...BuilderOption) *testA
 		func(core zapcore.Core) zapcore.Core {
 			return zapcore.NewTee(core, observer)
 		},
-	)), zaptest.Level(zap.WarnLevel))
+	)))
 
 	ctrl := gomock.NewController(tb)
 	tab := &testAtxBuilder{

diff --git a/systest/tests/distributed_post_verification_test.go b/systest/tests/distributed_post_verification_test.go
@@ -159,8 +159,9 @@ func TestPostMalfeasanceProof(t *testing.T) {
 	localDb := localsql.InMemory()
 	certClient := activation.NewCertifierClient(db, localDb, logger.Named("certifier"))
 	certifier := activation.NewCertifier(localDb, logger, certClient)
+	poetDb := activation.NewPoetDb(db, zap.NewNop())
 	poetClient, err := activation.NewPoetClient(
-		activation.NewPoetDb(db, zap.NewNop()),
+		poetDb,
 		types.PoetServer{
 			Address: cluster.MakePoetGlobalEndpoint(ctx.Namespace, 0),
 		}, cfg.POET,
@@ -169,12 +170,26 @@ func TestPostMalfeasanceProof(t *testing.T) {
 	)
 	require.NoError(t, err)
 
+	verifyingOpts := activation.DefaultPostVerifyingOpts()
+	verifyingOpts.Workers = 1
+	verifier, err := activation.NewPostVerifier(cfg.POST, logger, activation.WithVerifyingOpts(verifyingOpts))
+	require.NoError(t, err)
+
+	validator := activation.NewValidator(
+		db,
+		poetDb,
+		cfg.POST,
+		cfg.SMESHING.Opts.Scrypt,
+		verifier,
+	)
+
 	nipostBuilder, err := activation.NewNIPostBuilder(
 		localDb,
 		grpcPostService,
 		logger.Named("nipostBuilder"),
 		cfg.POET,
 		clock,
+		validator,
 		activation.WithPoetClients(poetClient),
 	)
 	require.NoError(t, err)
@@ -216,19 +231,26 @@ func TestPostMalfeasanceProof(t *testing.T) {
 		}
 		break
 	}
-
-	nipost, err := nipostBuilder.BuildNIPost(ctx, signer, challenge.PublishEpoch, challenge.Hash())
+	nipostChallenge := &types.NIPostChallenge{
+		PublishEpoch:   challenge.PublishEpoch,
+		PrevATXID:      types.EmptyATXID,
+		PositioningATX: challenge.PositioningATXID,
+		CommitmentATX:  challenge.CommitmentATXID,
+		InitialPost: &types.Post{
+			Nonce:   challenge.InitialPost.Nonce,
+			Indices: challenge.InitialPost.Indices,
+			Pow:     challenge.InitialPost.Pow,
+		}}
+
+	nipost, err := nipostBuilder.BuildNIPost(ctx, signer, challenge.Hash(), nipostChallenge)
 	require.NoError(t, err)
 
 	// 2.2 Create ATX with invalid POST
 	for i := range nipost.Post.Indices {
 		nipost.Post.Indices[i] += 1
 	}
+
 	// Sanity check that the POST is invalid
-	verifyingOpts := activation.DefaultPostVerifyingOpts()
-	verifyingOpts.Workers = 1
-	verifier, err := activation.NewPostVerifier(cfg.POST, logger, activation.WithVerifyingOpts(verifyingOpts))
-	require.NoError(t, err)
 	err = verifier.Verify(ctx, (*shared.Proof)(nipost.Post), &shared.ProofMetadata{
 		NodeId:          signer.NodeID().Bytes(),
 		CommitmentAtxId: challenge.CommitmentATXID.Bytes(),