| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright 2015 The Chromium Authors. All rights reserved. | 2 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """ | 6 """ |
| 7 Unit tests for the contents of shared_prefs.py (mostly SharedPrefs). | 7 Unit tests for the contents of shared_prefs.py (mostly SharedPrefs). |
| 8 """ | 8 """ |
| 9 | 9 |
| 10 import logging | 10 import logging |
| 11 import os | 11 import os |
| 12 import sys | 12 import sys |
| 13 import unittest | 13 import unittest |
| 14 | 14 |
| 15 from devil.android import device_utils | 15 from devil.android import device_utils |
| 16 from devil.android.sdk import shared_prefs | 16 from devil.android.sdk import shared_prefs |
| 17 from pylib import constants | 17 from pylib import constants |
| 18 | 18 |
| 19 sys.path.append(os.path.join( | 19 sys.path.append(os.path.join( |
| 20 constants.DIR_SOURCE_ROOT, 'third_party', 'pymock')) | 20 constants.DIR_SOURCE_ROOT, 'third_party', 'pymock')) |
| 21 import mock | 21 import mock # pylint: disable=import-error |
| 22 | 22 |
| 23 | 23 |
| 24 def MockDeviceWithFiles(files=None): | 24 def MockDeviceWithFiles(files=None): |
| 25 if files is None: | 25 if files is None: |
| 26 files = {} | 26 files = {} |
| 27 | 27 |
| 28 def file_exists(path): | 28 def file_exists(path): |
| 29 return path in files | 29 return path in files |
| 30 | 30 |
| 31 # pylint: disable=unused-argument |
| 31 def write_file(path, contents, **_kwargs): | 32 def write_file(path, contents, **_kwargs): |
| 32 files[path] = contents | 33 files[path] = contents |
| 33 | 34 |
| 35 # pylint: disable=unused-argument |
| 34 def read_file(path, **_kwargs): | 36 def read_file(path, **_kwargs): |
| 35 return files[path] | 37 return files[path] |
| 36 | 38 |
| 37 device = mock.MagicMock(spec=device_utils.DeviceUtils) | 39 device = mock.MagicMock(spec=device_utils.DeviceUtils) |
| 38 device.FileExists = mock.Mock(side_effect=file_exists) | 40 device.FileExists = mock.Mock(side_effect=file_exists) |
| 39 device.WriteFile = mock.Mock(side_effect=write_file) | 41 device.WriteFile = mock.Mock(side_effect=write_file) |
| 40 device.ReadFile = mock.Mock(side_effect=read_file) | 42 device.ReadFile = mock.Mock(side_effect=read_file) |
| 41 return device | 43 return device |
| 42 | 44 |
| 43 | 45 |
| (...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 160 | 162 |
| 161 self.assertEquals(self.device.WriteFile.call_args_list, []) # did not write | 163 self.assertEquals(self.device.WriteFile.call_args_list, []) # did not write |
| 162 with shared_prefs.SharedPrefs( | 164 with shared_prefs.SharedPrefs( |
| 163 self.device, 'com.some.package', 'prefs.xml') as prefs: | 165 self.device, 'com.some.package', 'prefs.xml') as prefs: |
| 164 # contents were not modified | 166 # contents were not modified |
| 165 self.assertEquals(prefs.AsDict(), self.expected_data) | 167 self.assertEquals(prefs.AsDict(), self.expected_data) |
| 166 | 168 |
| 167 if __name__ == '__main__': | 169 if __name__ == '__main__': |
| 168 logging.getLogger().setLevel(logging.DEBUG) | 170 logging.getLogger().setLevel(logging.DEBUG) |
| 169 unittest.main(verbosity=2) | 171 unittest.main(verbosity=2) |
| OLD | NEW |