| OLD | NEW |
| 1 # Copyright (C) 2011 Google Inc. All rights reserved. | 1 # Copyright (C) 2011 Google Inc. All rights reserved. |
| 2 # | 2 # |
| 3 # Redistribution and use in source and binary forms, with or without | 3 # Redistribution and use in source and binary forms, with or without |
| 4 # modification, are permitted provided that the following conditions are | 4 # modification, are permitted provided that the following conditions are |
| 5 # met: | 5 # met: |
| 6 # | 6 # |
| 7 # * Redistributions of source code must retain the above copyright | 7 # * Redistributions of source code must retain the above copyright |
| 8 # notice, this list of conditions and the following disclaimer. | 8 # notice, this list of conditions and the following disclaimer. |
| 9 # * Redistributions in binary form must reproduce the above | 9 # * Redistributions in binary form must reproduce the above |
| 10 # copyright notice, this list of conditions and the following disclaimer | 10 # copyright notice, this list of conditions and the following disclaimer |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 43 platform = platform_str | 43 platform = platform_str |
| 44 if windows_version_tuple: | 44 if windows_version_tuple: |
| 45 getwindowsversion = lambda x: windows_version_tuple | 45 getwindowsversion = lambda x: windows_version_tuple |
| 46 | 46 |
| 47 return FakeSysModule() | 47 return FakeSysModule() |
| 48 | 48 |
| 49 | 49 |
| 50 def fake_platform(mac_version_string='10.6.3', release_string='bar', linux_versi
on='trusty'): | 50 def fake_platform(mac_version_string='10.6.3', release_string='bar', linux_versi
on='trusty'): |
| 51 | 51 |
| 52 class FakePlatformModule(object): | 52 class FakePlatformModule(object): |
| 53 |
| 53 def mac_ver(self): | 54 def mac_ver(self): |
| 54 return tuple([mac_version_string, tuple(['', '', '']), 'i386']) | 55 return tuple([mac_version_string, tuple(['', '', '']), 'i386']) |
| 55 | 56 |
| 56 def linux_distribution(self): | 57 def linux_distribution(self): |
| 57 return tuple([None, None, linux_version]) | 58 return tuple([None, None, linux_version]) |
| 58 | 59 |
| 59 def platform(self): | 60 def platform(self): |
| 60 return 'foo' | 61 return 'foo' |
| 61 | 62 |
| 62 def release(self): | 63 def release(self): |
| 63 return release_string | 64 return release_string |
| 64 | 65 |
| 65 return FakePlatformModule() | 66 return FakePlatformModule() |
| 66 | 67 |
| 67 | 68 |
| 68 def fake_executive(output=None): | 69 def fake_executive(output=None): |
| 69 if output: | 70 if output: |
| 70 return MockExecutive2(output=output) | 71 return MockExecutive2(output=output) |
| 71 return MockExecutive2(exception=SystemError) | 72 return MockExecutive2(exception=SystemError) |
| 72 | 73 |
| 73 | 74 |
| 74 class TestPlatformInfo(unittest.TestCase): | 75 class TestPlatformInfo(unittest.TestCase): |
| 76 |
| 75 def make_info(self, sys_module=None, platform_module=None, filesystem_module
=None, executive=None): | 77 def make_info(self, sys_module=None, platform_module=None, filesystem_module
=None, executive=None): |
| 76 return PlatformInfo(sys_module or fake_sys(), platform_module or fake_pl
atform(), filesystem_module or MockFileSystem(), executive or fake_executive()) | 78 return PlatformInfo(sys_module or fake_sys(), platform_module or fake_pl
atform(), filesystem_module or MockFileSystem(), executive or fake_executive()) |
| 77 | 79 |
| 78 def test_real_code(self): | 80 def test_real_code(self): |
| 79 # This test makes sure the real (unmocked) code actually works. | 81 # This test makes sure the real (unmocked) code actually works. |
| 80 info = PlatformInfo(sys, platform, FileSystem(), Executive()) | 82 info = PlatformInfo(sys, platform, FileSystem(), Executive()) |
| 81 self.assertNotEquals(info.os_name, '') | 83 self.assertNotEquals(info.os_name, '') |
| 82 self.assertNotEquals(info.os_version, '') | 84 self.assertNotEquals(info.os_version, '') |
| 83 self.assertNotEquals(info.display_name(), '') | 85 self.assertNotEquals(info.display_name(), '') |
| 84 self.assertTrue(info.is_mac() or info.is_win() or info.is_linux() or inf
o.is_freebsd()) | 86 self.assertTrue(info.is_mac() or info.is_win() or info.is_linux() or inf
o.is_freebsd()) |
| (...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 206 self.assertEqual(info.total_bytes_memory(), 1234) | 208 self.assertEqual(info.total_bytes_memory(), 1234) |
| 207 | 209 |
| 208 info = self.make_info(fake_sys('win32', tuple([6, 1, 7600]))) | 210 info = self.make_info(fake_sys('win32', tuple([6, 1, 7600]))) |
| 209 self.assertIsNone(info.total_bytes_memory()) | 211 self.assertIsNone(info.total_bytes_memory()) |
| 210 | 212 |
| 211 info = self.make_info(fake_sys('linux2')) | 213 info = self.make_info(fake_sys('linux2')) |
| 212 self.assertIsNone(info.total_bytes_memory()) | 214 self.assertIsNone(info.total_bytes_memory()) |
| 213 | 215 |
| 214 info = self.make_info(fake_sys('freebsd9')) | 216 info = self.make_info(fake_sys('freebsd9')) |
| 215 self.assertIsNone(info.total_bytes_memory()) | 217 self.assertIsNone(info.total_bytes_memory()) |
| OLD | NEW |