Add: test monitor error

This commit is contained in:
ignatella 2024-11-04 19:54:56 +01:00
parent 7014843161
commit ffa09824d5
1 changed files with 9 additions and 2 deletions

View File

@ -61,6 +61,7 @@ func TestMonitor(t *testing.T) {
var gotRule *nftables.Rule var gotRule *nftables.Rule
wg := sync.WaitGroup{} wg := sync.WaitGroup{}
wg.Add(1) wg.Add(1)
var errMonitor error
go func() { go func() {
defer wg.Done() defer wg.Done()
count := int32(0) count := int32(0)
@ -74,13 +75,13 @@ func TestMonitor(t *testing.T) {
fileName := filepath.Base(os.Args[0]) fileName := filepath.Base(os.Args[0])
if genMsg.ProcComm != fileName { if genMsg.ProcComm != fileName {
err = fmt.Errorf("procComm: %s, want: %s", genMsg.ProcComm, fileName) errMonitor = fmt.Errorf("procComm: %s, want: %s", genMsg.ProcComm, fileName)
return return
} }
for _, change := range event.Changes { for _, change := range event.Changes {
if change.Error != nil { if change.Error != nil {
err = fmt.Errorf("monitor err: %s", change.Error) errMonitor = fmt.Errorf("monitor err: %s", change.Error)
return return
} }
@ -141,7 +142,13 @@ func TestMonitor(t *testing.T) {
if err := c.Flush(); err != nil { if err := c.Flush(); err != nil {
t.Fatal(err) t.Fatal(err)
} }
wg.Wait() wg.Wait()
if errMonitor != nil {
t.Fatal("monitor err", errMonitor)
}
if gotTable.Family != nat.Family || gotTable.Name != nat.Name { if gotTable.Family != nat.Family || gotTable.Name != nat.Name {
t.Fatal("no want table", gotTable.Family, gotTable.Name) t.Fatal("no want table", gotTable.Family, gotTable.Name)
} }