Moving tests from src folder.
Adding the reboot behavior to mock hypervisor.