mirror of https://github.com/poanetwork/gecko.git
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package snowball
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
|
|
"github.com/ava-labs/gecko/ids"
|
|
)
|
|
|
|
// ByzantineFactory implements Factory by returning a byzantine struct
|
|
type ByzantineFactory struct{}
|
|
|
|
// New implements Factory
|
|
func (ByzantineFactory) New() Consensus { return &Byzantine{} }
|
|
|
|
// Byzantine is a naive implementation of a multi-choice snowball instance
|
|
type Byzantine struct {
|
|
// params contains all the configurations of a snowball instance
|
|
params Parameters
|
|
|
|
// Hardcode the preference
|
|
preference ids.ID
|
|
}
|
|
|
|
// Initialize implements the Consensus interface
|
|
func (b *Byzantine) Initialize(params Parameters, choice ids.ID) {
|
|
b.params = params
|
|
b.preference = choice
|
|
}
|
|
|
|
// Parameters implements the Consensus interface
|
|
func (b *Byzantine) Parameters() Parameters { return b.params }
|
|
|
|
// Add implements the Consensus interface
|
|
func (b *Byzantine) Add(choice ids.ID) {}
|
|
|
|
// Preference implements the Consensus interface
|
|
func (b *Byzantine) Preference() ids.ID { return b.preference }
|
|
|
|
// RecordPoll implements the Consensus interface
|
|
func (b *Byzantine) RecordPoll(votes ids.Bag) {}
|
|
|
|
// RecordUnsuccessfulPoll implements the Consensus interface
|
|
func (b *Byzantine) RecordUnsuccessfulPoll() {}
|
|
|
|
// Finalized implements the Consensus interface
|
|
func (b *Byzantine) Finalized() bool { return true }
|
|
func (b *Byzantine) String() string { return b.preference.String() }
|
|
|
|
var (
|
|
Red = ids.Empty.Prefix(0)
|
|
Blue = ids.Empty.Prefix(1)
|
|
Green = ids.Empty.Prefix(2)
|
|
)
|
|
|
|
func ParamsTest(t *testing.T, factory Factory) {
|
|
sb := factory.New()
|
|
|
|
params := Parameters{
|
|
Metrics: prometheus.NewRegistry(),
|
|
K: 2, Alpha: 2, BetaVirtuous: 1, BetaRogue: 2, ConcurrentRepolls: 1,
|
|
}
|
|
sb.Initialize(params, Red)
|
|
|
|
if p := sb.Parameters(); p.K != params.K {
|
|
t.Fatalf("Wrong K parameter")
|
|
} else if p.Alpha != params.Alpha {
|
|
t.Fatalf("Wrong Alpha parameter")
|
|
} else if p.BetaVirtuous != params.BetaVirtuous {
|
|
t.Fatalf("Wrong Beta1 parameter")
|
|
} else if p.BetaRogue != params.BetaRogue {
|
|
t.Fatalf("Wrong Beta2 parameter")
|
|
} else if p.ConcurrentRepolls != params.ConcurrentRepolls {
|
|
t.Fatalf("Wrong Repoll parameter")
|
|
}
|
|
}
|