Added sumbol http status codes.

NOX73 · Dec 31, 2012 · fd6995d · fd6995d
1 parent 60b4451
commit fd6995d
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 0 deletions.
diff --git a/celluloid-http.gemspec b/celluloid-http.gemspec
@@ -17,6 +17,7 @@ Gem::Specification.new do |gem|
   gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
   gem.require_paths = ["lib"]
 
+  gem.add_runtime_dependency('rack')
   gem.add_runtime_dependency('celluloid-io')
   gem.add_runtime_dependency('http_parser.rb')
 

diff --git a/lib/celluloid-http.rb b/lib/celluloid-http.rb
@@ -2,6 +2,7 @@
 require "uri"
 require "http_parser.rb"
 require "celluloid/io"
+require "rack"
 
 module Celluloid
   module Http

diff --git a/lib/celluloid-http/response.rb b/lib/celluloid-http/response.rb
@@ -1,6 +1,8 @@
 class Celluloid::Http::Response
   extend Forwardable
 
+  STATUS_CODES          = Rack::Utils::HTTP_STATUS_CODES
+
   def_delegators :parser, :<<
 
   attr_writer :status
@@ -35,4 +37,12 @@ def parser
     @parser ||= Http::Parser.new(self)
   end
 
+  def reason
+    STATUS_CODES[status]
+  end
+
+  def sym_status
+    reason.downcase.gsub(/\s|-/, '_').to_sym
+  end
+
 end
diff --git a/test/unit/response_test.rb b/test/unit/response_test.rb
@@ -27,5 +27,17 @@ def test_parse_http_response
     assert_equal "en", response.headers["Content-Language"]
   end
 
+  def test_symbol_ok_http_status
+    response = Celluloid::Http::Response.new(200)
+
+    assert_equal :ok, response.sym_status
+  end
+
+  def test_symbol_not_found_http_status
+    response = Celluloid::Http::Response.new(404)
+
+    assert_equal :not_found, response.sym_status
+  end
+
 
 end