mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 15:27:13 +03:00
4c76d686a8
Summary: Automatic upgrade to remove `version` override and silence errors. Reviewed By: grievejia Differential Revision: D17956249 fbshipit-source-id: d5c8b5aa73151b3fea67aec35d70f332030da2c9
237 lines
8.6 KiB
Python
237 lines
8.6 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
import pathlib
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import typing
|
|
import unittest
|
|
|
|
import pexpect
|
|
import toml
|
|
from eden.test_support.environment_variable import EnvironmentVariableMixin
|
|
from eden.test_support.temporary_directory import TemporaryDirectoryMixin
|
|
|
|
from .lib.edenfs_systemd import EdenFSSystemdMixin
|
|
from .lib.find_executables import FindExe
|
|
from .lib.pexpect import PexpectAssertionMixin
|
|
from .lib.systemd import SystemdUserServiceManagerMixin
|
|
|
|
|
|
class SystemdTest(
|
|
unittest.TestCase,
|
|
EnvironmentVariableMixin,
|
|
TemporaryDirectoryMixin,
|
|
SystemdUserServiceManagerMixin,
|
|
EdenFSSystemdMixin,
|
|
PexpectAssertionMixin,
|
|
):
|
|
"""Test Eden's systemd service for Linux."""
|
|
|
|
def setUp(self) -> None:
|
|
super().setUp()
|
|
|
|
self.set_environment_variable("EDEN_EXPERIMENTAL_SYSTEMD", "1")
|
|
self.eden_dir = self.make_temporary_directory()
|
|
self.etc_eden_dir = self.make_temporary_directory()
|
|
self.home_dir = self.make_temporary_directory()
|
|
|
|
# TODO(T33122320): Delete this test when systemd is properly integrated.
|
|
# TODO(T33122320): Test without --foreground.
|
|
def test_eden_start_says_systemd_mode_is_enabled(self) -> None:
|
|
def test(start_args: typing.List[str]) -> None:
|
|
with self.subTest(start_args=start_args):
|
|
start_process: "pexpect.spawn[str]" = pexpect.spawn(
|
|
FindExe.EDEN_CLI,
|
|
self.get_required_eden_cli_args()
|
|
+ ["start", "--foreground"]
|
|
+ start_args,
|
|
encoding="utf-8",
|
|
logfile=sys.stderr,
|
|
)
|
|
start_process.expect_exact("Running in experimental systemd mode")
|
|
start_process.expect_exact("Started edenfs")
|
|
|
|
test(start_args=["--", "--allowRoot"])
|
|
test(
|
|
start_args=[
|
|
"--daemon-binary",
|
|
typing.cast(str, FindExe.FAKE_EDENFS), # T38947910
|
|
]
|
|
)
|
|
|
|
# TODO(T33122320): Delete this test when systemd is properly integrated.
|
|
def test_eden_start_with_systemd_disabled_does_not_say_systemd_mode_is_enabled(
|
|
self
|
|
) -> None:
|
|
self.unset_environment_variable("EDEN_EXPERIMENTAL_SYSTEMD")
|
|
|
|
def test(start_args: typing.List[str]) -> None:
|
|
with self.subTest(start_args=start_args):
|
|
start_process: "pexpect.spawn[str]" = pexpect.spawn(
|
|
FindExe.EDEN_CLI,
|
|
self.get_required_eden_cli_args()
|
|
+ ["start", "--foreground"]
|
|
+ start_args,
|
|
encoding="utf-8",
|
|
logfile=sys.stderr,
|
|
)
|
|
start_process.expect_exact("Started edenfs")
|
|
self.assertNotIn(
|
|
"Running in experimental systemd mode", start_process.before
|
|
)
|
|
|
|
test(start_args=["--", "--allowRoot"])
|
|
test(
|
|
start_args=[
|
|
"--daemon-binary",
|
|
typing.cast(str, FindExe.FAKE_EDENFS), # T38947910
|
|
]
|
|
)
|
|
|
|
def test_eden_start_starts_systemd_service(self) -> None:
|
|
self.set_up_edenfs_systemd_service()
|
|
subprocess.check_call(
|
|
[typing.cast(str, FindExe.EDEN_CLI)] # T38947910
|
|
+ self.get_required_eden_cli_args()
|
|
+ [
|
|
"start",
|
|
"--daemon-binary",
|
|
typing.cast(str, FindExe.FAKE_EDENFS), # T38947910
|
|
]
|
|
)
|
|
self.assert_systemd_service_is_active(eden_dir=pathlib.Path(self.eden_dir))
|
|
|
|
def test_systemd_service_is_failed_if_edenfs_crashes_on_start(self) -> None:
|
|
self.set_up_edenfs_systemd_service()
|
|
self.assert_systemd_service_is_stopped(eden_dir=pathlib.Path(self.eden_dir))
|
|
subprocess.call(
|
|
[typing.cast(str, FindExe.EDEN_CLI)] # T38947910
|
|
+ self.get_required_eden_cli_args()
|
|
+ [
|
|
"start",
|
|
"--daemon-binary",
|
|
typing.cast(str, FindExe.FAKE_EDENFS), # T38947910
|
|
"--",
|
|
"--failDuringStartup",
|
|
]
|
|
)
|
|
self.assert_systemd_service_is_failed(eden_dir=pathlib.Path(self.eden_dir))
|
|
|
|
def test_eden_start_reports_service_failure_if_edenfs_fails_during_startup(
|
|
self
|
|
) -> None:
|
|
self.set_up_edenfs_systemd_service()
|
|
start_process = self.spawn_start_with_fake_edenfs(
|
|
extra_args=["--", "--failDuringStartup"]
|
|
)
|
|
start_process.expect(
|
|
r"error: Starting the fb-edenfs@.+?\.service systemd service "
|
|
r"failed \(reason: exit-code\)"
|
|
)
|
|
self.assertNotIn(
|
|
"journalctl",
|
|
start_process.before,
|
|
"journalctl doesn't work and should not be mentioned",
|
|
)
|
|
remaining_output = start_process.read()
|
|
self.assertNotIn(
|
|
"journalctl",
|
|
remaining_output,
|
|
"journalctl doesn't work and should not be mentioned",
|
|
)
|
|
|
|
def test_eden_start_reports_error_if_systemd_is_dead(self) -> None:
|
|
self.set_up_edenfs_systemd_service()
|
|
systemd = self.systemd
|
|
assert systemd is not None
|
|
systemd.exit()
|
|
self.assertTrue(
|
|
(systemd.xdg_runtime_dir / "systemd" / "private").exists(),
|
|
"systemd's socket file should still exist",
|
|
)
|
|
|
|
self.spoof_user_name("testuser")
|
|
start_process = self.spawn_start_with_fake_edenfs()
|
|
start_process.expect_exact(
|
|
"error: The systemd user manager is not running. Run the following "
|
|
"command to\r\nstart it, then try again:"
|
|
)
|
|
start_process.expect_exact("sudo systemctl start user@testuser.service")
|
|
|
|
def test_eden_start_reports_error_if_systemd_is_dead_and_cleaned_up(self) -> None:
|
|
self.set_up_edenfs_systemd_service()
|
|
systemd = self.systemd
|
|
assert systemd is not None
|
|
systemd.exit()
|
|
shutil.rmtree(systemd.xdg_runtime_dir)
|
|
|
|
self.spoof_user_name("testuser")
|
|
start_process = self.spawn_start_with_fake_edenfs()
|
|
start_process.expect_exact(
|
|
"error: The systemd user manager is not running. Run the following "
|
|
"command to\r\nstart it, then try again:"
|
|
)
|
|
start_process.expect_exact("sudo systemctl start user@testuser.service")
|
|
|
|
def test_eden_start_uses_fallback_if_systemd_environment_is_missing(self) -> None:
|
|
self.set_up_edenfs_systemd_service()
|
|
systemd = self.systemd
|
|
assert systemd is not None
|
|
|
|
fallback_xdg_runtime_dir = str(systemd.xdg_runtime_dir)
|
|
self.set_eden_config(
|
|
{"service": {"fallback_systemd_xdg_runtime_dir": fallback_xdg_runtime_dir}}
|
|
)
|
|
self.unset_environment_variable("XDG_RUNTIME_DIR")
|
|
|
|
start_process = self.spawn_start_with_fake_edenfs()
|
|
start_process.expect_exact(
|
|
f"warning: The XDG_RUNTIME_DIR environment variable is not set; "
|
|
f"using fallback: '{fallback_xdg_runtime_dir}'"
|
|
)
|
|
start_process.expect_exact("Started edenfs")
|
|
self.assert_process_succeeds(start_process)
|
|
self.assert_systemd_service_is_active(eden_dir=pathlib.Path(self.eden_dir))
|
|
|
|
def spawn_start_with_fake_edenfs(
|
|
self, extra_args: typing.Sequence[str] = ()
|
|
) -> "pexpect.spawn[str]":
|
|
return pexpect.spawn(
|
|
FindExe.EDEN_CLI,
|
|
self.get_required_eden_cli_args()
|
|
+ [
|
|
"start",
|
|
"--daemon-binary",
|
|
typing.cast(str, FindExe.FAKE_EDENFS), # T38947910
|
|
]
|
|
+ list(extra_args),
|
|
encoding="utf-8",
|
|
logfile=sys.stderr,
|
|
)
|
|
|
|
def get_required_eden_cli_args(self) -> typing.List[str]:
|
|
return [
|
|
"--config-dir",
|
|
self.eden_dir,
|
|
"--etc-eden-dir",
|
|
self.etc_eden_dir,
|
|
"--home-dir",
|
|
self.home_dir,
|
|
]
|
|
|
|
def set_eden_config(self, config) -> None:
|
|
config_d = pathlib.Path(self.etc_eden_dir) / "config.d"
|
|
config_d.mkdir()
|
|
with open(config_d / "systemd.toml", "w") as config_file:
|
|
# pyre-fixme[6]: Expected `_Writable` for 2nd param but got `IO[Any]`.
|
|
toml.dump(config, config_file)
|
|
|
|
def spoof_user_name(self, user_name: str) -> None:
|
|
self.set_environment_variable("LOGNAME", user_name)
|
|
self.set_environment_variable("USER", user_name)
|