lib/restspec/values/status_code.rb in restspec-0.0.4 vs lib/restspec/values/status_code.rb in restspec-0.1

@@ -1,13 +1,28 @@
 module Restspec
   module Values
-    class StatusCode < Struct.new(:number_or_symbol)
+    # A value object that transforms a http status code (201) or
+    # a symbol with the status code message (:created) to a simple
+    # number (201).
+    class StatusCode
+      def initialize(number_or_symbol)
+        self.number_or_symbol = number_or_symbol
+      end
+
+      # @example
+      #   StatusCode.new(201).value # 201
+      #   StatusCode.new(:created).value # 201
+      # @return [Fixnum] the status code
       def value
         if number_or_symbol.is_a?(Symbol)
           Rack::Utils::SYMBOL_TO_STATUS_CODE.fetch(number_or_symbol)
         else
           number_or_symbol
         end
       end
+
+      private
+
+      attr_accessor :number_or_symbol
     end
   end
 end