1 files changed, 11 insertions, 0 deletions
diff --git a/lib/generators/nifty/scaffold/templates/tests/testunit/actions/update.rb b/lib/generators/nifty/scaffold/templates/tests/testunit/actions/update.rb
new file mode 100644
index 0000000..b588bfc
--- /dev/null
+++ b/lib/generators/nifty/scaffold/templates/tests/testunit/actions/update.rb
@@ -0,0 +1,11 @@
+  def test_update_invalid
+    <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+    put :update, :id => <%= class_name %>.first
+    assert_template 'edit'
+  end
+
+  def test_update_valid
+    <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+    put :update, :id => <%= class_name %>.first
+    assert_redirected_to <%= item_path_for_test('url') %>
+  end