forked from submariner-io/submariner
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move TCP connectivity test functions to new module
Moved RunConnectivityTest, RunNoConnectivityTest, et al from dataplane to a new "tcp" module/package. This separates the actual connectivity code from the dataplane Ginko specs so one can use the connectivity test code from other specs w/o also causing the dataplane specs to run. Signed-off-by: Tom Pantelis <tompantelis@gmail.com>
- Loading branch information
Showing
4 changed files
with
109 additions
and
102 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,92 @@ | ||
| package tcp | ||
|
|
||
| import ( | ||
| "fmt" | ||
|
|
||
| "github.com/submariner-io/submariner/test/e2e/framework" | ||
|
|
||
| . "github.com/onsi/ginkgo" | ||
| . "github.com/onsi/gomega" | ||
| ) | ||
|
|
||
| func RunConnectivityTest(f *framework.Framework, useService bool, networkType bool, listenerScheduling framework.NetworkPodScheduling, connectorScheduling framework.NetworkPodScheduling, listenerCluster framework.ClusterIndex, connectorCluster framework.ClusterIndex) (*framework.NetworkPod, *framework.NetworkPod) { | ||
|
|
||
| listenerPod, connectorPod := createPods(f, useService, networkType, listenerScheduling, connectorScheduling, listenerCluster, connectorCluster, | ||
| framework.TestContext.ConnectionTimeout, framework.TestContext.ConnectionAttempts) | ||
| listenerPod.CheckSuccessfulFinish() | ||
| connectorPod.CheckSuccessfulFinish() | ||
|
|
||
| By("Verifying that the listener got the connector's data and the connector got the listener's data") | ||
| Expect(listenerPod.TerminationMessage).To(ContainSubstring(connectorPod.Config.Data)) | ||
| Expect(connectorPod.TerminationMessage).To(ContainSubstring(listenerPod.Config.Data)) | ||
|
|
||
| // Normally, Submariner preserves the sourceIP for inter-cluster traffic. However, when a POD is using the | ||
| // HostNetwork, it does not get an IPAddress from the podCIDR and it uses the HostIP. Submariner, for such PODs, | ||
| // would MASQUERADE the external traffic from that POD to the corresponding CNI interface ip-address on that | ||
| // Host. So, we skip source-ip validation for PODs using HostNetworking. | ||
| if networkType == framework.PodNetworking { | ||
| By("Verifying the output of listener pod which must contain the source IP") | ||
| Expect(listenerPod.TerminationMessage).To(ContainSubstring(connectorPod.Pod.Status.PodIP)) | ||
| } | ||
|
|
||
| // Return the pods in case further verification is needed | ||
| return listenerPod, connectorPod | ||
| } | ||
|
|
||
| func RunNoConnectivityTest(f *framework.Framework, useService bool, listenerScheduling framework.NetworkPodScheduling, connectorScheduling framework.NetworkPodScheduling, listenerCluster framework.ClusterIndex, connectorCluster framework.ClusterIndex) (*framework.NetworkPod, *framework.NetworkPod) { | ||
| listenerPod, connectorPod := createPods(f, useService, framework.PodNetworking, listenerScheduling, connectorScheduling, listenerCluster, connectorCluster, 5, 1) | ||
|
|
||
| By("Verifying that listener pod exits with non-zero code and timed out message") | ||
| Expect(listenerPod.TerminationMessage).To(ContainSubstring("nc: timeout")) | ||
| Expect(listenerPod.TerminationCode).To(Equal(int32(1))) | ||
|
|
||
| By("Verifying that connector pod exists with zero code but times out") | ||
| Expect(connectorPod.TerminationMessage).To(ContainSubstring("Connection timed out")) | ||
| Expect(connectorPod.TerminationCode).To(Equal(int32(0))) | ||
|
|
||
| // Return the pods in case further verification is needed | ||
| return listenerPod, connectorPod | ||
| } | ||
|
|
||
| func createPods(f *framework.Framework, useService bool, networkType bool, listenerScheduling framework.NetworkPodScheduling, connectorScheduling framework.NetworkPodScheduling, listenerCluster framework.ClusterIndex, | ||
| connectorCluster framework.ClusterIndex, connectionTimeout uint, connectionAttempts uint) (*framework.NetworkPod, *framework.NetworkPod) { | ||
|
|
||
| By(fmt.Sprintf("Creating a listener pod in cluster %q, which will wait for a handshake over TCP", framework.TestContext.ClusterIDs[listenerCluster])) | ||
| listenerPod := f.NewNetworkPod(&framework.NetworkPodConfig{ | ||
| Type: framework.ListenerPod, | ||
| Cluster: listenerCluster, | ||
| Scheduling: listenerScheduling, | ||
| ConnectionTimeout: connectionTimeout, | ||
| ConnectionAttempts: connectionAttempts, | ||
| }) | ||
|
|
||
| remoteIP := listenerPod.Pod.Status.PodIP | ||
| if useService { | ||
| By(fmt.Sprintf("Pointing a service ClusterIP to the listener pod in cluster %q", framework.TestContext.ClusterIDs[listenerCluster])) | ||
| service := listenerPod.CreateService() | ||
| remoteIP = service.Spec.ClusterIP | ||
| } | ||
|
|
||
| framework.Logf("Will send traffic to IP: %v", remoteIP) | ||
|
|
||
| By(fmt.Sprintf("Creating a connector pod in cluster %q, which will attempt the specific UUID handshake over TCP", framework.TestContext.ClusterIDs[connectorCluster])) | ||
| connectorPod := f.NewNetworkPod(&framework.NetworkPodConfig{ | ||
| Type: framework.ConnectorPod, | ||
| Cluster: connectorCluster, | ||
| Scheduling: connectorScheduling, | ||
| RemoteIP: remoteIP, | ||
| ConnectionTimeout: connectionTimeout, | ||
| ConnectionAttempts: connectionAttempts, | ||
| NetworkType: networkType, | ||
| }) | ||
|
|
||
| By(fmt.Sprintf("Waiting for the listener pod %q to exit, returning what listener sent", listenerPod.Pod.Name)) | ||
| listenerPod.AwaitFinish() | ||
|
|
||
| By(fmt.Sprintf("Waiting for the connector pod %q to exit, returning what connector sent", connectorPod.Pod.Name)) | ||
| connectorPod.AwaitFinish() | ||
|
|
||
| framework.Logf("Connector pod has IP: %s", connectorPod.Pod.Status.PodIP) | ||
|
|
||
| return listenerPod, connectorPod | ||
| } |