1 files changed, 48 insertions, 0 deletions
diff --git a/lib/extensions/testing.rb b/lib/extensions/testing.rb
new file mode 100644
index 0000000..8f7e73c
--- /dev/null
+++ b/lib/extensions/testing.rb
@@ -0,0 +1,48 @@
+module LeapWebCore
+  module AssertResponses
+
+    # response that works with different TestCases:
+    # ActionController::TestCase has @response
+    # ActionDispatch::IntegrationTest has @response
+    # Rack::Test::Methods defines last_response
+    def get_response
+      @response || last_response
+    end
+
+    def assert_attachement_filename(name)
+      assert_equal %Q(attachment; filename="#{name}"),
+        get_response.headers["Content-Disposition"]
+    end
+
+    def json_response
+      response = JSON.parse(get_response.body)
+      response.respond_to?(:with_indifferent_access) ?
+        response.with_indifferent_access :
+        response
+    end
+
+    def assert_json_response(object)
+      assert_equal 'application/json',
+        get_response.content_type.to_s.split(';').first
+      if object.is_a? Hash
+        object.stringify_keys! if object.respond_to? :stringify_keys!
+        assert_equal object, json_response
+      else
+        assert_equal object.to_json, get_response.body
+      end
+    end
+
+    def assert_json_error(object)
+      object.stringify_keys! if object.respond_to? :stringify_keys!
+      assert_json_response :errors => object
+    end
+  end
+end
+
+class ::ActionController::TestCase
+  include LeapWebCore::AssertResponses
+end
+
+class ::ActionDispatch::IntegrationTest
+  include LeapWebCore::AssertResponses
+end