Fixed hash empty value bug w/ spec

unclebob · Feb 29, 2012 · 0068c9c · 0068c9c
1 parent 1213e54
commit 0068c9c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lib/rubyslim/table_to_hash_converter.rb b/lib/rubyslim/table_to_hash_converter.rb
@@ -27,6 +27,8 @@ def self.create_hash_from_rows(rows)
   def self.add_row_to_hash(hash, row)
     columns = row.get_elements('td')
     raise ArgumentError if columns.size != 2
+    # Handle empty values.
+    columns[1].text = '' if columns[1].text == nil
     hash[columns[0].text.strip.to_sym] = columns[1].text.strip
   end
 end
diff --git a/spec/table_to_hash_converter_spec.rb b/spec/table_to_hash_converter_spec.rb
@@ -23,7 +23,7 @@ def shouldNotChange(string)
     ["<table><tr><td>name</td><td>bob</td></tr></table>", {:name=>"bob"}],
     [" <table> <tr> <td> name </td> <td> bob </td> </tr> </table> ", {:name=>"bob"}],
     ["<table><tr><td>name</td><td>bob</td></tr><tr><td>addr</td><td>here</td></tr></table>", {:name=>'bob', :addr=>'here'}],
-    ["<table><tr><td>name</td><td> </td></tr></table>", {:name=>""}],
+    ["<table class=\"hash_table\"><tr><td>name</td><td></td></tr></table>", {:name=>""}],
   ].each do |table, hash|
     it "should match #{table} to #{hash}" do
       TableToHashConverter.convert(table).should == hash