def test_check_umount_umount(self, cmd_exec_mock, check_mounted_mock, logger_mock):
"""Test umount for a single umount and no creation."""
# pylint: disable=protected-access
secure_mount._MOUNTED = ["/secdir"]
check_mounted_mock.return_value = False
secure_mount.umount()
self.assertEqual(secure_mount._MOUNTED, [])
cmd_exec_mock.run.assert_not_called()
logger_mock.error.assert_not_called()
logger_mock.warning.assert_called_once()
def test_check_umount_mount_busy(self, cmd_exec_mock, check_mounted_mock,
logger_mock):
"""Test umount for a single mount that fail."""
# pylint: disable=protected-access
secure_mount._MOUNTED = ["/secdir"]
check_mounted_mock.return_value = True
cmd_exec_mock.run.return_value = {"code": 1, "reterr": "Device busy"}
secure_mount.umount()
self.assertEqual(secure_mount._MOUNTED, [])
cmd_exec_mock.run.assert_called_once()
logger_mock.error.assert_called_once()
logger_mock.warning.assert_not_called()
def shutdown_handler(*_):
logger.info("TERM Signal received, shutting down...")
logger.debug("Stopping revocation notifier...")
revocation_process.terminate()
logger.debug("Shutting down HTTP server...")
server.shutdown()
server.server_close()
serverthread.join()
logger.debug("HTTP server stopped...")
revocation_process.join()
logger.debug("Revocation notifier stopped...")
secure_mount.umount()
logger.debug("Umounting directories...")
instance_tpm.flush_keys()
logger.debug("Flushed keys successfully")
sys.exit(0)
def test_check_umount_empty(self):
"""Test umount when there are nothing to clean."""
secure_mount.umount()
# pylint: disable=protected-access
self.assertEqual(secure_mount._MOUNTED, [])
