Skip to content

Commit 7871e22

Browse files
committed
Fixed TestProcessNetwork
1 parent b325c41 commit 7871e22

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

integration-tests/suites/process_network.go

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import (
99
"github.com/stackrox/collector/integration-tests/pkg/types"
1010

1111
sensorAPI "github.com/stackrox/rox/generated/internalapi/sensor"
12+
"github.com/stackrox/rox/generated/storage"
1213
)
1314

1415
type ProcessNetworkTestSuite struct {
@@ -74,7 +75,7 @@ func (s *ProcessNetworkTestSuite) SetupSuite() {
7475
s.serverPort, err = s.getPort("nginx")
7576
s.Require().NoError(err)
7677

77-
_, err = s.execContainer("nginx-curl", []string{"curl", fmt.Sprintf("%s:%s", s.serverIP, s.serverPort)}, false)
78+
_, err = s.execContainer("nginx-curl", []string{"curl", fmt.Sprintf("%s:%d", s.serverIP, s.serverPort)}, false)
7879
s.Require().NoError(err)
7980

8081
s.clientIP, err = s.getIPAddress("nginx-curl")
@@ -151,7 +152,8 @@ func (s *ProcessNetworkTestSuite) TestNetworkFlows() {
151152
s.Sensor().ExpectConnections(s.T(), s.serverContainer, 10*time.Second,
152153
&sensorAPI.NetworkConnection{
153154
LocalAddress: types.CreateNetworkAddress("", "", s.serverPort),
154-
RemoteAddress: types.CreateNetworkAddress(s.clientIP, "", s.serverPort),
155+
RemoteAddress: types.CreateNetworkAddress(s.clientIP, "", 0),
156+
Protocol: storage.L4Protocol_L4_PROTOCOL_TCP,
155157
Role: sensorAPI.ClientServerRole_ROLE_SERVER,
156158
SocketFamily: sensorAPI.SocketFamily_SOCKET_FAMILY_UNKNOWN,
157159
CloseTimestamp: nil,
@@ -161,7 +163,8 @@ func (s *ProcessNetworkTestSuite) TestNetworkFlows() {
161163
s.Sensor().ExpectConnections(s.T(), s.clientContainer, 10*time.Second,
162164
&sensorAPI.NetworkConnection{
163165
LocalAddress: types.CreateNetworkAddress("", "", 0),
164-
RemoteAddress: types.CreateNetworkAddress(s.clientIP, "", s.serverPort),
166+
RemoteAddress: types.CreateNetworkAddress(s.serverIP, "", s.serverPort),
167+
Protocol: storage.L4Protocol_L4_PROTOCOL_TCP,
165168
Role: sensorAPI.ClientServerRole_ROLE_CLIENT,
166169
SocketFamily: sensorAPI.SocketFamily_SOCKET_FAMILY_UNKNOWN,
167170
CloseTimestamp: nil,

0 commit comments

Comments
 (0)