active_mocker 2.4.0.pre1 → 2.4.0.pre2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d76ea3471b67a416239349d655e5e3e9061e2cf3
-  data.tar.gz: 1f10949f52554aa237ae3cf60ed7e7b56cc0c2fd
+  metadata.gz: 9696a956e1bea0e39fab858b45697d76cef725a7
+  data.tar.gz: e71122a7b31abd042d9ad33d6bfba2d5afc16a0b
 SHA512:
-  metadata.gz: 357f966e759449cf9ae50fba27ac6f319ddee7d06f719a3c5311c33666f6b385ed0291238c2f516b20d3baf9d171b96c6ae0030865839ff94379b0039882423e
-  data.tar.gz: f056acf32872ddcb7ec7c3f349241c7a93c459b8abf70cc14b3da61de3d1f0f5e982c4d97ddd33bc69d6ba9775f99ebdf38dd5f7b8e49bafb064eec0eef0cdc2
+  metadata.gz: 18bb1cf82a31fc4a39726f3eda57c8bd1466ae608c58f03f012774b2e2fafc2b58351c6d48bc8f623c7271553c9d934115e83d4a517afc17bf1fabf7b7cbe3e8
+  data.tar.gz: 2155bcea9852bb130941a57fd0736b0bd4bdc856902627afbb4f33356b6c317acd6ba6907b489201ecb04914a3e407dd4b9744ee520c23398d193c47f1a417ec

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 # Changelog
 All notable changes to this project will be documented in this file.
 
+## 2.4.0.pre2 - 2016-10-17
+### Enhancement
+- ActiveRecord Enum support
+Compatible with ActiveRecord versions 5 and 4. Generating using different version of ActiveRecord creates different mock files.
+
 ## 2.4.0.pre1 - 2016-10-13
 ### Enhancement
 - Option to delete all records before each example
@@ -21,7 +26,7 @@ All notable changes to this project will be documented in this file.
     ```ruby
     ActiveMocker::LoadedMocks.features.enable(:timestamps)
     ```
-    
+
 ## 2.3.3 - 2016-10-13
 ### Enhancement
 -  Auto stubbing of ActiveRecord::RecordNotFound with requiring "active_mocker/rspec_helper"

data/README.md

