[#1659] test package previous button

ckan · Jun 9, 2014 · 96e600e · 96e600e
1 parent 1fe7338
commit 96e600e
Showing 1 changed file with 63 additions and 23 deletions.
diff --git a/ckan/new_tests/controllers/test_package.py b/ckan/new_tests/controllers/test_package.py
@@ -5,59 +5,99 @@
 import ckan.new_tests.helpers as helpers
 import ckan.new_tests.factories as factories
 
+webtest_submit = helpers.webtest_submit
 
 class TestPackageController(helpers.FunctionalTestBaseClass):
     def test_create_form_renders(self):
         user = factories.Sysadmin()
-        extra_environ = {
-            'REMOTE_USER': user['name'].encode('ascii'),
-        }
+        env = {'REMOTE_USER': user['name'].encode('ascii')}
         response = self.app.get(
             url=url_for(controller='package', action='new'),
-            extra_environ=extra_environ,
+            extra_environ=env,
         )
         assert_true('dataset-edit' in response.forms)
 
     def test_create_form_next_button_works(self):
         user = factories.Sysadmin()
-        extra_environ = {
-            'REMOTE_USER': user['name'].encode('ascii'),
-        }
+        env = {'REMOTE_USER': user['name'].encode('ascii')}
         response = self.app.get(
             url=url_for(controller='package', action='new'),
-            extra_environ=extra_environ,
+            extra_environ=env,
         )
         form = response.forms['dataset-edit']
         form['name'] = u'next-button-works'
         form['title'] = u'Next button works'
 
-        response = form.submit(
-            'save',
-            status=302,
-            extra_environ=extra_environ,
-        )
+        response = webtest_submit(form, 'save', status=302, extra_environ=env)
         assert_true('Location' in dict(response.headers))
 
     def test_create_form_resource_form_renders(self):
         user = factories.Sysadmin()
-        extra_environ = {
-            'REMOTE_USER': user['name'].encode('ascii'),
-        }
+        env = {'REMOTE_USER': user['name'].encode('ascii')}
         response = self.app.get(
             url=url_for(controller='package', action='new'),
-            extra_environ=extra_environ,
+            extra_environ=env,
         )
         form = response.forms['dataset-edit']
         form['name'] = u'resource-form-renders'
         form['title'] = u'Resource form renders'
 
-        response = form.submit(
-            'save',
-            status=302,
-            extra_environ=extra_environ,
-        )
+        response = webtest_submit(form, 'save', status=302, extra_environ=env)
         response = self.app.get(
             url=dict(response.headers)['Location'],
-            extra_environ=extra_environ,
+            extra_environ=env,
         )
         assert_true('resource-edit' in response.forms)
+
+    def test_create_form_previous_button_works(self):
+        user = factories.Sysadmin()
+        env = {'REMOTE_USER': user['name'].encode('ascii')}
+        response = self.app.get(
+            url=url_for(controller='package', action='new'),
+            extra_environ=env,
+        )
+        form = response.forms['dataset-edit']
+        form['name'] = u'previous-button-works'
+        form['title'] = u'Previous button works'
+
+        response = webtest_submit(form, 'save', status=302, extra_environ=env)
+        response = self.app.get(
+            url=dict(response.headers)['Location'],
+            extra_environ=env,
+        )
+        form = response.forms['resource-edit']
+        response = webtest_submit(form, 'save', value='go-dataset',
+            status=302, extra_environ=env)
+        response = self.app.get(
+            url=dict(response.headers)['Location'],
+            extra_environ=env,
+        )
+        assert_true('dataset-edit' in response.forms)
+
+    def test_create_form_previous_button_populates_form(self):
+        user = factories.Sysadmin()
+        env = {'REMOTE_USER': user['name'].encode('ascii')}
+        response = self.app.get(
+            url=url_for(controller='package', action='new'),
+            extra_environ=env,
+        )
+        form = response.forms['dataset-edit']
+        form['name'] = u'previous-button-populates-form'
+        form['title'] = u'Previous button populates form'
+
+        response = webtest_submit(form, 'save', status=302, extra_environ=env)
+        response = self.app.get(
+            url=dict(response.headers)['Location'],
+            extra_environ=env,
+        )
+        form = response.forms['resource-edit']
+        response = webtest_submit(form, 'save', value='go-dataset',
+            status=302, extra_environ=env)
+        response = self.app.get(
+            url=dict(response.headers)['Location'],
+            extra_environ=env,
+        )
+        form = response.forms['dataset-edit']
+        assert_true('title' in form.fields)
+        assert_equal(form['title'].value, u'Previous button populates form')
+        assert_equal(form['name'].value, u'previous-button-populates-form')