Skip to content

Commit

Permalink
check: add third client scheduled on a different node
Browse files Browse the repository at this point in the history
for some upcoming tests we will need a new client scheduled on a
different node than the first 2 clients. This commit adds this new
client.

Signed-off-by: Gilberto Bertin <[email protected]>
  • Loading branch information
jibi committed Dec 14, 2023
1 parent 301179b commit e27002a
Showing 1 changed file with 49 additions and 2 deletions.
51 changes: 49 additions & 2 deletions connectivity/check/deployment.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ const (

clientDeploymentName = "client"
client2DeploymentName = "client2"
client3DeploymentName = "client3"
clientCPDeployment = "client-cp"

DNSTestServerContainerName = "dns-test-server"
Expand Down Expand Up @@ -854,7 +855,48 @@ func (ct *ConnectivityTest) deploy(ctx context.Context) error {
}
}

// 3rd client scheduled on the control plane
if !ct.params.SingleNode {
// 3rd client scheduled on a different node than the first 2 clients
_, err = ct.clients.src.GetDeployment(ctx, ct.params.TestNamespace, client3DeploymentName, metav1.GetOptions{})
if err != nil {
ct.Logf("✨ [%s] Deploying %s deployment...", ct.clients.src.ClusterName(), client3DeploymentName)
clientDeployment := newDeployment(deploymentParameters{
Name: client3DeploymentName,
Kind: kindClientName,
NamedPort: "http-8080",
Port: 8080,
Image: ct.params.CurlImage,
Command: []string{"/bin/ash", "-c", "sleep 10000000"},
Labels: map[string]string{"other": "client-other-node"},
Annotations: ct.params.DeploymentAnnotations.Match(client3DeploymentName),
Affinity: &corev1.Affinity{
PodAntiAffinity: &corev1.PodAntiAffinity{
RequiredDuringSchedulingIgnoredDuringExecution: []corev1.PodAffinityTerm{
{
LabelSelector: &metav1.LabelSelector{
MatchExpressions: []metav1.LabelSelectorRequirement{
{Key: "name", Operator: metav1.LabelSelectorOpIn, Values: []string{clientDeploymentName}},
},
},
TopologyKey: corev1.LabelHostname,
},
},
},
},
NodeSelector: ct.params.NodeSelector,
})
_, err = ct.clients.src.CreateServiceAccount(ctx, ct.params.TestNamespace, k8s.NewServiceAccount(client3DeploymentName), metav1.CreateOptions{})
if err != nil {
return fmt.Errorf("unable to create service account %s: %s", client3DeploymentName, err)
}
_, err = ct.clients.src.CreateDeployment(ctx, ct.params.TestNamespace, clientDeployment, metav1.CreateOptions{})
if err != nil {
return fmt.Errorf("unable to create deployment %s: %s", client3DeploymentName, err)
}
}
}

// 4th client scheduled on the control plane
if ct.params.K8sLocalHostTest {
ct.Logf("✨ [%s] Deploying %s deployment...", ct.clients.src.ClusterName(), clientCPDeployment)
clientDeployment := newDeployment(deploymentParameters{
Expand Down Expand Up @@ -884,7 +926,7 @@ func (ct *ConnectivityTest) deploy(ctx context.Context) error {
}
}

if !ct.params.SingleNode || ct.params.MultiCluster != "" {
if !ct.params.SingleNode {
_, err = ct.clients.dst.GetService(ctx, ct.params.TestNamespace, echoOtherNodeDeploymentName, metav1.GetOptions{})
if err != nil {
ct.Logf("✨ [%s] Deploying echo-other-node service...", ct.clients.dst.ClusterName())
Expand Down Expand Up @@ -1063,6 +1105,9 @@ func (ct *ConnectivityTest) deploy(ctx context.Context) error {
func (ct *ConnectivityTest) deploymentList() (srcList []string, dstList []string) {
if !ct.params.Perf {
srcList = []string{clientDeploymentName, client2DeploymentName, echoSameNodeDeploymentName}
if !ct.params.SingleNode {
srcList = append(srcList, client3DeploymentName)
}
} else {
perfNm := newPerfDeploymentNameManager(&ct.params)
srcList = []string{perfNm.ClientName(), perfNm.ServerName()}
Expand Down Expand Up @@ -1099,10 +1144,12 @@ func (ct *ConnectivityTest) deleteDeployments(ctx context.Context, client *k8s.C
_ = client.DeleteDeployment(ctx, ct.params.TestNamespace, echoOtherNodeDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteDeployment(ctx, ct.params.TestNamespace, clientDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteDeployment(ctx, ct.params.TestNamespace, client2DeploymentName, metav1.DeleteOptions{})
_ = client.DeleteDeployment(ctx, ct.params.TestNamespace, client3DeploymentName, metav1.DeleteOptions{})
_ = client.DeleteServiceAccount(ctx, ct.params.TestNamespace, echoSameNodeDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteServiceAccount(ctx, ct.params.TestNamespace, echoOtherNodeDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteServiceAccount(ctx, ct.params.TestNamespace, clientDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteServiceAccount(ctx, ct.params.TestNamespace, client2DeploymentName, metav1.DeleteOptions{})
_ = client.DeleteServiceAccount(ctx, ct.params.TestNamespace, client3DeploymentName, metav1.DeleteOptions{})
_ = client.DeleteService(ctx, ct.params.TestNamespace, echoSameNodeDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteService(ctx, ct.params.TestNamespace, echoOtherNodeDeploymentName, metav1.DeleteOptions{})
_ = client.DeleteConfigMap(ctx, ct.params.TestNamespace, corednsConfigMapName, metav1.DeleteOptions{})
Expand Down

0 comments on commit e27002a

Please sign in to comment.