chozo 0.4.0 → 0.4.1

data/lib/chozo/varia_model.rb

@@ -6,6 +6,11 @@ module Chozo
   # @author Jamie Winsor <jamie@vialstudios.com>
   module VariaModel
     module ClassMethods
+      ASSIGNMENT_MODES = [
+        :whitelist,
+        :carefree
+      ]
+
       # @return [HashWithIndifferentAccess]
       def attributes
         @attributes ||= Hashie::Mash.new
@@ -16,6 +21,26 @@ module Chozo
         @validations ||= HashWithIndifferentAccess.new
       end
 
+      # @return [Symbol]
+      def assignment_mode
+        @assignment_mode ||= :whitelist
+      end
+
+      # Set the attribute mass assignment mode
+      #   * :whitelist - only attributes defined on the class will have values set
+      #   * :carefree  - values will be set for attributes that are not explicitly defined
+      #                  in the class definition 
+      #
+      # @param [Symbol] mode
+      #   an assignment mode to use @see {ASSIGNMENT_MODES}
+      def set_assignment_mode(mode)
+        unless ASSIGNMENT_MODES.include?(mode)
+          raise ArgumentError, "unknown assignment mode: #{mode}"
+        end
+
+        @assignment_mode = mode
+      end
+
       # @param [#to_s] name
       # @option options [Symbol, Array<Symbol>] :type
       # @option options [Boolean] :required
@@ -172,11 +197,11 @@ module Chozo
     end
 
     def mass_assign(new_attrs = {})
-      attributes.dotted_paths.each do |dotted_path|
-        value = new_attrs.dig(dotted_path)
-        next if value.nil?
-
-        set_attribute(dotted_path, value)
+      case self.class.assignment_mode
+      when :whitelist
+        whitelist_assign(new_attrs)
+      when :carefree
+        carefree_assign(new_attrs)
       end
     end
     alias_method :attributes=, :mass_assign
@@ -233,6 +258,21 @@ module Chozo
       def add_error(attribute, message)
         self.errors[attribute] ||= Array.new
         self.errors[attribute] << message
-      end    
+      end
+
+    private
+
+      def carefree_assign(new_attrs = {})
+        attributes.deep_merge!(new_attrs)
+      end
+
+      def whitelist_assign(new_attrs = {})
+        attributes.dotted_paths.each do |dotted_path|
+          value = new_attrs.dig(dotted_path)
+          next if value.nil?
+
+          set_attribute(dotted_path, value)
+        end
+      end
   end
 end

data/lib/chozo/version.rb

@@ -1,3 +1,3 @@
 module Chozo
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

data/spec/unit/chozo/varia_model_spec.rb

@@ -90,6 +90,32 @@ describe Chozo::VariaModel do
           validations.should be_empty
         end
       end
+
+      describe "#assignment_mode" do
+        it "returns the default assignment mode :whitelist" do
+          subject.assignment_mode.should eql(:whitelist)
+        end
+      end
+
+      describe "#set_assignment_mode" do
+        it "sets the assignment_mode to whitelist" do
+          subject.set_assignment_mode(:whitelist)
+
+          subject.assignment_mode.should eql(:whitelist)
+        end
+
+        it "sets the assignment_mode to carefree" do
+          subject.set_assignment_mode(:carefree)
+
+          subject.assignment_mode.should eql(:carefree)
+        end
+
+        it "raises if given an invalid assignment mode" do
+          expect {
+            subject.set_assignment_mode(:not_a_real_mode)
+          }.to raise_error(ArgumentError)
+        end
+      end
     end
 
     describe "::validate_kind_of" do
@@ -498,8 +524,28 @@ describe Chozo::VariaModel do
       }
 
       subject.mass_assign(new_attrs)
+      subject.attributes[:undefined_attribute].should be_nil
       subject.should_not respond_to(:undefined_attribute)
     end
+
+    context "when in carefree assignment mode" do
+      subject do
+        Class.new do
+          include Chozo::VariaModel
+
+          set_assignment_mode :carefree
+        end.new
+      end
+
+      it "does not ignore values which are not defined" do
+        new_attrs = {
+          undefined_attribute: "value"
+        }
+
+        subject.mass_assign(new_attrs)
+        subject.attributes[:undefined_attribute].should eql("value")
+      end
+    end
   end
 
   describe "#from_json" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chozo
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
   prerelease: 
 platform: ruby
 authors:
@@ -125,7 +125,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -4510203401888880792
+      hash: 3156158446945270058
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.23