memberset_test.gno

 1package commondao_test
 2
 3import (
 4	"testing"
 5
 6	"gno.land/p/nt/uassert"
 7
 8	"gno.land/p/devx000/wip/nt/commondao"
 9)
10
11func TestMemberSetSize(t *testing.T) {
12	storage := commondao.NewMemberStorage()
13	members := commondao.NewMemberSet(storage)
14	uassert.Equal(t, 0, members.Size(), "expect size 0")
15
16	storage.Add("g1jg8mtutu9khhfwc4nxmuhcpftf0pajdhfvsqf5")
17	members = commondao.NewMemberSet(storage)
18	uassert.Equal(t, 1, members.Size(), "expect size 1")
19
20	storage.Add("g1w4ek2u33ta047h6lta047h6lta047h6ldvdwpn")
21	members = commondao.NewMemberSet(storage)
22	uassert.Equal(t, 2, members.Size(), "expect size 2")
23}
24
25func TestMemberSetHas(t *testing.T) {
26	storage := commondao.NewMemberStorage()
27	storage.Add("g1jg8mtutu9khhfwc4nxmuhcpftf0pajdhfvsqf5")
28
29	members := commondao.NewMemberSet(storage)
30
31	uassert.True(t, members.Has("g1jg8mtutu9khhfwc4nxmuhcpftf0pajdhfvsqf5"), "expect member is found")
32	uassert.False(t, members.Has("g1w4ek2u33ta047h6lta047h6lta047h6ldvdwpn"), "expect member is not found")
33}
34
35func TestMemberSetIterateByOffset(t *testing.T) {
36	storage := commondao.NewMemberStorage()
37	storage.Add("g1jg8mtutu9khhfwc4nxmuhcpftf0pajdhfvsqf5")
38	storage.Add("g1w4ek2u33ta047h6lta047h6lta047h6ldvdwpn")
39	storage.Add("g1w4ek2u3jta047h6lta047h6lta047h6l9huexc")
40
41	commondao.NewMemberSet(storage).IterateByOffset(1, 1, func(addr address) bool {
42		uassert.Equal(t, "g1w4ek2u33ta047h6lta047h6lta047h6ldvdwpn", addr.String(), "expect address to match")
43		return true
44	})
45}