couchbase-orm 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f69d88d6820ca854f9650369e01b4ccb7c431a22
-  data.tar.gz: c80410e6473d97aee26722c763286bd0a322e9ea
+  metadata.gz: 4bcf87271b7e719b9420b7f90e4f764dda78f00b
+  data.tar.gz: f3939395af9277a42940d39653861e0ce73df53a
 SHA512:
-  metadata.gz: 5e157bc072a22300558ef1b6bca1ed2033a4c74e8651b78c09306d43687081b0e3560199c10ab185b8b35ae4fca46056a067ddcbf79af09eacdd2d036f225dce
-  data.tar.gz: 3543a615ad4e0f117139ec75658e714f119e15aef615859fa9a5dd161edc90a54d7ba99606cc29b15d39872a79a31a7f3fdcea41527cfc374abb864aaa3f7ab8
+  metadata.gz: 4dd10a65d7dbd402d60ab4963ce211bc5da327ed17feeb2063948af6397da3a4205825a62c7414c8aa4aab30c59fc79fcef59f80b9587be330435ad4f2b6e771
+  data.tar.gz: 11d1bfb0fc6983159bebced2a8c33e5394326063b6461f2cf19e5ca1707d6d1f34d3b857355529941224c5bb2ba219d7bbff615c00511bdb77ac103380e1c9d4

data/lib/couchbase-orm.rb

@@ -7,6 +7,19 @@ module CouchbaseOrm
     autoload :Connection,  'couchbase-orm/connection'
     autoload :IdGenerator, 'couchbase-orm/id_generator'
     autoload :Base,        'couchbase-orm/base'
+
+    def self.try_load(id)
+        result = id.respond_to?(:cas) ? id : CouchbaseOrm::Base.bucket.get(id, quiet: true, extended: true)
+        if result && result.value.is_a?(Hash) && result.value[:type]
+            ddoc = result.value[:type]
+            ::CouchbaseOrm::Base.descendants.each do |model|
+                if model.design_document == ddoc
+                    return model.new(result)
+                end
+            end
+        end
+        nil
+    end
 end
 
 # Provide Boolean conversion function

data/lib/couchbase-orm/associations.rb

@@ -29,7 +29,11 @@ module CouchbaseOrm
                 # Define reader
                 define_method(name) do
                     return instance_variable_get(instance_var) if instance_variable_defined?(instance_var)
-                    val = assoc.constantize.find(self.send(ref), quiet: true)
+                    val = if options[:polymorphic]
+                        ::CouchbaseOrm.try_load(self.send(ref))
+                    else
+                        assoc.constantize.find(self.send(ref), quiet: true)
+                    end
                     instance_variable_set(instance_var, val)
                     val
                 end
@@ -38,6 +42,10 @@ module CouchbaseOrm
                 attr_writer name
                 define_method(:"#{name}=") do |value|
                     if value
+                        if !options[:polymorphic]
+                            klass = assoc.constantize
+                            raise ArgumentError, "type mismatch on association: #{klass.design_document} != #{value.class.design_document}" if klass.design_document != value.class.design_document
+                        end
                         self.send(ref_ass, value.id)
                     else
                         self.send(ref_ass, nil)

data/lib/couchbase-orm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true, encoding: ASCII-8BIT
 
 module CouchbaseOrm
-    VERSION = '0.0.2'
+    VERSION = '0.0.3'
 end

data/spec/associations_spec.rb

@@ -7,6 +7,10 @@ class Parent < CouchbaseOrm::Base
     attribute :name
 end
 
+class RandomOtherType < CouchbaseOrm::Base
+    attribute :name
+end
+
 class Child < CouchbaseOrm::Base
     attribute :name
 
@@ -70,6 +74,15 @@ describe CouchbaseOrm::Associations do
         expect(Parent.exists?(id)).to be(false)
     end
 
+    it "should raise an error if an invalid type is being assigned" do
+        begin
+            parent = RandomOtherType.create!(name: 'joe')
+            expect { Child.create!(name: 'bob', parent: parent) }.to raise_error(ArgumentError)
+        ensure
+            parent.delete
+        end
+    end
+
     describe Parent do
         it_behaves_like "ActiveModel"
     end

data/spec/base_spec.rb

@@ -62,6 +62,16 @@ describe CouchbaseOrm::Base do
         base.destroy
     end
 
+    it "should try to load a model with nothing but an ID" do
+        begin
+            base = BaseTest.create!(name: 'joe')
+            obj = CouchbaseOrm.try_load(base.id)
+            expect(obj).to eq(base)
+        ensure
+            base.destroy
+        end
+    end
+
     describe BaseTest do
         it_behaves_like "ActiveModel"
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: couchbase-orm
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Stephen von Takach