updated operator and loader tests

evilsocket · Feb 27, 2021 · 1d277ef · 1d277ef
1 parent 44e8561
commit 1d277ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/daemon/rule/loader_test.go b/daemon/rule/loader_test.go
@@ -23,6 +23,7 @@ func TestRuleLoader(t *testing.T) {
 	var list []Operator
 	dur1s := Duration("1s")
 	dummyOper, _ := NewOperator(Simple, false, OpTrue, "", list)
+	dummyOper.Compile()
 	inMem1sRule := Create("000-xxx-name", true, false, Allow, dur1s, dummyOper)
 	inMemUntilRestartRule := Create("000-aaa-name", true, false, Allow, Restart, dummyOper)
 

diff --git a/daemon/rule/operator_test.go b/daemon/rule/operator_test.go
@@ -139,6 +139,10 @@ func TestNewOperatorSimple(t *testing.T) {
 			t.Error("NewOperator simple.dstHost.sensitive err should be nil: ", err)
 			t.Fail()
 		}
+		if err = opSimple.Compile(); err != nil {
+			t.Error("NewOperator simple.dstHost.sensitive Compile() err:", err)
+			t.Fail()
+		}
 		conn.DstHost = "OpEnsNitCh.io"
 		if opSimple.Match(conn) == false {
 			t.Error("Test NewOperator() simple.dstHost.sensitive doesn't match")