feature: generate alias_method

abak-press · Jun 1, 2016 · ddfa482 · ddfa482
1 parent 8768c09
commit ddfa482
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lib/treasury/fields/extractor.rb b/lib/treasury/fields/extractor.rb
@@ -34,6 +34,7 @@ def extract_object(params)
 
       def extract_attribute_name(name)
         class_attribute :attribute_name, instance_writer: false
+        singleton_class.send(:alias_method, "extract_#{name}".to_sym, :extract_object)
 
         self.attribute_name = name
       end

diff --git a/spec/lib/treasury/fields/extractor_spec.rb b/spec/lib/treasury/fields/extractor_spec.rb
@@ -30,6 +30,10 @@
       it { expect { class_with_extractor.extract_object(object: {another_object: 1}) }.to raise_error(ArgumentError) }
     end
 
+    context 'when call alias method' do
+      it { expect(class_with_extractor.extract_user(object: {user_id: 1})).to eq 1 }
+    end
+
     it 'return object id' do
       user = double(:user, id: 1)