lib/serialization_helper.rb in yaml_db-0.2.0 vs lib/serialization_helper.rb in yaml_db-0.2.1

@@ -113,16 +113,20 @@
 
     def self.convert_booleans(records, columns)
       records.each do |record|
         columns.each do |column|
           next if is_boolean(record[column])
-          record[column] = (record[column] == 't' or record[column] == '1')
+          record[column] = convert_boolean(record[column])
         end
       end
       records
     end
 
+    def self.convert_boolean(value)
+      ['t', '1', true, 1].include?(value)
+    end
+
     def self.boolean_columns(table)
       columns = ActiveRecord::Base.connection.columns(table).reject { |c| silence_warnings { c.type != :boolean } }
       columns.map { |c| c.name }
     end
 
@@ -190,6 +194,6 @@
       ActiveRecord::Base.connection.select_one("SELECT COUNT(*) FROM #{SerializationHelper::Utils.quote_table(table)}").values.first.to_i
     end
 
   end
 
-end
\ No newline at end of file
+end