def setUp(self):
self._dl = download.BaseDownloader()
# filesystem
self.filesystem = fake_filesystem.FakeFilesystem()
self.filesystem.CreateFile(r'C:\input.ini', contents=_TEST_INI)
download.os = fake_filesystem.FakeOsModule(self.filesystem)
download.open = fake_filesystem.FakeFileOpen(self.filesystem)
def setUp(self):
self.cache = cache.Cache()
fakefs = fake_filesystem.FakeFilesystem()
fakefs.CreateDirectory(r'C:\Directory')
os_module = fake_filesystem.FakeOsModule(fakefs)
self.mock_open = fake_filesystem.FakeFileOpen(fakefs)
cache.os = os_module
cache.open = self.mock_open
def setUp(self):
self.buildinfo = buildinfo.BuildInfo()
# filesystem
self.filesystem = fake_filesystem.FakeFilesystem()
runner.os = fake_filesystem.FakeOsModule(self.filesystem)
runner.open = fake_filesystem.FakeFileOpen(self.filesystem)
runner.shutil = fake_filesystem_shutil.FakeShutilModule(
self.filesystem)
self.cr = runner.ConfigRunner(self.buildinfo)
self.cr._task_list_path = '/tmp/task_list.yaml'
def setUp(self):
# fake filesystem
fakefs = fake_filesystem.FakeFilesystem()
fakefs.CreateDirectory(r'/stage')
fakefs.CreateFile(r'/file1.txt', contents='file1')
fakefs.CreateFile(r'/file2.txt', contents='file2')
self.fake_open = fake_filesystem.FakeFileOpen(fakefs)
file_system.os = fake_filesystem.FakeOsModule(fakefs)
file_system.shutil = fake_filesystem_shutil.FakeShutilModule(fakefs)
file_system.open = self.fake_open
self.fakefs = fakefs
def setUp(self):
# fake filesystem
self.filesystem = fake_filesystem.FakeFilesystem()
self.filesystem.CreateDirectory('/dev')
buildinfo.os = fake_filesystem.FakeOsModule(self.filesystem)
buildinfo.open = fake_filesystem.FakeFileOpen(self.filesystem)
# setup
mock_wmi = mock.patch.object(buildinfo.hw_info.wmi_query,
'WMIQuery',
autospec=True)
self.addCleanup(mock_wmi.stop)
mock_wmi.start()
self.buildinfo = buildinfo.BuildInfo()
def setUp(self, unused_tk):
self.ui = chooser.Chooser(_TEST_CONF, preload=False)
v1 = mock.Mock()
v1.Value.return_value = 'value1'
v2 = mock.Mock()
v2.Value.return_value = 'value2'
v3 = mock.Mock()
v3.Value.return_value = 'value3'
self.ui.fields = {'field1': v1, 'field2': v2, 'field3': v3}
self.fs = fake_filesystem.FakeFilesystem()
chooser.resources.os = fake_filesystem.FakeOsModule(self.fs)
self.fs.CreateFile('/resources/logo.gif')
def testSetupTaskList(self):
cache = autobuild.constants.SYS_CACHE
filesystem = fake_filesystem.FakeFilesystem()
filesystem.CreateFile(r'X:\task_list.yaml')
autobuild.os = fake_filesystem.FakeOsModule(filesystem)
self.assertEqual(self.autobuild._SetupTaskList(),
'%s\\task_list.yaml' % cache)
autobuild.FLAGS.preserve_tasks = True
self.assertEqual(self.autobuild._SetupTaskList(),
'%s\\task_list.yaml' % cache)
autobuild.FLAGS.environment = 'WinPE'
self.assertEqual(self.autobuild._SetupTaskList(), r'X:\task_list.yaml')
self.assertTrue(autobuild.os.path.exists(r'X:\task_list.yaml'))
autobuild.FLAGS.preserve_tasks = False
self.assertEqual(self.autobuild._SetupTaskList(), r'X:\task_list.yaml')
self.assertFalse(autobuild.os.path.exists(r'X:\task_list.yaml'))
def testBuildInfoSave(self, build_info, reg):
fs = fake_filesystem.FakeFilesystem()
installer.open = fake_filesystem.FakeFileOpen(fs)
installer.os = fake_filesystem.FakeOsModule(fs)
fs.CreateFile(
'/tmp/build_info.yaml',
contents='{BUILD: {opt 1: true, opt 2: some value, opt 3: 12345}}\n')
build_info.Cache.return_value.Path.return_value = '/tmp'
s = installer.BuildInfoSave(None, build_info)
s.Run()
reg.return_value.SetKeyValue.assert_has_calls(
[
mock.call(installer.constants.REG_ROOT, 'opt 1', True),
mock.call(installer.constants.REG_ROOT, 'opt 2', 'some value'),
mock.call(installer.constants.REG_ROOT, 'opt 3', 12345),
],
any_order=True)
s.Run()
def setUp(self):
self.fs = fake_filesystem.FakeFilesystem()
powershell.os = fake_filesystem.FakeOsModule(self.fs)
powershell.resources.os = fake_filesystem.FakeOsModule(self.fs)
self.fs.CreateFile('/resources/bin/script.ps1')
self.ps = powershell.PowerShell()
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem()
files.open = fake_filesystem.FakeFileOpen(self.filesystem)
files.file_util.os = fake_filesystem.FakeOsModule(self.filesystem)
def setUp(self):
self.fs = fake_filesystem.FakeFilesystem()
resources.os = fake_filesystem.FakeOsModule(self.fs)
self.fs.CreateFile('/test/file.txt')
