1 files changed, 34 insertions, 64 deletions
diff --git a/test/functional/my_controller_test.rb b/test/functional/my_controller_test.rb
index ff509674b..cac6a14ad 100644
--- a/test/functional/my_controller_test.rb
+++ b/test/functional/my_controller_test.rb
@@ -137,71 +137,41 @@ class MyControllerTest < ActionController::TestCase
     assert_equal ['documents', 'calendar', 'latestnews'], User.find(2).pref[:my_page_layout]['left']
   end
 
-  context "POST to reset_rss_key" do
-    context "with an existing rss_token" do
-      setup do
-        @previous_token_value = User.find(2).rss_key # Will generate one if it's missing
-        post :reset_rss_key
-      end
-
-      should "destroy the existing token" do
-        assert_not_equal @previous_token_value, User.find(2).rss_key
-      end
-
-      should "create a new token" do
-        assert User.find(2).rss_token
-      end
-
-      should_set_the_flash_to /reset/
-      should_redirect_to('my account') {'/my/account' }
-    end
-
-    context "with no rss_token" do
-      setup do
-        assert_nil User.find(2).rss_token
-        post :reset_rss_key
-      end
-
-      should "create a new token" do
-        assert User.find(2).rss_token
-      end
-
-      should_set_the_flash_to /reset/
-      should_redirect_to('my account') {'/my/account' }
-    end
+  def test_reset_rss_key_with_existing_key
+    @previous_token_value = User.find(2).rss_key # Will generate one if it's missing
+    post :reset_rss_key
+
+    assert_not_equal @previous_token_value, User.find(2).rss_key
+    assert User.find(2).rss_token
+    assert_match /reset/, flash[:notice]
+    assert_redirected_to '/my/account'
+  end
+
+  def test_reset_rss_key_without_existing_key
+    assert_nil User.find(2).rss_token
+    post :reset_rss_key
+
+    assert User.find(2).rss_token
+    assert_match /reset/, flash[:notice]
+    assert_redirected_to '/my/account'
   end
 
-  context "POST to reset_api_key" do
-    context "with an existing api_token" do
-      setup do
-        @previous_token_value = User.find(2).api_key # Will generate one if it's missing
-        post :reset_api_key
-      end
-
-      should "destroy the existing token" do
-        assert_not_equal @previous_token_value, User.find(2).api_key
-      end
-
-      should "create a new token" do
-        assert User.find(2).api_token
-      end
-
-      should_set_the_flash_to /reset/
-      should_redirect_to('my account') {'/my/account' }
-    end
-
-    context "with no api_token" do
-      setup do
-        assert_nil User.find(2).api_token
-        post :reset_api_key
-      end
-
-      should "create a new token" do
-        assert User.find(2).api_token
-      end
-
-      should_set_the_flash_to /reset/
-      should_redirect_to('my account') {'/my/account' }
-    end
+  def test_reset_api_key_with_existing_key
+    @previous_token_value = User.find(2).api_key # Will generate one if it's missing
+    post :reset_api_key
+
+    assert_not_equal @previous_token_value, User.find(2).api_key
+    assert User.find(2).api_token
+    assert_match /reset/, flash[:notice]
+    assert_redirected_to '/my/account'
+  end
+
+  def test_reset_api_key_without_existing_key
+    assert_nil User.find(2).api_token
+    post :reset_api_key
+
+    assert User.find(2).api_token
+    assert_match /reset/, flash[:notice]
+    assert_redirected_to '/my/account'
   end
 end