From 0aee0a97aaddd2b055dcb312088e4cd6c2e03bc7 Mon Sep 17 00:00:00 2001 From: Henrik Hautakoski Date: Mon, 15 Jul 2024 22:57:05 +0200 Subject: [PATCH] internal/types/blacklist.go: adding Empty() --- internal/types/blacklist.go | 4 ++++ internal/types/blacklist_test.go | 10 ++++++++++ 2 files changed, 14 insertions(+) diff --git a/internal/types/blacklist.go b/internal/types/blacklist.go index 2b0812f..12ee19d 100644 --- a/internal/types/blacklist.go +++ b/internal/types/blacklist.go @@ -2,6 +2,10 @@ package types type Blacklist map[string][]string +func (bl Blacklist) Empty() bool { + return len(bl) < 1 +} + func (bl Blacklist) Add(contract string, action string) { if len(bl[contract]) < 1 { bl[contract] = []string{} diff --git a/internal/types/blacklist_test.go b/internal/types/blacklist_test.go index f156e78..1cbcbfe 100644 --- a/internal/types/blacklist_test.go +++ b/internal/types/blacklist_test.go @@ -6,6 +6,16 @@ import ( "github.com/stretchr/testify/require" ) +func TestBlacklist_Empty(t *testing.T) { + bl := Blacklist{} + + require.True(t, bl.Empty()) + + bl.Add("contract", "action1") + + require.False(t, bl.Empty()) +} + func TestBlacklist_Add(t *testing.T) { bl := Blacklist{} bl.Add("contract", "action1")