mongo_mapper-strict_keys 0.0.1 → 0.0.2

data/lib/mongo_mapper/strict_keys/force_plugin.rb

@@ -0,0 +1,4 @@
+require 'mongo_mapper/strict_keys'
+
+MongoMapper::Document.plugin(MongoMapper::Plugins::StrictKeys)
+MongoMapper::EmbeddedDocument.plugin(MongoMapper::Plugins::StrictKeys)

data/lib/mongo_mapper/strict_keys/version.rb

@@ -1,7 +1,7 @@
 module MongoMapper
   module Plugins
     module StrictKeys
-      VERSION = "0.0.1"
+      VERSION = "0.0.2"
     end
   end
 end

data/lib/mongo_mapper/strict_keys.rb

@@ -11,7 +11,6 @@ module MongoMapper
 
       module InstanceMethods
         def ensure_key_exists(name)
-          # Do nothing.
           unless respond_to?("#{name}=")
             raise ArgumentError.new("Attribute #{name} has not been explicitly defined as a key.")
           end

data/spec/mongo_mapper/force_plugin_spec.rb

@@ -0,0 +1,50 @@
+require File.expand_path("../spec_helper", File.dirname(__FILE__))
+
+require 'mongo_mapper/strict_keys/force_plugin'
+
+describe 'MongoMapper::Plugins::StrictKeys -- Forced Plugin' do
+  class ForcedModel
+    include MongoMapper::Document
+
+    key :string_key, String
+  end
+
+  describe 'Instance attributes' do
+    before(:each) do
+      @fm = ForcedModel.new
+    end
+
+    context 'attributes-as-methods' do
+      it 'should accept attributes-as-methods for pre-defined keys' do
+        lambda {
+          @fm.string_key = "I work!"
+        }.should_not raise_error
+
+        @fm.string_key.should == "I work!"
+      end
+
+      it 'should reject attributes-as-methods for never-defined keys' do
+        lambda {
+          @fm.not_defined = 'blowed up'
+        }.should raise_error(NoMethodError)
+      end
+    end
+
+    context 'attributes-in-brackets' do
+      it 'should accept attributes-in-brackets for pre-defined keys' do
+        lambda {
+          @fm[:string_key] = "I work!"
+        }.should_not raise_error
+
+        @fm[:string_key].should == "I work!"
+      end
+
+      it 'should reject attributes-in-brackets for never-defined keys' do
+        lambda {
+          @fm[:not_defined] = 'blowed up'
+        }.should raise_error(ArgumentError)
+      end
+    end
+
+  end
+end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Scott Gonyea
@@ -102,8 +102,10 @@ files:
 - Rakefile
 - lib/mongo_mapper-strict_keys.rb
 - lib/mongo_mapper/strict_keys.rb
+- lib/mongo_mapper/strict_keys/force_plugin.rb
 - lib/mongo_mapper/strict_keys/version.rb
 - mongo_mapper-strict_keys.gemspec
+- spec/mongo_mapper/force_plugin_spec.rb
 - spec/mongo_mapper/strict_keys_spec.rb
 - spec/spec_helper.rb
 has_rdoc: true
@@ -139,5 +141,6 @@ signing_key:
 specification_version: 3
 summary: Mongo Mapper Strict Keys Plugin
 test_files: 
+- spec/mongo_mapper/force_plugin_spec.rb
 - spec/mongo_mapper/strict_keys_spec.rb
 - spec/spec_helper.rb