add ExpectAllEgress

jayunit100 · Feb 19, 2020 · ce2494f · ce2494f
1 parent 77c02e2
commit ce2494f
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/netpol/pkg/utils/reachability.go b/netpol/pkg/utils/reachability.go
@@ -177,7 +177,15 @@ func (r *Reachability) Expect(pod1 Pod, pod2 Pod, isConnected bool) {
 func (r *Reachability) ExpectAllIngress(pod Pod, connected bool) {
 	r.Expected.SetAllTo(string(pod), connected)
 	if !connected {
-		log.Infof("Blacklisting all traffic to %s", pod)
+		log.Infof("Blacklisting all traffic *to* %s", pod)
+	}
+}
+
+// ExpectAllEgress defines that any traffic going out of the pod will be allowed/denied (true/false)
+func (r *Reachability)ExpectAllEgress(pod Pod, connected bool) {
+	r.Expected.SetAllFrom(string(pod), connected)
+	if !connected {
+		log.Infof("Blacklisting all traffic *from* %s", pod)
 	}
 }