| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 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 gmerge.""" | 7 """Unit tests for gmerge.""" |
| 8 | 8 |
| 9 import gmerge | 9 import gmerge |
| 10 import os | 10 import os |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 'CHROMEOS_RELEASE_BOARD=x86-mario\r\n', | 23 'CHROMEOS_RELEASE_BOARD=x86-mario\r\n', |
| 24 'CHROMEOS_DEVSERVER=http://localhost:8080/\n'] | 24 'CHROMEOS_DEVSERVER=http://localhost:8080/\n'] |
| 25 | 25 |
| 26 def testLsbRelease(self): | 26 def testLsbRelease(self): |
| 27 merger = gmerge.GMerger(self.lsb_release_lines) | 27 merger = gmerge.GMerger(self.lsb_release_lines) |
| 28 self.assertEqual({'CHROMEOS_RELEASE_BOARD': 'x86-mario', | 28 self.assertEqual({'CHROMEOS_RELEASE_BOARD': 'x86-mario', |
| 29 'CHROMEOS_DEVSERVER': 'http://localhost:8080/'}, | 29 'CHROMEOS_DEVSERVER': 'http://localhost:8080/'}, |
| 30 merger.lsb_release) | 30 merger.lsb_release) |
| 31 | 31 |
| 32 def testPostData(self): | 32 def testPostData(self): |
| 33 original_use = os.environ.get('USE', '') | 33 old_env = os.environ |
| 34 os.environ = {} |
| 34 os.environ['USE'] = 'a b c d +e' | 35 os.environ['USE'] = 'a b c d +e' |
| 35 gmerge.FLAGS = Flags({'accept_stable': 'blah'}) | 36 gmerge.FLAGS = Flags({'accept_stable': 'blah'}) |
| 36 | 37 |
| 37 merger = gmerge.GMerger(self.lsb_release_lines) | 38 merger = gmerge.GMerger(self.lsb_release_lines) |
| 38 self.assertEqual( | 39 self.assertEqual( |
| 39 'use=a+b+c+d+%2Be&pkg=package_name&board=x86-mario&accept_stable=blah', | 40 'use=a+b+c+d+%2Be&pkg=package_name&board=x86-mario&accept_stable=blah', |
| 40 merger.GeneratePackageRequest('package_name')) | 41 merger.GeneratePackageRequest('package_name')) |
| 41 os.environ['USE'] = original_use | 42 os.environ = old_env |
| 42 | 43 |
| 43 | 44 |
| 44 if __name__ == '__main__': | 45 if __name__ == '__main__': |
| 45 unittest.main() | 46 unittest.main() |
| OLD | NEW |