Implement review suggestions, fix tests

2020-10-15 15:25:34 +02:00
parent 5d0db1c250
commit be75c2d99b
4 changed files with 8 additions and 6 deletions
--- a/profile/endpoints/endpoints_test.go
+++ b/profile/endpoints/endpoints_test.go
@@ -17,6 +17,8 @@ func TestMain(m *testing.M) {
 }

 func testEndpointMatch(t *testing.T, ep Endpoint, entity *intel.Entity, expectedResult EPResult) {
+	entity.SetDstPort(entity.Port)
+
 	result, _ := ep.Matches(context.TODO(), entity)
 	if result != expectedResult {
 		t.Errorf(