def test_image_extract(self):
     img1 = self._create_tar({
         'file1': BytesIO(b'a'),
         'file2': BytesIO(b'b')
     })
     img1_dev = '/dev/1.%s' % os.path.basename(img1)
     img2 = self._create_tar({
         'file3': BytesIO(b'a'),
         'file4': BytesIO(b'b')
     })
     img2_dev = '/dev/2.%s' % os.path.basename(img2)
     opts = '--zvm-image=%s --zvm-image=%s,/lib' % (img1, img2)
     self.program = 'file2'
     self.argv = [ZVSH]
     self.argv.extend(opts.split())
     self.argv.append(self.program)
     cmd_line = self.program
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(
             join_path(shell.zvsh.tmpdir, 'manifest.1'))
         boot = os.path.join(os.path.abspath(shell.zvsh.tmpdir), 'boot.1')
         reference = self._reference_manifest(shell.zvsh.tmpdir,
                                              executable=boot)
         channels = reference['channel']
         img1_chan = [
             os.path.abspath(img1), img1_dev, '3', '0',
             str(DEFAULT_LIMITS['reads']),
             str(DEFAULT_LIMITS['rbytes']),
             str(DEFAULT_LIMITS['writes']),
             str(DEFAULT_LIMITS['wbytes'])
         ]
         img2_chan = [
             os.path.abspath(img2), img2_dev, '3', '0',
             str(DEFAULT_LIMITS['reads']),
             str(DEFAULT_LIMITS['rbytes']),
             str(DEFAULT_LIMITS['writes']),
             str(DEFAULT_LIMITS['wbytes'])
         ]
         # insert in reverse order
         channels.insert(3, img2_chan)
         channels.insert(3, img1_chan)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir, 'nvram.1'))
         reference = _reference_nvram(cmd_line,
                                      images=[(img1_dev, '/', 'ro', 'no'),
                                              (img2_dev, '/lib', 'ro', 'no')
                                              ])
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
Exemple #2
0
 def test_image_extract(self):
     img1 = self._create_tar({'file1': BytesIO(b'a'),
                              'file2': BytesIO(b'b')})
     img1_dev = '/dev/1.%s' % os.path.basename(img1)
     img2 = self._create_tar({'file3': BytesIO(b'a'),
                              'file4': BytesIO(b'b')})
     img2_dev = '/dev/2.%s' % os.path.basename(img2)
     opts = '--zvm-image=%s --zvm-image=%s,/lib' % (img1, img2)
     self.program = 'file2'
     self.argv = [ZVSH]
     self.argv.extend(opts.split())
     self.argv.append(self.program)
     cmd_line = self.program
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(join_path(shell.zvsh.tmpdir,
                                             'manifest.1'))
         boot = os.path.join(os.path.abspath(shell.zvsh.tmpdir), 'boot.1')
         reference = self._reference_manifest(shell.zvsh.tmpdir,
                                              executable=boot)
         channels = reference['channel']
         img1_chan = [os.path.abspath(img1),
                      img1_dev,
                      '3',
                      '0',
                      str(DEFAULT_LIMITS['reads']),
                      str(DEFAULT_LIMITS['rbytes']),
                      str(DEFAULT_LIMITS['writes']),
                      str(DEFAULT_LIMITS['wbytes'])]
         img2_chan = [os.path.abspath(img2),
                      img2_dev,
                      '3',
                      '0',
                      str(DEFAULT_LIMITS['reads']),
                      str(DEFAULT_LIMITS['rbytes']),
                      str(DEFAULT_LIMITS['writes']),
                      str(DEFAULT_LIMITS['wbytes'])]
         # insert in reverse order
         channels.insert(3, img2_chan)
         channels.insert(3, img1_chan)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir,
                                       'nvram.1'))
         reference = _reference_nvram(
             cmd_line,
             images=[(img1_dev, '/', 'ro', 'no'),
                     (img2_dev, '/lib', 'ro', 'no')])
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
Exemple #3
0
 def test_input_output_file(self):
     in_fd, in_name = mkstemp(dir=self.testdir)
     os.write(in_fd, b'test')
     os.close(in_fd)
     out_fd, out_name = mkstemp(dir=self.testdir)
     os.close(out_fd)
     os.unlink(out_name)
     self.assertFalse(os.path.exists(out_name))
     opts = '-i @%s -o @%s' % (in_name, out_name)
     in_dev_name = '/dev/1.%s' % os.path.basename(in_name)
     out_dev_name = '/dev/2.%s' % os.path.basename(out_name)
     cmd_line = ' '.join([self.program, '-i %s -o %s'
                                        % (in_dev_name, out_dev_name)])
     self.argv.extend(opts.split())
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(join_path(shell.zvsh.tmpdir,
                                             'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         channels = reference['channel']
         file_chan = [out_name,
                      out_dev_name,
                      '3',
                      '0',
                      str(DEFAULT_LIMITS['reads']),
                      str(DEFAULT_LIMITS['rbytes']),
                      str(DEFAULT_LIMITS['writes']),
                      str(DEFAULT_LIMITS['wbytes'])]
         channels.insert(3, file_chan)
         file_chan = [in_name,
                      in_dev_name,
                      '3',
                      '0',
                      str(DEFAULT_LIMITS['reads']),
                      str(DEFAULT_LIMITS['rbytes']),
                      str(DEFAULT_LIMITS['writes']),
                      str(DEFAULT_LIMITS['wbytes'])]
         channels.insert(3, file_chan)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir,
                                       'nvram.1'))
         reference = _reference_nvram(cmd_line,
                                      [{'channel': in_dev_name,
                                        'mode': 'file'},
                                       {'channel': out_dev_name,
                                        'mode': 'file'}])
         self.assertEqual(nvram, reference)
         self.assertTrue(os.path.exists(out_name))
     finally:
         shell.zvsh.orig_cleanup()
 def test_one_command(self):
     cmd_line = self.program
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(
             join_path(shell.zvsh.tmpdir, 'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir, 'nvram.1'))
         reference = _reference_nvram(cmd_line)
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
Exemple #5
0
 def test_one_command(self):
     cmd_line = self.program
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(join_path(shell.zvsh.tmpdir,
                                             'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir,
                                       'nvram.1'))
         reference = _reference_nvram(cmd_line)
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
 def test_command_with_args(self):
     opts = '-t --test test'
     cmd_line = ' '.join([self.program, opts])
     self.argv.extend(opts.split())
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(
             join_path(shell.zvsh.tmpdir, 'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir, 'nvram.1'))
         reference = _reference_nvram(cmd_line)
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
Exemple #7
0
 def test_command_with_args(self):
     opts = '-t --test test'
     cmd_line = ' '.join([self.program, opts])
     self.argv.extend(opts.split())
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(join_path(shell.zvsh.tmpdir,
                                             'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir,
                                       'nvram.1'))
         reference = _reference_nvram(cmd_line)
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
Exemple #8
0
 def test_wo_image(self):
     img1 = self._create_tar({'file1': BytesIO(b'a'),
                              'file2': BytesIO(b'b')})
     img1_dev = '/dev/1.%s' % os.path.basename(img1)
     opts = '--zvm-image=%s,/,wo' % img1
     self.argv = [ZVSH]
     self.argv.extend(opts.split())
     self.argv.append(self.program)
     cmd_line = self.program
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(join_path(shell.zvsh.tmpdir,
                                             'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         channels = reference['channel']
         img1_chan = [os.path.abspath(img1),
                      img1_dev,
                      '3',
                      '0',
                      str(DEFAULT_LIMITS['reads']),
                      str(DEFAULT_LIMITS['rbytes']),
                      str(DEFAULT_LIMITS['writes']),
                      str(DEFAULT_LIMITS['wbytes'])]
         channels.insert(3, img1_chan)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir,
                                       'nvram.1'))
         reference = _reference_nvram(
             cmd_line,
             images=[(img1_dev, '/', 'wo', 'no')])
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()
Exemple #9
0
 def test_existing_file(self):
     fd, name = mkstemp(dir=self.testdir)
     os.write(fd, b'test')
     os.close(fd)
     opts = '-f @%s' % name
     dev_name = '/dev/1.%s' % os.path.basename(name)
     cmd_line = ' '.join([self.program, '-f %s' % dev_name])
     self.argv.extend(opts.split())
     shell = Shell(self.argv)
     try:
         with pytest.raises(SystemExit):
             shell.run()
         files = os.listdir(shell.zvsh.tmpdir)
         self.assertTrue('manifest.1' in files)
         self.assertTrue('nvram.1' in files)
         manifest = _read_manifest(join_path(shell.zvsh.tmpdir,
                                             'manifest.1'))
         reference = self._reference_manifest(shell.zvsh.tmpdir)
         channels = reference['channel']
         file_chan = [name,
                      dev_name,
                      '3',
                      '0',
                      str(DEFAULT_LIMITS['reads']),
                      str(DEFAULT_LIMITS['rbytes']),
                      str(DEFAULT_LIMITS['writes']),
                      str(DEFAULT_LIMITS['wbytes'])]
         channels.insert(3, file_chan)
         self.assertEqual(manifest, reference)
         nvram = _read_nvram(join_path(shell.zvsh.tmpdir,
                                       'nvram.1'))
         reference = _reference_nvram(cmd_line,
                                      [{'channel': dev_name,
                                        'mode': 'file'}])
         self.assertEqual(nvram, reference)
     finally:
         shell.zvsh.orig_cleanup()