yaml_db_anonymizer 0.0.1 → 0.0.2

data/lib/yaml_db/anonymizer.rb

@@ -36,7 +36,11 @@ module YamlDb
       def anonymize(table, column, value)
         return value unless self.definition.key?(table.to_s) and self.definition[table.to_s].key?(column.to_s)
         block_or_value = self.definition[table.to_s][column.to_s]
-        block_or_value.is_a?(Proc) ? block_or_value.call(value) : block_or_value
+        if block_or_value.is_a?(Proc)
+          block_or_value.arity == 1 ? block_or_value.call(value) : block_or_value.call
+        else
+          block_or_value
+        end
       end
     end
   end

data/lib/yaml_db/anonymizer/dump.rb

@@ -3,7 +3,9 @@ module YamlDb
     class Dump < ::YamlDb::Dump
 
       def self.dump_table_records(io, table)
+        return super(io, table) unless Anonymizer.definition.is_a? Hash
         return if Anonymizer.definition[table.to_s] == :truncate
+
         table_record_header(io)
 
         column_names = table_column_names(table)
@@ -23,7 +25,7 @@ module YamlDb
       end
 
       def self.dump_table_columns(io, table)
-        super(io, table) unless Anonymizer.definition[table.to_s] == :truncate
+        super(io, table) unless Anonymizer.definition.is_a?(Hash) && Anonymizer.definition[table.to_s] == :truncate
       end
 
     end

data/lib/yaml_db/anonymizer/version.rb

@@ -1,5 +1,5 @@
 module YamlDb
   module Anonymizer
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/spec/yaml_db/anonymizer_spec.rb

@@ -8,6 +8,7 @@ describe YamlDb::Anonymizer do
           remove :a
           replace :b, with: 0
           replace :c, with: lambda { |val| "Anonymized #{val}" }
+          replace :d, with: lambda { "Anonymized value" }
         end
       end
     end
@@ -23,6 +24,10 @@ describe YamlDb::Anonymizer do
     it "should anonymize with blocks" do
       YamlDb::Anonymizer.anonymize(:mytable, :c, "bar").should == "Anonymized bar"
     end
+
+    it "should anonymize with blocks" do
+      YamlDb::Anonymizer.anonymize(:mytable, :d, "bar").should == "Anonymized value"
+    end
   end
 
   context 'export' do
@@ -87,5 +92,24 @@ EOYAML
       end
     end
   end
+
+  context 'fallback' do
+    let(:io)    { double('io') }
+    let(:table) { double('table') }
+
+    before do
+      YamlDb::Anonymizer.definition = nil
+    end
+
+    it 'falls back to YamlDb::Dump.dump_table_records if definition is missing' do
+      YamlDb::Dump.should_receive(:dump_table_records).with(io, table)
+      YamlDb::Anonymizer::Dump.dump_table_records(io, table)
+    end
+
+    it 'falls back to YamlDb::Dump.dump_table_columns if definition is missing' do
+      YamlDb::Dump.should_receive(:dump_table_columns).with(io, table)
+      YamlDb::Anonymizer::Dump.dump_table_columns(io, table)
+    end
+  end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: yaml_db_anonymizer
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-23 00:00:00.000000000 Z
+date: 2012-11-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml_db