[az] Add base vm tests for set available

Author: Sploder12

tests/test_base_virtual_machine.cpp

@@ -1460,4 +1460,39 @@ TEST_F(BaseVM, sshExecRethrowsSSHExceptionsWhenConnected)
                          mpt::match_what(HasSubstr("intentional")));
 }
 
+TEST_F(BaseVM, setUnavailableShutsdownRunning)
+{
+    vm.start();
+    ASSERT_EQ(vm.current_state(), St::running);
+
+    EXPECT_CALL(vm, shutdown(mp::VirtualMachine::ShutdownPolicy::Poweroff)).Times(1);
+
+    vm.set_available(false);
+    EXPECT_EQ(vm.current_state(), St::unavailable);
+}
+
+TEST_F(BaseVM, setAvailableRestartsRunning)
+{
+    vm.start();
+    ASSERT_EQ(vm.current_state(), St::running);
+    vm.set_available(false);
+    ASSERT_EQ(vm.current_state(), St::unavailable);
+
+    EXPECT_CALL(vm, start).Times(1);
+
+    vm.set_available(true);
+    EXPECT_EQ(vm.current_state(), St::running);
+}
+
+TEST_F(BaseVM, setAvailableKeepsStoppedStopped)
+{
+    vm.shutdown(mp::VirtualMachine::ShutdownPolicy::Poweroff);
+    ASSERT_EQ(vm.current_state(), St::stopped);
+    vm.set_available(false);
+    ASSERT_EQ(vm.current_state(), St::unavailable);
+
+    vm.set_available(true);
+    EXPECT_EQ(vm.current_state(), St::stopped);
+}
+
 } // namespace