fix failing test.

Author: guettli

controllers/controllers_suite_test.go

@@ -93,11 +93,12 @@ var _ = BeforeSuite(func() {
 	}).SetupWithManager(ctx, testEnv.Manager, controller.Options{})).To(Succeed())
 
 	Expect((&HetznerBareMetalHostReconciler{
-		Client:              testEnv.Manager.GetClient(),
-		APIReader:           testEnv.Manager.GetAPIReader(),
-		RobotClientFactory:  testEnv.RobotClientFactory,
-		SSHClientFactory:    testEnv.BaremetalSSHClientFactory,
-		PreProvisionCommand: "dummy-pre-provision-command",
+		Client:               testEnv.Manager.GetClient(),
+		APIReader:            testEnv.Manager.GetAPIReader(),
+		RobotClientFactory:   testEnv.RobotClientFactory,
+		SSHClientFactory:     testEnv.BaremetalSSHClientFactory,
+		PreProvisionCommand:  "dummy-pre-provision-command",
+		SSHAfterInstallImage: true,
 	}).SetupWithManager(ctx, testEnv.Manager, controller.Options{})).To(Succeed())
 
 	Expect((&HetznerBareMetalMachineReconciler{

controllers/hetznerbaremetalmachine_controller_test.go

@@ -429,6 +429,8 @@ var _ = Describe("HetznerBareMetalMachineReconciler", func() {
 					return isPresentAndTrue(key, bmMachine, infrav1.HostReadyCondition)
 				}, timeout).Should(BeTrue())
 
+				osSSHClient.On("GetHostName").Unset()
+
 				osSSHClient.On("GetHostName").Return(sshclient.Output{
 					StdOut: "some-unexpected-hostname",
 					StdErr: "",