Merge pull request #121 from seancribbs/bugfix/date-header-required

Fixes missing Date header for relevant responses

lib/webmachine/decision/fsm.rb

@@ -70,6 +70,7 @@ def respond(code, headers={})
         end
 
         ensure_content_length
+        ensure_date_header
       end
 
       # When tracing is disabled, this does nothing.

lib/webmachine/decision/helpers.rb

@@ -1,4 +1,5 @@
 require 'stringio'
+require 'time'
 require 'webmachine/streaming'
 require 'webmachine/media_type'
 require 'webmachine/quoted_string'
@@ -102,6 +103,13 @@ def ensure_content_length
         end
       end
 
+      # Ensures that responses have an appropriate Date header
+      def ensure_date_header
+        if (200..499).include?(response.code)
+          response.headers['Date'] ||= Time.now.httpdate
+        end
+      end
+
       # Sets the Content-Length header on the response
       def set_content_length
         if response.body.respond_to?(:bytesize)

spec/webmachine/decision/flow_spec.rb

@@ -11,6 +11,15 @@
   let(:default_resource) { resource_with }
   let(:missing_resource) { missing_resource_with }
 
+  # http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.18:
+  # Origin servers MUST include a Date header field in all responses
+  # ... [except 1xx or 5xx]
+  after(:each) do
+    unless response.code < 200 || response.code >= 500
+      response.headers.should have_key('Date')
+    end
+  end
+
   def resource_with(&block)
     klass = Class.new(Webmachine::Resource) do
       def to_html; "test resource"; end