whitelist_test.gno

 1package agent_test
 2
 3import (
 4	"testing"
 5
 6	"gno.land/p/demo/gnorkle/agent"
 7	"gno.land/p/demo/uassert"
 8)
 9
10func TestWhitelist(t *testing.T) {
11	var whitelist agent.Whitelist
12
13	uassert.False(t, whitelist.HasDefinition(), "whitelist should not be defined initially")
14
15	whitelist.AddAddresses([]string{"a", "b"})
16	uassert.True(t, whitelist.HasAddress("a"), `whitelist should have address "a"`)
17	uassert.True(t, whitelist.HasAddress("b"), `whitelist should have address "b"`)
18	uassert.True(t, whitelist.HasDefinition(), "whitelist should be defined after adding addresses")
19
20	whitelist.RemoveAddress("a")
21	uassert.False(t, whitelist.HasAddress("a"), `whitelist should not have address "a"`)
22	uassert.True(t, whitelist.HasAddress("b"), `whitelist should still have address "b"`)
23
24	whitelist.ClearAddresses()
25	uassert.False(t, whitelist.HasAddress("a"), `whitelist cleared; should not have address "a"`)
26	uassert.False(t, whitelist.HasAddress("b"), `whitelist cleared; should still have address "b"`)
27	uassert.False(t, whitelist.HasDefinition(), "whitelist cleared; should not be defined")
28}