| OLD | NEW |
| 1 #!/usr/bin/python -tt | 1 #!/usr/bin/python -tt |
| 2 # | 2 # |
| 3 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. | 3 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. |
| 4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
| 5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
| 6 | 6 |
| 7 """Unit tests for bmpblk_utility. | 7 """Unit tests for bmpblk_utility. |
| 8 """ | 8 """ |
| 9 | 9 |
| 10 import os | 10 import os |
| (...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 self.assertEqual(0, rc) | 82 self.assertEqual(0, rc) |
| 83 rc, out, err = runprog(prog, '-x', '-d', './FOO_DIR', 'FOO') | 83 rc, out, err = runprog(prog, '-x', '-d', './FOO_DIR', 'FOO') |
| 84 self.assertEqual(0, rc) | 84 self.assertEqual(0, rc) |
| 85 os.chdir('./FOO_DIR') | 85 os.chdir('./FOO_DIR') |
| 86 rc, out, err = runprog(prog, '-c', 'config.yaml', 'BAR') | 86 rc, out, err = runprog(prog, '-c', 'config.yaml', 'BAR') |
| 87 self.assertEqual(0, rc) | 87 self.assertEqual(0, rc) |
| 88 rc, out, err = runprog('/usr/bin/cmp', '../FOO', 'BAR') | 88 rc, out, err = runprog('/usr/bin/cmp', '../FOO', 'BAR') |
| 89 self.assertEqual(0, rc) | 89 self.assertEqual(0, rc) |
| 90 os.chdir('..') | 90 os.chdir('..') |
| 91 | 91 |
| 92 def testPackUnpackZ(self): | 92 def doPackUnpackZ(self, comp): |
| 93 """Create, unpack, recreate with explicit compression""" | 93 """Create, unpack, recreate with a given compression""" |
| 94 rc, out, err = runprog(prog, '-z', '1', '-c', 'case_simple.yaml', 'FOO') | 94 rc, out, err = runprog(prog, '-z', comp, '-c', 'case_simple.yaml', 'FOO') |
| 95 self.assertEqual(0, rc) | 95 self.assertEqual(0, rc) |
| 96 rc, out, err = runprog(prog, '-x', '-d', './FOO_DIR', 'FOO') | 96 rc, out, err = runprog(prog, '-x', '-d', './FOO_DIR', 'FOO') |
| 97 self.assertEqual(0, rc) | 97 self.assertEqual(0, rc) |
| 98 os.chdir('./FOO_DIR') | 98 os.chdir('./FOO_DIR') |
| 99 rc, out, err = runprog(prog, '-z', '1', '-c', 'config.yaml', 'BAR') | 99 rc, out, err = runprog(prog, '-z', comp, '-c', 'config.yaml', 'BAR') |
| 100 self.assertEqual(0, rc) | 100 self.assertEqual(0, rc) |
| 101 rc, out, err = runprog('/usr/bin/cmp', '../FOO', 'BAR') | 101 rc, out, err = runprog('/usr/bin/cmp', '../FOO', 'BAR') |
| 102 self.assertEqual(0, rc) | 102 self.assertEqual(0, rc) |
| 103 os.chdir('..') | 103 os.chdir('..') |
| 104 | 104 |
| 105 def testPackUnpackZ1(self): |
| 106 """Create, unpack, recreate with EFIv1 compression""" |
| 107 self.doPackUnpackZ('1'); |
| 108 |
| 109 def testPackUnpackZ2(self): |
| 110 """Create, unpack, recreate with LZMA compression""" |
| 111 self.doPackUnpackZ('2'); |
| 112 |
| 105 def tearDown(self): | 113 def tearDown(self): |
| 106 rc, out, err = runprog('/bin/rm', '-rf', './FOO_DIR', 'FOO') | 114 rc, out, err = runprog('/bin/rm', '-rf', './FOO_DIR', 'FOO') |
| 107 self.assertEqual(0, rc) | 115 self.assertEqual(0, rc) |
| 108 | 116 |
| 109 | 117 |
| 110 # Run these tests | 118 # Run these tests |
| 111 if __name__ == '__main__': | 119 if __name__ == '__main__': |
| 112 varname = 'BMPBLK' | 120 varname = 'BMPBLK' |
| 113 if varname not in os.environ: | 121 if varname not in os.environ: |
| 114 print('You must specify the path to bmpblk_utility in the $%s ' | 122 print('You must specify the path to bmpblk_utility in the $%s ' |
| 115 'environment variable.' % varname) | 123 'environment variable.' % varname) |
| 116 sys.exit(1) | 124 sys.exit(1) |
| 117 prog = os.environ[varname] | 125 prog = os.environ[varname] |
| 118 print "Testing prog...", prog | 126 print "Testing prog...", prog |
| 119 unittest.main() | 127 unittest.main() |
| 120 | 128 |
| OLD | NEW |