def test_start_extension_command_should_use_systemd_if_not_failing(self):
original_popen = subprocess.Popen
def mock_popen(*args, **kwargs):
return original_popen(*args, **kwargs)
with tempfile.TemporaryFile(dir=self.tmp_dir, mode="w+b") as stdout:
with tempfile.TemporaryFile(dir=self.tmp_dir,
mode="w+b") as stderr:
with patch("azurelinuxagent.common.cgroupapi.subprocess.Popen",
side_effect=mock_popen) as patch_mock_popen:
process, extension_cgroups = SystemdCgroupsApi(
).start_extension_command(
extension_name="Microsoft.Compute.TestExtension-1.2.3",
command="date",
shell=True,
cwd=self.tmp_dir,
env={},
stdout=stdout,
stderr=stderr)
process.wait()
ret = process.poll()
# We should have invoked the extension command only once and succeeded
self.assertEquals(0, ret)
self.assertEquals(1, patch_mock_popen.call_count)
args = patch_mock_popen.call_args[0][0]
self.assertIn("systemd-run --unit", args)
self.assert_cgroups_created(extension_cgroups)
def test_start_extension_command_should_not_use_systemd_if_failing(self):
original_popen = subprocess.Popen
def mock_popen(*args, **kwargs):
# Inject a syntax error to the call
systemd_command = args[0].replace('systemd-run',
'systemd-run syntax_error')
new_args = (systemd_command, )
return original_popen(new_args, **kwargs)
with tempfile.TemporaryFile(dir=self.tmp_dir, mode="w+b") as stdout:
with tempfile.TemporaryFile(dir=self.tmp_dir,
mode="w+b") as stderr:
with patch("azurelinuxagent.common.cgroupapi.add_event"
) as mock_add_event:
with patch(
"azurelinuxagent.common.cgroupapi.subprocess.Popen",
side_effect=mock_popen) as patch_mock_popen:
# We expect this call to fail because of the syntax error
process, extension_cgroups = SystemdCgroupsApi(
).start_extension_command(
extension_name=
"Microsoft.Compute.TestExtension-1.2.3",
command="date",
shell=True,
cwd=self.tmp_dir,
env={},
stdout=stdout,
stderr=stderr)
process.wait()
args, kwargs = mock_add_event.call_args
self.assertIn(
"Failed to run systemd-run for unit Microsoft.Compute.TestExtension_1.2.3",
kwargs['message'])
self.assertIn(
"Failed to find executable syntax_error: No such file or directory",
kwargs['message'])
self.assertEquals(False, kwargs['is_success'])
self.assertEquals('InvokeCommandUsingSystemd',
kwargs['op'])
# We expect the process to ultimately succeed since the fallback option worked
ret = process.poll()
self.assertEquals(0, ret)
# We expect two calls to Popen, first for the systemd-run call, second for the fallback option
self.assertEquals(2, patch_mock_popen.call_count)
first_call_args = patch_mock_popen.mock_calls[0][1][0]
second_call_args = patch_mock_popen.mock_calls[1][1][0]
self.assertIn("systemd-run --unit", first_call_args)
self.assertNotIn("systemd-run --unit",
second_call_args)
# No cgroups should have been created
self.assertEquals(extension_cgroups, [])
def test_start_extension_command_should_use_systemd_if_not_failing(self):
original_popen = subprocess.Popen
def mock_popen(*args, **kwargs):
return original_popen(*args, **kwargs)
with tempfile.TemporaryFile(dir=self.tmp_dir, mode="w+b") as stdout:
with tempfile.TemporaryFile(dir=self.tmp_dir, mode="w+b") as stderr:
with patch("azurelinuxagent.common.cgroupapi.subprocess.Popen", side_effect=mock_popen) as patch_mock_popen:
process, extension_cgroups = SystemdCgroupsApi().start_extension_command(
extension_name="Microsoft.Compute.TestExtension-1.2.3",
command="date",
shell=True,
cwd=self.tmp_dir,
env={},
stdout=stdout,
stderr=stderr)
process.wait()
ret = process.poll()
# We should have invoked the extension command only once and succeeded
self.assertEquals(0, ret)
self.assertEquals(1, patch_mock_popen.call_count)
# Assert that the extension's cgroups were created as well
self.assertEqual(len(extension_cgroups), 2, 'start_extension_command did not return the expected number of cgroups')
cpu_found = memory_found = False
for cgroup in extension_cgroups:
match = re.match(r'^/sys/fs/cgroup/(cpu|memory)/system.slice/Microsoft.Compute.TestExtension_1\.2\.3\_([a-f0-9-]+)\.scope$', cgroup.path)
self.assertTrue(match is not None, "Unexpected path for cgroup: {0}".format(cgroup.path))
if match.group(1) == 'cpu':
cpu_found = True
if match.group(1) == 'memory':
memory_found = True
self.assertTrue(cpu_found, 'start_extension_command did not return a cpu cgroup')
self.assertTrue(memory_found, 'start_extension_command did not return a memory cgroup')