| 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 unittest | 11 import unittest |
| 11 | 12 |
| 12 class Flags(object): | 13 class Flags(object): |
| 13 def __init__(self, dictionary): | 14 def __init__(self, dictionary): |
| 14 self.__dict__.update(dictionary) | 15 self.__dict__.update(dictionary) |
| 15 | 16 |
| 16 | 17 |
| 17 class GMergeTest(unittest.TestCase): | 18 class GMergeTest(unittest.TestCase): |
| 18 """Test for gmerge.""" | 19 """Test for gmerge.""" |
| 19 | 20 |
| 20 def setUp(self): | 21 def setUp(self): |
| 21 self.lsb_release_lines = [ | 22 self.lsb_release_lines = [ |
| 22 'CHROMEOS_RELEASE_BOARD=x86-mario\r\n', | 23 'CHROMEOS_RELEASE_BOARD=x86-mario\r\n', |
| 23 'CHROMEOS_DEVSERVER=http://localhost:8080/\n'] | 24 'CHROMEOS_DEVSERVER=http://localhost:8080/\n'] |
| 24 | 25 |
| 25 def testLsbRelease(self): | 26 def testLsbRelease(self): |
| 26 merger = gmerge.GMerger(self.lsb_release_lines) | 27 merger = gmerge.GMerger(self.lsb_release_lines) |
| 27 self.assertEqual({'CHROMEOS_RELEASE_BOARD': 'x86-mario', | 28 self.assertEqual({'CHROMEOS_RELEASE_BOARD': 'x86-mario', |
| 28 'CHROMEOS_DEVSERVER': 'http://localhost:8080/'}, | 29 'CHROMEOS_DEVSERVER': 'http://localhost:8080/'}, |
| 29 merger.lsb_release) | 30 merger.lsb_release) |
| 30 | 31 |
| 31 def testPostData(self): | 32 def testPostData(self): |
| 32 gmerge.FLAGS = Flags({'use': 'a b c d +e', | 33 original_use = os.environ.get('USE', '') |
| 33 'accept_stable': 'blah'}) | 34 os.environ['USE'] = 'a b c d +e' |
| 35 gmerge.FLAGS = Flags({'accept_stable': 'blah'}) |
| 34 | 36 |
| 35 merger = gmerge.GMerger(self.lsb_release_lines) | 37 merger = gmerge.GMerger(self.lsb_release_lines) |
| 36 self.assertEqual( | 38 self.assertEqual( |
| 37 'use=a+b+c+d+%2Be&pkg=package_name&board=x86-mario&accept_stable=blah', | 39 'use=a+b+c+d+%2Be&pkg=package_name&board=x86-mario&accept_stable=blah', |
| 38 merger.GeneratePackageRequest('package_name')) | 40 merger.GeneratePackageRequest('package_name')) |
| 41 os.environ['USE'] = original_use |
| 39 | 42 |
| 40 | 43 |
| 41 if __name__ == '__main__': | 44 if __name__ == '__main__': |
| 42 unittest.main() | 45 unittest.main() |
| OLD | NEW |