def testTimeout(self):
"""Verifies that cleanup code correctly times out.
Cleanup can fail because of I/O caches and other similar things keeping
the mount active. Test that the cleanup code properly times out in
these scenarios.
"""
host_dir = lxc.SharedHostDir(self.shared_host_path)
# Create a process in the shared dir to force unmounting to fail.
p = subprocess.Popen(['sleep', '2'], cwd=self.shared_host_path)
# Cleanup should time out.
with self.assertRaises(error.CmdError):
logging.debug('attempting cleanup (should fail)')
# Use a short timeout so the test doesn't take forever.
host_dir.cleanup(timeout=1)
logging.debug('cleanup did not fail')
# Kill the process occupying the mount.
p.terminate()
# Cleanup should succeed.
try:
# Use the default timeout so this doesn't raise false errors.
host_dir.cleanup()
except error.CmdError as e:
self.fail('Unexpected cleanup error: %r' % e)
def testHostDirAccess(self):
"""Verifies that sudo is not required to write to the shared host dir.
"""
try:
host_dir = lxc.SharedHostDir(self.shared_host_path)
tempfile.NamedTemporaryFile(dir=host_dir.path)
except OSError:
self.fail('Unable to write to shared host dir.\n%s' %
error.format_error())
finally:
host_dir.cleanup()
def testHostDirNotMounted(self):
"""Verifies that an unmounted host dir does not cause container bucket
construction to crash.
"""
# Create the shared host dir, but do not mount it.
os.makedirs(self.shared_host_path)
# Setup then destroy the HPM. This should not emit any exceptions.
try:
host_dir = lxc.SharedHostDir(self.shared_host_path)
host_dir.cleanup()
except:
self.fail('SharedHostDir crashed.\n%s' % error.format_error())
def testHostDirMissing(self):
"""Verifies that a missing host dir does not cause cleanup to crash.
"""
host_dir = lxc.SharedHostDir(self.shared_host_path)
# Manually destroy the host path
utils.run('sudo umount %(path)s && sudo rmdir %(path)s' %
{'path': self.shared_host_path})
# Verify that the host path does not exist.
self.assertFalse(os.path.exists(self.shared_host_path))
try:
host_dir.cleanup()
except:
self.fail('SharedHostDir.cleanup crashed.\n%s' %
error.format_error())
def testHostDirCreationAndCleanup(self):
"""Verifies that the host dir is properly created and cleaned up when
the container bucket is set up and destroyed.
"""
# Precondition: host path nonexistent
self.assertFalse(os.path.isdir(self.shared_host_path))
host_dir = lxc.SharedHostDir(self.shared_host_path)
# Verify the host path in the host_dir.
self.assertEqual(os.path.realpath(host_dir.path),
os.path.realpath(self.shared_host_path))
self.assertTrue(os.path.isdir(self.shared_host_path))
# Clean up, verify that the path is removed.
host_dir.cleanup()
self.assertFalse(os.path.isdir(self.shared_host_path))
def setUpClass(cls):
cls.test_dir = tempfile.mkdtemp(dir=lxc.DEFAULT_CONTAINER_PATH,
prefix='zygote_unittest_')
# Check if a base container exists on this machine and download one if
# necessary.
image = lxc.BaseImage()
try:
cls.base_container = image.get()
cls.cleanup_base_container = False
except error.ContainerError:
image.setup()
cls.base_container = image.get()
cls.cleanup_base_container = True
assert (cls.base_container is not None)
# Set up the zygote host path.
cls.shared_host_dir = lxc.SharedHostDir(
os.path.join(cls.test_dir, 'host'))
def _start(args):
"""Starts up the container pool service.
This function instantiates and starts up the pool service on the current
thread (i.e. the function will block, and not return until the service is
shut down).
"""
# TODO(dshi): crbug.com/459344 Set remove this enforcement when test
# container can be unprivileged container.
if utils.sudo_require_password():
logging.warning('SSP requires root privilege to run commands, please '
'grant root access to this process.')
utils.run('sudo true')
# Configure logging.
config = server_logging_config.ServerLoggingConfig()
config.configure_logging(verbose=args.verbose)
config.add_debug_file_handlers(log_dir=_LOG_LOCATION, log_name=_LOG_NAME)
# Pool code is heavily multi-threaded. This will help debugging.
logging_config.add_threadname_in_log()
host_dir = lxc.SharedHostDir()
service = container_pool.Service(host_dir)
# Catch signals, and send the appropriate stop request to the service
# instead of killing the main thread.
# - SIGINT is generated by Ctrl-C
# - SIGTERM is generated by an upstart stopping event.
for sig in (signal.SIGINT, signal.SIGTERM):
signal.signal(sig, lambda s, f: service.stop())
with ts_mon_config.SetupTsMonGlobalState(service_name='lxc_pool_service',
indirect=True,
short_lived=False):
# Start the service. This blocks and does not return till the service
# shuts down.
service.start(pool_size=args.size)
def _create_client(timeout=3):
logging.debug('Creating client...')
address = os.path.join(lxc.SharedHostDir().path,
lxc.DEFAULT_CONTAINER_POOL_SOCKET)
with container_pool.Client.connect(address, timeout) as connection:
yield connection