status-im/go-libp2p-pubsub
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update tests for quadratic p4

Browse Source
This commit is contained in:
vyzo 2020-05-08 20:07:05 +03:00
parent 1c85190323
commit 2dc46af415
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gossipsub_spam_test.go
View File

@ -669,7 +669,7 @@ func TestGossipsubAttackInvalidMessageSpam(t *testing.T) {
time.Sleep(100*time.Millisecond + GossipSubHeartbeatInitialDelay) time.Sleep(100*time.Millisecond + GossipSubHeartbeatInitialDelay)
// The attackers score should now have fallen below zero // The attackers score should now have fallen below zero
if attackerScore() > 0 { if attackerScore() >= 0 {
t.Fatalf("Expected attacker score to be less than zero but it's %f", attackerScore()) t.Fatalf("Expected attacker score to be less than zero but it's %f", attackerScore())
} }
// There should be several rejected messages (because the signature was invalid) // There should be several rejected messages (because the signature was invalid)

15
score_test.go
View File

@ -1,6 +1,7 @@
package pubsub package pubsub
import ( import (
"math"
"sync" "sync"
"testing" "testing"
"time" "time"
@ -457,7 +458,7 @@ func TestScoreInvalidMessageDeliveries(t *testing.T) {
topicScoreParams := &TopicScoreParams{ topicScoreParams := &TopicScoreParams{
TopicWeight: 1, TopicWeight: 1,
TimeInMeshQuantum: time.Second, TimeInMeshQuantum: time.Second,
InvalidMessageDeliveriesWeight: 1, InvalidMessageDeliveriesWeight: -1,
InvalidMessageDeliveriesDecay: 1.0, InvalidMessageDeliveriesDecay: 1.0,
} }
params.Topics[mytopic] = topicScoreParams params.Topics[mytopic] = topicScoreParams
@ -478,7 +479,7 @@ func TestScoreInvalidMessageDeliveries(t *testing.T) {
ps.refreshScores() ps.refreshScores()
aScore := ps.Score(peerA) aScore := ps.Score(peerA)
expected := topicScoreParams.TopicWeight * topicScoreParams.InvalidMessageDeliveriesWeight * float64(nMessages) expected := topicScoreParams.TopicWeight * topicScoreParams.InvalidMessageDeliveriesWeight * float64(nMessages*nMessages)
if aScore != expected { if aScore != expected {
t.Fatalf("Score: %f. Expected %f", aScore, expected) t.Fatalf("Score: %f. Expected %f", aScore, expected)
} }
@ -494,7 +495,7 @@ func TestScoreInvalidMessageDeliveriesDecay(t *testing.T) {
topicScoreParams := &TopicScoreParams{ topicScoreParams := &TopicScoreParams{
TopicWeight: 1, TopicWeight: 1,
TimeInMeshQuantum: time.Second, TimeInMeshQuantum: time.Second,
InvalidMessageDeliveriesWeight: 1, InvalidMessageDeliveriesWeight: -1,
InvalidMessageDeliveriesDecay: 0.9, InvalidMessageDeliveriesDecay: 0.9,
} }
params.Topics[mytopic] = topicScoreParams params.Topics[mytopic] = topicScoreParams
@ -515,7 +516,7 @@ func TestScoreInvalidMessageDeliveriesDecay(t *testing.T) {
ps.refreshScores() ps.refreshScores()
aScore := ps.Score(peerA) aScore := ps.Score(peerA)
expected := topicScoreParams.TopicWeight * topicScoreParams.InvalidMessageDeliveriesWeight * topicScoreParams.InvalidMessageDeliveriesDecay * float64(nMessages) expected := topicScoreParams.TopicWeight * topicScoreParams.InvalidMessageDeliveriesWeight * math.Pow(topicScoreParams.InvalidMessageDeliveriesDecay*float64(nMessages), 2)
if aScore != expected { if aScore != expected {
t.Fatalf("Score: %f. Expected %f", aScore, expected) t.Fatalf("Score: %f. Expected %f", aScore, expected)
} }
@ -523,7 +524,7 @@ func TestScoreInvalidMessageDeliveriesDecay(t *testing.T) {
// refresh scores a few times to apply decay // refresh scores a few times to apply decay
for i := 0; i < 10; i++ { for i := 0; i < 10; i++ {
ps.refreshScores() ps.refreshScores()
expected *= topicScoreParams.InvalidMessageDeliveriesDecay expected *= math.Pow(topicScoreParams.InvalidMessageDeliveriesDecay, 2)
} }
aScore = ps.Score(peerA) aScore = ps.Score(peerA)
if aScore != expected { if aScore != expected {
@ -651,13 +652,13 @@ func TestScoreRejectMessageDeliveries(t *testing.T) {
ps.RejectMessage(&msg, rejectValidationFailed) ps.RejectMessage(&msg, rejectValidationFailed)
aScore = ps.Score(peerA) aScore = ps.Score(peerA)
expected = -2.0 expected = -4.0
if aScore != expected { if aScore != expected {
t.Fatalf("Score: %f. Expected %f", aScore, expected) t.Fatalf("Score: %f. Expected %f", aScore, expected)
} }
bScore = ps.Score(peerB) bScore = ps.Score(peerB)
expected = -2.0 expected = -4.0
if bScore != expected { if bScore != expected {
t.Fatalf("Score: %f. Expected %f", bScore, expected) t.Fatalf("Score: %f. Expected %f", bScore, expected)
} }