@@ -362,7 +362,7 @@ See [Documentation](http://rdoc.info/github/zeisler/active_mocker/master/ActiveM
   * attributes
   * update
   * save/save!
-  * write_attribute/read_attribute - (protected, can be used within modules)
+  * write_attribute/read_attribute
   * delete
   * new_record?
   * persisted?
@@ -373,6 +373,7 @@ See [Documentation](http://rdoc.info/github/zeisler/active_mocker/master/ActiveM
   * slice
   * attribute_alias?
   * alias_attributes
+  * touch
 
 **has_one/belongs_to/has_many**
 

data/lib/active_mocker.rb

@@ -28,3 +28,5 @@ require "active_mocker/mock_creator"
 require "active_mocker/error_object"
 require "active_mocker/display_errors"
 require "active_mocker/generate"
+require "active_mocker/attribute_types/enum"
+require "active_mocker/attribute"

data/lib/active_mocker/attribute.rb

@@ -0,0 +1,2 @@
+ActiveRecordSchemaScrapper::Attribute.attribute :attribute_writer, String
+ActiveRecordSchemaScrapper::Attribute.attribute :attribute_reader, String

data/lib/active_mocker/attribute_types/enum.rb

@@ -0,0 +1,64 @@
+module ActiveMocker
+  module AttributeTypes
+    class Enum < Virtus::Attribute
+      class << self
+        def build(db_value_type:, table_name:, attribute:, enums:, ignore_value: false)
+          klass               = Class.new(ActiveMocker::AttributeTypes::Enum)
+          klass.table_name    = table_name.to_sym
+          klass.attribute     = attribute.to_sym
+          klass.ignore_value  = ignore_value
+          enums               = if enums.is_a?(Array)
+                                  enums.each_with_object({}).with_index { |(k, h), i| h[k] = i }
+                                else
+                                  enums
+                                end
+          klass.key_type      = String
+          klass.db_value_type = db_value_type
+          klass.enums         = Hash[enums.map do |k, v|
+            [Virtus::Attribute.build(klass.key_type).coerce(k),
+             Virtus::Attribute.build(klass.db_value_type).coerce(v)]
+          end]
+          klass
+        end
+
+        def to_s
+          "ActiveMocker::AttributeTypes::Enum.build(ignore_value: #{ignore_value}, db_value_type: #{db_value_type}, table_name: :#{table_name}, attribute: :#{attribute}, enums: #{enums.inspect})"
+        end
+
+        attr_accessor :enums, :table_name, :attribute, :db_value_type, :key_type, :ignore_value
+      end
+
+      def coerce(key)
+        return if key.nil?
+        coerced_key = key_type.coerce(key)
+        if key && self.class.enums.key?(coerced_key)
+          if self.class.ignore_value
+            coerced_key
+          else
+            get_value(key)
+          end
+        else
+          raise ArgumentError, "'#{coerced_key}' is not a valid #{self.class.attribute}"
+        end
+      end
+
+      def get_key(value)
+        self.class.enums.invert[db_value_type.coerce(value)]
+      end
+
+      def get_value(key)
+        self.class.enums[key_type.coerce(key)]
+      end
+
+      def key_type
+        Virtus::Attribute.build(self.class.key_type)
+      end
+
+      def db_value_type
+        Virtus::Attribute.build(self.class.db_value_type)
+      end
+    end
+  end
+end
+
+ActiveRecordSchemaScrapper::Attributes.register_type(name: :enum, klass: String)

data/lib/active_mocker/mock/base.rb

@@ -362,7 +362,7 @@ module ActiveMocker
         @associations[attr.to_sym] = value
       end
 
-      protected :read_attribute, :write_attribute, :read_association, :write_association
+      protected :read_association, :write_association
     end
 
     include PropertiesGetterAndSetter

data/lib/active_mocker/mock_creator.rb

@@ -217,25 +217,74 @@ module ActiveMocker
 
     module Attributes
       def attributes
-        @attribute ||= begin
+        @attributes ||= begin
           a = schema_scrapper.attributes.to_a
           a << primary_key unless a.any? { |aa| aa.name == "id" }
+          a.map(&method(:process_attr))
           a
         end
       end
+
+      def process_attr(attr)
+        enums                 = enums(attr.name)
+        attr.default          = Virtus::Attribute.build(attr.type).coerce(attr.default)
+        attr.attribute_writer = "write_attribute(:#{attr.name}, val)"
+        attr.attribute_reader = "read_attribute(:#{attr.name})"
+
+        unless enums.empty?
+          enum_type = ActiveMocker::AttributeTypes::Enum.build(
+            enums:         enums,
+            table_name:    table_name,
+            attribute:     attr.name,
+            db_value_type: attr.type,
+          )
+          if ActiveRecord::VERSION::MAJOR == 5
+            enum_type.ignore_value = true
+            attr.type              = enum_type
+            if attr.default
+              attr.default = Virtus::Attribute.build(enum_type).get_key(attr.default)
+            end
+          elsif ActiveRecord::VERSION::MAJOR == 4
+            attr.attribute_writer = "@#{attr.name}_enum_type ||= Virtus::Attribute.build(#{enum_type})\nwrite_attribute(:#{attr.name}, @#{attr.name}_enum_type.coerce(val))"
+            attr.attribute_reader = "@#{attr.name}_enum_type ||= Virtus::Attribute.build(#{enum_type})\n@#{attr.name}_enum_type.get_key(read_attribute(:#{attr.name}))"
+            if attr.default
+              attr.default = Virtus::Attribute.build(attr.type).coerce(attr.default)
+            end
+          end
+          attr
+        end
+      end
+
+      def enums(attribute)
+        @enums ||= begin
+          raw_enums = class_introspector
+                        .class_macros
+                        .select { |hash| hash.key?(:enum) }
+          if raw_enums
+            raw_enums
+              .map { |hash| hash[:enum].flatten.first }
+              .each_with_object({}) { |v, h| h.merge!(v) }
+          else
+            {}
+          end
+        end
+
+        @enums.fetch(attribute.to_sym, {})
+      end
     end
 
     module ClassMethods
       include Attributes
 
       def attributes_with_defaults
+        types_hash
         attributes.each_with_object({}) do |attr, hash|
-          hash[attr.name] = Virtus::Attribute.build(attr.type).coerce(attr.default)
+          hash[attr.name] = attr.default
         end
       end
 
       def types_hash
-        attributes.each_with_object(HashNewStyle.new) do |attr, types|
+        @types_hash ||= attributes.each_with_object(HashNewStyle.new) do |attr, types|
           types[attr.name] = attr.type.to_s
         end.inspect
       end

data/lib/active_mocker/mock_template/_attributes.erb

@@ -1,11 +1,11 @@
 # _attributes.erb
 <% attributes.each do |meth| -%>
   def <%= meth.name %>
-    read_attribute(:<%= meth.name %>)
+    <%= meth.attribute_reader %>
   end
 
 
   def <%= meth.name %>=(val)
-    write_attribute(:<%= meth.name %>, val)
+    <%= meth.attribute_writer %>
   end
-<% end -%>
+<% end -%>

data/lib/active_mocker/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module ActiveMocker
-  VERSION = "2.4.0.pre1"
+  VERSION = "2.4.0.pre2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_mocker
 version: !ruby/object:Gem::Version
-  version: 2.4.0.pre1
+  version: 2.4.0.pre2
 platform: ruby
 authors:
 - Dustin Zeisler
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-10-14 00:00:00.000000000 Z
+date: 2016-10-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -184,6 +184,8 @@ files:
 - LICENSE.txt
 - README.md
 - lib/active_mocker.rb
+- lib/active_mocker/attribute.rb
+- lib/active_mocker/attribute_types/enum.rb
 - lib/active_mocker/config.rb
 - lib/active_mocker/deprecated_components/mock_abilities.rb
 - lib/active_mocker/deprecated_components/rspec.rb