value_object_ar 0.0.2 → 0.0.3

data/lib/methods.rb

@@ -20,9 +20,11 @@ module ValueObjectAR
       end
     
       def define_value_object_ar_writer(attribute, klass)
-        define_method("#{attribute}=") do |value|
-          if value.is_a?(klass) 
-            value = value.send(attribute)
+        define_method("#{attribute}=") do |value_or_object|
+          value = if value_or_object.is_a?(klass) 
+            value_or_object.value
+          else
+            value_or_object
           end
           attributes[attribute.to_s] = value
         end

data/test/value_object_ar_test.rb

@@ -0,0 +1,56 @@
+require "test/unit"
+require_relative "../lib/value_object_ar"
+
+class TestValueObjectAr < Test::Unit::TestCase
+
+  def setup
+    @product = Product.new(currency: 'DKK')
+  end
+
+  def test_call_methods_on_value_object
+    assert_equal @product.currency.hipstier_than_bitcoin?, "DKK? Probably not"
+  end
+  
+  def test_set_with_value
+    @product.currency = 'EUR'
+    assert_equal @product.currency.hipstier_than_bitcoin?, "EUR? Probably not"
+  end
+  
+  def test_set_with_value_object
+    @product.currency = CurrencyValueObject.new('USD')
+    assert_equal @product.currency.hipstier_than_bitcoin?, "USD? Probably not"
+  end
+
+end
+
+class CurrencyValueObject
+  
+  attr_accessor :value
+  
+  def initialize(value)
+    @value = value
+  end
+  
+  def hipstier_than_bitcoin?
+    "#{value}? Probably not"
+  end
+  
+end
+
+class Product
+  
+  include ValueObjectAR::Methods
+  value_object :currency, CurrencyValueObject
+
+  # ActiveRecord behaviour below:
+  attr_accessor :attributes
+    
+  def initialize(options={})
+    options.each {|k,v| self.public_send("#{k}=",v)}
+  end
+  
+  def attributes
+    @attributes ||= {'currency' => nil}
+  end
+  
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: value_object_ar
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -51,6 +51,7 @@ extra_rdoc_files: []
 files:
 - lib/methods.rb
 - lib/value_object_ar.rb
+- test/value_object_ar_test.rb
 homepage: http://github.com/kasperbn/value_object_ar
 licenses: []
 post_install_message: 
@@ -76,4 +77,5 @@ rubygems_version: 1.8.25
 signing_key: 
 specification_version: 3
 summary: Value objects for your active models
-test_files: []
+test_files:
+- test/value_object_ar_test.rb