Accept strings as keys for hash

Author: PoTe

lib/hashifiable.rb

@@ -9,7 +9,7 @@ def hashify(*args)
 
       args.each do |argument|
         case argument
-        when Symbol
+        when Symbol, String
           ## Calls the specified method on the object and stores it
           ## under the method name.
           hash_representation[argument] = self.send(argument)

spec/hashifiable_spec.rb

@@ -1,16 +1,18 @@
 require 'spec_helper'
 
-class TestUser < Struct.new(:id, :name, :credit_card, :secret_token)
+class TestUser < Struct.new(:id, :name, :credit_card, :secret_token, :quote)
   extend Hashifiable
   hashify :id,
           :name,
+          'quote',
+          'random' => Proc.new { [1, 2, 3].sample },
           :two_times_two => Proc.new { 2 * 2 },
           :encrypted_token => Proc.new { secret_token + ' secret sauce' },
           :lambdas_at_work => ->() { 2 * 2 }
 end
 
 describe Hashifiable do
-  let(:user) { TestUser.new(1, 'pote', '1123123241241', 'i2j34i2j34302843') }
+  let(:user) { TestUser.new(1, 'pote', '1123123241241', 'i2j34i2j34302843', "It's bigger on the inside!") }
 
   it 'should have a to_h method' do
     user.should respond_to(:to_h)
@@ -36,10 +38,18 @@ class TestUser < Struct.new(:id, :name, :credit_card, :secret_token)
     user.to_h[:id].should == 2
   end
 
+  it 'should include strings as well as symbols on the hash representation' do
+    user.to_h['quote'].should == "It's bigger on the inside!"
+  end
+
   it 'should include procs in the hash representation' do
     user.to_h.keys.should include(:two_times_two)
   end
 
+  it 'should allow procs to have strings as keys' do
+    user.to_h['random'].class.should == Fixnum
+  end
+
   it 'should return the output of the function and not a proc' do
     user.to_h[:two_times_two].should == 4
   end