chore: add tests for Entry

QuokkaStake · Jun 13, 2024 · 70bb1a1 · 70bb1a1
1 parent 35823be
commit 70bb1a1
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pkg/types/entry_test.go b/pkg/types/entry_test.go
@@ -211,3 +211,16 @@ func TestEntriesSetTotalVotingPower(t *testing.T) {
  assert.InDelta(t, float64(1), entries["firstaddr"].CumulativeVotingPowerPercent, 0.001)
  assert.InDelta(t, 0.75, entries["secondaddr"].CumulativeVotingPowerPercent, 0.001)
 }
+
+func TestEntriesGetByValidatorAddresses(t *testing.T) {
+ t.Parallel()
+
+ entries := Entries{
+ "firstaddr": {Validator: &Validator{OperatorAddress: "firstaddr"}},
+ "secondaddr": {Validator: &Validator{OperatorAddress: "secondaddr"}},
+ "thirdaddr": {Validator: &Validator{OperatorAddress: "thirdaddr"}},
+ }
+
+ filteredEntries := entries.ByValidatorAddresses([]string{"firstaddr", "secondaddr"})
+ assert.Len(t, filteredEntries, 2)
+}