Fix tests

This commit is contained in:
Aitor Magán 2014-10-21 17:50:16 +02:00
parent ebfb70331a
commit 8d9208544f
2 changed files with 6 additions and 4 deletions

View File

@ -199,7 +199,7 @@ class PrivateDatasets(p.SingletonPlugin, tk.DefaultDatasetForm):
'use_cache': False}, 'use_cache': False},
{'id': package_id}) {'id': package_id})
# Prevent acquired datasets jumping to the first position # Prevent acquired datasets jumping to the first position
new_pkg_dict['metadata_modified'] = new_pkg_dict['revision_timestamp'] new_pkg_dict['metadata_modified'] = new_pkg_dict.get('revision_timestamp', '')
self.indexer.update_dict(new_pkg_dict) self.indexer.update_dict(new_pkg_dict)
return pkg_dict return pkg_dict

View File

@ -282,8 +282,10 @@ class PluginTest(unittest.TestCase):
package_id = 'package_id' package_id = 'package_id'
# Configure mocks # Configure mocks
default_dict = {'a': '0', 'b': 1, 'm': True} default_dict = {'a': '0', 'b': 1, 'm': True, 'revision_timestamp': '2000'}
package_show = MagicMock(return_value=default_dict) expected_dict = default_dict.copy()
expected_dict['metadata_modified'] = default_dict['revision_timestamp']
package_show = MagicMock(return_value=default_dict.copy())
plugin.tk.get_action = MagicMock(return_value=package_show) plugin.tk.get_action = MagicMock(return_value=package_show)
# Each time 'AllowedUser' is called, we must get a new instance # Each time 'AllowedUser' is called, we must get a new instance
@ -336,7 +338,7 @@ class PluginTest(unittest.TestCase):
self.assertEquals(0, self.privateDatasets.indexer.update_dict.call_count) self.assertEquals(0, self.privateDatasets.indexer.update_dict.call_count)
else: else:
# Check that the cache has been updated # Check that the cache has been updated
self.privateDatasets.indexer.update_dict.assert_called_once_with(default_dict) self.privateDatasets.indexer.update_dict.assert_called_once_with(expected_dict)
@parameterized.expand([ @parameterized.expand([
# One element # One element