Add tests

ckan · Aug 6, 2018 · 469d068 · 469d068
1 parent 8210da1
commit 469d068
Showing 1 changed file with 36 additions and 3 deletions.
diff --git a/ckan/tests/lib/dictization/test_model_dictize.py b/ckan/tests/lib/dictization/test_model_dictize.py
@@ -5,7 +5,7 @@
 
 from nose.tools import assert_equal
 
-from ckan.lib.dictization import model_dictize
+from ckan.lib.dictization import model_dictize, model_save
 from ckan import model
 from ckan.lib import search
 
@@ -401,8 +401,7 @@ def test_package_dictize_resource(self):
 
         result = model_dictize.package_dictize(dataset_obj, context)
 
-        assert_equal_for_keys(result['resources'][0], resource,
-                              'name', 'url')
+        assert_equal_for_keys(result['resources'][0], resource, 'name', 'url')
         expected_dict = {
             u'cache_last_updated': None,
             u'cache_url': None,
@@ -422,6 +421,40 @@ def test_package_dictize_resource(self):
         }
         self.assert_equals_expected(expected_dict, result['resources'][0])
 
+    def test_package_dictize_resource_upload_and_striped(self):
+        dataset = factories.Dataset()
+        resource = factories.Resource(package=dataset['id'],
+                                      name='test_pkg_dictize',
+                                      url_type='upload',
+                                      url='some_filename.csv')
+
+        context = {'model': model, 'session': model.Session}
+
+        result = model_save.resource_dict_save(resource, context)
+
+        expected_dict = {
+            u'url': u'some_filename.csv',
+            u'url_type': u'upload'
+        }
+        assert expected_dict['url'] == result.url
+
+    def test_package_dictize_resource_upload_with_url_and_striped(self):
+        dataset = factories.Dataset()
+        resource = factories.Resource(package=dataset['id'],
+                                      name='test_pkg_dictize',
+                                      url_type='upload',
+                                      url='http://some_filename.csv')
+
+        context = {'model': model, 'session': model.Session}
+
+        result = model_save.resource_dict_save(resource, context)
+
+        expected_dict = {
+            u'url': u'some_filename.csv',
+            u'url_type': u'upload'
+        }
+        assert expected_dict['url'] == result.url
+
     def test_package_dictize_tags(self):
         dataset = factories.Dataset(tags=[{'name': 'fish'}])
         dataset_obj = model.Package.get(dataset['id'])