Fix pruning candidate (#1821)

kaspanet · Aug 23, 2021 · ba5880f · ba5880f
1 parent 7b5720a
commit ba5880f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/domain/consensus/datastructures/pruningstore/pruning_store.go b/domain/consensus/datastructures/pruningstore/pruning_store.go
@@ -60,7 +60,7 @@ func (ps *pruningStore) PruningPointCandidate(dbContext model.DBReader, stagingA
 		return ps.pruningPointCandidateCache, nil
 	}
 
-	candidateBytes, err := dbContext.Get(ps.pruningBlockHashKey)
+	candidateBytes, err := dbContext.Get(ps.candidatePruningPointHashKey)
 	if err != nil {
 		return nil, err
 	}

diff --git a/domain/consensus/processes/pruningmanager/pruning_test.go b/domain/consensus/processes/pruningmanager/pruning_test.go
@@ -2,6 +2,7 @@ package pruningmanager_test
 
 import (
 	"encoding/json"
+	"github.com/kaspanet/kaspad/infrastructure/db/database"
 	"os"
 	"path/filepath"
 	"testing"
@@ -107,7 +108,9 @@ func TestPruning(t *testing.T) {
 				blockHashToID[*blockHash] = dagBlock.ID
 
 				pruningPointCandidate, err := tc.PruningStore().PruningPointCandidate(tc.DatabaseContext(), stagingArea)
-				if err != nil {
+				if database.IsNotFoundError(err) {
+					pruningPointCandidate = consensusConfig.GenesisHash
+				} else if err != nil {
 					return err
 				}