def testGetArtifactsFail(self):
sysinfo_recipe = sysinfo.SysinfoRecipe('failsysinfo')
#pylint: disable=protected-access
sysinfo_recipe._platform = 'darwin'
sysinfo_recipe._SYSTEM_PROFILER_CMD = [
'echo', '-n', self._SYSTEM_PROFILER_FAIL_STRING
]
artifacts = sysinfo_recipe.GetArtifacts()
self.assertEqual(len(artifacts), 1)
artifact = artifacts[0]
self.assertIsInstance(artifact, base.ProcessOutputArtifact)
self.assertEqual(artifact.name, 'system_info.txt')
self.assertEqual(artifact.remote_path, 'system_info.txt')
artifact_content = artifact.OpenStream().read()
self.assertEqual(artifact_content, self._SYSTEM_PROFILER_FAIL_STRING)
def testGetArtifacts(self):
sysinfo_recipe = sysinfo.SysinfoRecipe('sysinfo')
#pylint: disable=protected-access
sysinfo_recipe._platform = 'linux'
sysinfo_recipe._DMI_DECODE_CMD = [
'echo', '-n', self._DMIDECODE_OUTPUT_STRING
]
artifacts = sysinfo_recipe.GetArtifacts()
self.assertEqual(len(artifacts), 1)
artifact = artifacts[0]
self.assertIsInstance(artifact, base.ProcessOutputArtifact)
self.assertEqual(artifact.name, 'system_info.txt')
self.assertEqual(artifact.remote_path, 'system_info.txt')
artifact_content = artifact.OpenStream().read()
self.assertEqual(artifact_content, self._DMIDECODE_OUTPUT_STRING)
def testGetArtifacts(self):
sysinfo_recipe = sysinfo.SysinfoRecipe('sysinfo')
# pylint: disable=protected-access
sysinfo_recipe._platform = 'linux'
# pylint: disable=line-too-long
with mock.patch(
'auto_forensicate.recipes.base.ProcessOutputArtifact._RunCommand'
) as patched_run:
patched_run.return_value = self._DMIDECODE_OUTPUT_STRING
artifacts = sysinfo_recipe.GetArtifacts()
self.assertEqual(len(artifacts), 2)
artifact = artifacts[0]
self.assertIsInstance(artifact, base.ProcessOutputArtifact)
self.assertEqual(artifact.name, 'system_info.txt')
self.assertEqual(artifact.remote_path, 'system_info.txt')
artifact_content = artifact.OpenStream().read()
self.assertEqual(artifact_content, self._DMIDECODE_OUTPUT_STRING)