hash_db 0.0.2 → 0.0.3

data/lib/hash_db/model/association.rb

@@ -8,14 +8,20 @@ module HashDB
         klass.keys foreign_key
 
         define_method "#{name}" do
-          base_id = id
           @attributes[name] ||= begin
+            base_id = id
             [].tap do |array|
               array.define_singleton_method :<< do |object|
                 object.send("#{foreign_key}=", base_id)
                 super object
               end
 
+              array.define_singleton_method :new do |*args|
+                klass.new(*args).tap do |object|
+                  array << object
+                end
+              end
+
               array.define_singleton_method :create do |*args|
                 klass.create(*args).tap do |object|
                   array << object

data/lib/hash_db/version.rb

@@ -1,3 +1,3 @@
 module HashDB
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/hash_db/model/association_spec.rb

@@ -2,16 +2,24 @@ require "spec_helper"
 
 describe HashDB::Model do
   context ".has_many" do
-    it "should allow assigning and retrieving objects" do
-      post = Class.new do
+    let :post do
+      Class.new do
         include HashDB::Model
       end
+    end
 
-      user = Class.new do
+    let :user do
+      # Class.new creates a new scope, where we cannot access the let(:post)
+      # above. This is a workaround for it.
+      # TODO: Find a better way to do this.
+      _post = post
+      Class.new do
         include HashDB::Model
-        has_many :posts, foreign_key: :user_id, class: post
+        has_many :posts, foreign_key: :user_id, class: _post
       end
+    end
 
+    it "should allow assigning and retrieving objects" do
       u1 = user.create
       u2 = user.create
       p1 = post.create
@@ -22,19 +30,40 @@ describe HashDB::Model do
       u2.posts << p1
       p1.user_id.should eq u2.id
     end
+
+    it "should initialize related modules with .new" do
+      u1 = user.create
+      p1 = u1.posts.new
+
+      p1.user_id.should eq u1.id
+    end
+
+    it "should create related model with .create" do
+      u1 = user.create
+      p1 = u1.posts.create
+
+      p1.user_id.should eq u1.id
+      p1.id.should_not be_nil
+    end
   end
 
   context ".belongs_to" do
-    it "should allow assigning and retrieving objects" do
-      user = Class.new do
+    let :user do
+      Class.new do
         include HashDB::Model
       end
+    end
 
-      post = Class.new do
+    let :post do
+      # Same hack as mentioned above.
+      _user = user
+      Class.new do
         include HashDB::Model
-        belongs_to :user, key: :user_id, class: user
+        belongs_to :user, key: :user_id, class: _user
       end
+    end
 
+    it "should allow assigning and retrieving objects" do
       u1 = user.create
       u2 = user.create
       p1 = post.create

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hash_db
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-02-02 00:00:00.000000000 Z
+date: 2013-02-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -100,7 +100,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3250930581944786811
+      hash: -2794194441662757614
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -109,7 +109,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3250930581944786811
+      hash: -2794194441662757614
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24