Fix: fixed some random stuff

iotaledger · Apr 10, 2022 · d84c7fd · d84c7fd
1 parent f8f8194
commit d84c7fd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 225 deletions.
diff --git a/packages/consensus/finality/finality_gadget.go b/packages/consensus/finality/finality_gadget.go
@@ -10,23 +10,23 @@ import (
 	"github.com/iotaledger/hive.go/types"
 
 	"github.com/iotaledger/goshimmer/packages/consensus/gof"
-	"github.com/iotaledger/goshimmer/packages/ledgerstate"
+	"github.com/iotaledger/goshimmer/packages/ledger/branchdag"
 	"github.com/iotaledger/goshimmer/packages/markers"
 	"github.com/iotaledger/goshimmer/packages/tangle"
 )
 
 // Gadget is an interface that describes the finality gadget.
 type Gadget interface {
 	HandleMarker(marker *markers.Marker, aw float64) (err error)
-	HandleBranch(branchID ledgerstate.BranchID, aw float64) (err error)
+	HandleBranch(branchID branchdag.BranchID, aw float64) (err error)
 	tangle.ConfirmationOracle
 }
 
 // MessageThresholdTranslation is a function which translates approval weight to a gof.GradeOfFinality.
 type MessageThresholdTranslation func(aw float64) gof.GradeOfFinality
 
 // BranchThresholdTranslation is a function which translates approval weight to a gof.GradeOfFinality.
-type BranchThresholdTranslation func(branchID ledgerstate.BranchID, aw float64) gof.GradeOfFinality
+type BranchThresholdTranslation func(branchID branchdag.BranchID, aw float64) gof.GradeOfFinality
 
 const (
 	lowLowerBound    = 0.25
@@ -36,7 +36,7 @@ const (
 
 var (
 	// DefaultBranchGoFTranslation is the default function to translate the approval weight to gof.GradeOfFinality of a branch.
-	DefaultBranchGoFTranslation BranchThresholdTranslation = func(branchID ledgerstate.BranchID, aw float64) gof.GradeOfFinality {
+	DefaultBranchGoFTranslation BranchThresholdTranslation = func(branchID branchdag.BranchID, aw float64) gof.GradeOfFinality {
 		switch {
 		case aw >= lowLowerBound && aw < mediumLowerBound:
 			return gof.Low

diff --git a/packages/ledgerstate/branch_dag_test.go b/packages/ledgerstate/branch_dag_test.go