From 5b9e6dc1dd54b8bc0b07527a188c5c82a8cb617d Mon Sep 17 00:00:00 2001
From: Matt Boersma <Matt.Boersma@microsoft.com>
Date: Wed, 1 Nov 2023 13:56:14 -0600
Subject: [PATCH] Add self-hosted + ClusterClass test specs from CAPI

---
 test/e2e/capi_test.go | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/test/e2e/capi_test.go b/test/e2e/capi_test.go
index f54df8059db6..5554e5d031d4 100644
--- a/test/e2e/capi_test.go
+++ b/test/e2e/capi_test.go
@@ -147,6 +147,42 @@ var _ = Describe("Running the Cluster API E2E tests", func() {
 				}
 			})
 		})
+
+		Context("Running the self-hosted spec with ClusterClass", func() {
+			SelfHostedSpec(context.TODO(), func() SelfHostedSpecInput {
+				return SelfHostedSpecInput{
+					E2EConfig:                e2eConfig,
+					ClusterctlConfigPath:     clusterctlConfigPath,
+					BootstrapClusterProxy:    bootstrapClusterProxy,
+					ArtifactFolder:           artifactFolder,
+					SkipCleanup:              skipCleanup,
+					Flavor:                   "topology",
+					ControlPlaneMachineCount: ptr.To[int64](1),
+					WorkerMachineCount:       ptr.To[int64](1),
+					ControlPlaneWaiters: clusterctl.ControlPlaneWaiters{
+						WaitForControlPlaneInitialized: EnsureControlPlaneInitialized,
+					},
+				}
+			})
+		})
+
+		Context("Running the self-hosted spec with ClusterClass and an HA control plane", func() {
+			SelfHostedSpec(context.TODO(), func() SelfHostedSpecInput {
+				return SelfHostedSpecInput{
+					E2EConfig:                e2eConfig,
+					ClusterctlConfigPath:     clusterctlConfigPath,
+					BootstrapClusterProxy:    bootstrapClusterProxy,
+					ArtifactFolder:           artifactFolder,
+					SkipCleanup:              skipCleanup,
+					Flavor:                   "topology",
+					ControlPlaneMachineCount: ptr.To[int64](3),
+					WorkerMachineCount:       ptr.To[int64](1),
+					ControlPlaneWaiters: clusterctl.ControlPlaneWaiters{
+						WaitForControlPlaneInitialized: EnsureControlPlaneInitialized,
+					},
+				}
+			})
+		})
 	}
 
 	// TODO: Add test using KCPRemediationSpec