| 86 | } |
| 87 | |
| 88 | func TestIPList_Save(t *testing.T) { |
| 89 | var a = assert.NewAssertion(t) |
| 90 | |
| 91 | var list = waf.NewIPList(waf.IPListTypeAllow) |
| 92 | list.Add(waf.IPTypeAll, firewallconfigs.FirewallScopeGlobal, 1, "192.168.1.100", time.Now().Unix()+3600) |
| 93 | list.Add(waf.IPTypeAll, firewallconfigs.FirewallScopeGlobal, 2, "192.168.1.101", time.Now().Unix()+3600) |
| 94 | |
| 95 | var file = Tea.Root + "/data/waf.iplist.json" |
| 96 | err := list.Save(file) |
| 97 | if err != nil { |
| 98 | t.Fatal(err) |
| 99 | } |
| 100 | |
| 101 | var newList = waf.NewIPList(waf.IPListTypeAllow) |
| 102 | err = newList.Load(file) |
| 103 | if err != nil { |
| 104 | t.Fatal(err) |
| 105 | } |
| 106 | |
| 107 | a.IsTrue(newList.Contains(waf.IPTypeAll, firewallconfigs.FirewallScopeGlobal, 1, "192.168.1.100")) |
| 108 | |
| 109 | _ = os.Remove(file) |
| 110 | } |
| 111 | |
| 112 | func BenchmarkIPList_Add(b *testing.B) { |
| 113 | runtime.GOMAXPROCS(1) |