traitr 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 020595fcb015240ac313662cc5e284f10ee4ab06
-  data.tar.gz: 798775cefb913fc437972b29e8e412b792b29d9c
+  metadata.gz: 617169e1a6b866506fdb0e4804fe7d8e7eb22ad5
+  data.tar.gz: a54d3fa3d4ce4915eab40f90f89270efd2e7da23
 SHA512:
-  metadata.gz: 7c8371264c2dcc1ce8b3cf7af4e057567c6b69e765e2e14279b420631f94b0de1ddc01e46623f50901e3700616563aba45d72d61a99eaa3911adc8ff959e1b56
-  data.tar.gz: 0b71b50fe8da6da863269c9f6db7d82770a55ec6a1949f0b67eb4ce9a9f477058c6a0b73e6f5150ff2d14962ec3a57bdb72c67c456b720036f8a0b23ae915638
+  metadata.gz: 8fff94ed87623a0810bfd89ca3076dc8f2d3fd90cf21b284738d32a294dfe2cbd05f582b5fc143327f34ff90ec01bf82cd95e9fa7a00bbdfd1b697fc15240e4e
+  data.tar.gz: 5dfb5fb4b2af1e00372cddebdfec6b703364a9c20006d653a315d77e7ad6c3e5df8197df1b615ec9814a1b5b7da8b349df3a280bae0391147722fcc67aa80164

data/lib/traitor/config.rb

@@ -4,22 +4,21 @@ module Traitor
     @create_kwargs  = {}
     @build_kwargs = {}
     @build_with_list = false # use (**attributes.merge(build_kwargs)) instead of (attributes, **build_kwargs)
+    @no_callbacks = false
 
     class << self
-      attr_accessor :create_method, :create_kwargs, :build_kwargs, :build_with_list
+      attr_accessor :create_method, :create_kwargs, :build_kwargs, :build_with_list, :no_callbacks
 
       def configure_for_rails!
         @create_method  = :save
         @create_kwargs  = { validate: false }
         @build_kwargs = { without_protection: true }
-        @build_with_list = false
       end
 
       def configure_safe_for_rails!
         @create_method  = :save
         @create_kwargs  = {}
         @build_kwargs = {}
-        @build_with_list = false
       end
 
       def reset!
@@ -27,6 +26,19 @@ module Traitor
         @create_kwargs = {}
         @build_kwargs = {}
         @build_with_list = false
+        @no_callbacks = false
+      end
+
+      def stash!
+        @old_config = Hash[
+          self.instance_variables.map { |att| [att, self.instance_variable_get(att)] }
+        ]
+      end
+
+      def restore!
+        return unless @old_config
+        @old_config.each { |att, val| self.instance_variable_set(att, val) }
+        @old_config = nil
       end
     end
   end

data/lib/traitor/helpers/active_record.rb

@@ -0,0 +1,63 @@
+# note that we do not have activerecord as a dependency in this gem!
+# do not `require 'traitor/helpers/active_record'` unless you have ActiveRecord
+# in your gemfile
+
+require 'active_record'
+
+module Traitor
+  module Helpers
+    module ActiveRecord
+      def create_without_callbacks
+        case self.class.connection.class.name
+        when 'ActiveRecord::ConnectionAdapters::PostgreSQLAdapter'
+          create_without_callbacks_pg
+        when 'ActiveRecord::ConnectionAdapters::SQLite3Adapter'
+          create_without_callbacks_sqlite
+        else
+          create_without_callbacks_default
+        end
+      end
+
+      private
+
+      def create_without_callbacks_default
+        insert_sql = self.class.arel_table.create_insert.tap do |im|
+          im.insert(self.send(:arel_attributes_with_values_for_create, self.attribute_names))
+        end.to_sql
+
+        pk = self.class.primary_key
+        self.class.connection.execute(insert_sql)
+        id = self.maximum(pk)
+        self.send(:"#{pk}=", id)
+        self.clear_changes_information
+      end
+
+      def create_without_callbacks_sqlite
+        insert_sql = self.class.arel_table.create_insert.tap do |im|
+          im.insert(self.send(:arel_attributes_with_values_for_create, self.attribute_names))
+        end.to_sql
+
+        conn = self.class.connection
+        conn.execute(insert_sql)
+
+        pk = self.class.primary_key
+        id = conn.execute("SELECT last_insert_rowid() AS id")[0]['id']
+        self.send(:"#{pk}=", id)
+        self.clear_changes_information
+      end
+
+      def create_without_callbacks_pg
+        insert_sql = self.class.arel_table.create_insert.tap do |im|
+          im.insert(self.send(:arel_attributes_with_values_for_create, self.attribute_names))
+        end.to_sql
+
+        pk = self.class.primary_key
+        id = self.class.connection.execute(insert_sql + " RETURNING #{pk}")[0][pk]
+        self.send(:"#{pk}=", id)
+        self.clear_changes_information
+      end
+    end
+  end
+end
+
+ActiveRecord::Base.send :include, Traitor::Helpers::ActiveRecord

data/lib/traitor/helpers/rspec.rb

@@ -0,0 +1,29 @@
+require 'rspec'
+
+module Traitor
+  module Helpers
+    class RSpec
+      def self.configure!
+        ::RSpec.configure do |config|
+          config.around(:example) do |example|
+            traitor_configs = example.metadata && example.metadata.select { |key, conf| key.to_s.start_with? 'traitor_' }
+            if traitor_configs && traitor_configs.any?
+              Traitor::Config.stash!
+              traitor_configs.each do |key, conf|
+                Traitor::Config.send(:"#{key.to_s.sub('traitor_', '')}=", conf)
+              end
+
+              example.run
+
+              Traitor::Config.restore!
+            else
+              example.run
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+Traitor::Helpers::RSpec.configure!

data/lib/traitor.rb

@@ -6,139 +6,141 @@ module Traitor
   BLOCK_KEYS = [:after_build, :after_create]
 
   @trait_library = {}
+  @alternate_create_methods = {}
   @block_library = {}
   @class_cache = {}
   @trait_cache = {}
 
-  def self.reset!
-    @trait_library = {}
-    @block_library = {}
-    @class_cache = {}
-    @trait_cache = {}
-  end
+  class << self
+    def reset!
+      @trait_library = {}
+      @alternate_create_methods = {}
+      @block_library = {}
+      @class_cache = {}
+      @trait_cache = {}
+    end
 
-  def self.define(klass, **traits)
-    @trait_library[klass] ||= {}
+    def define(klass, **traits)
+      @trait_library[klass] ||= {}
 
-    (traits.keys & BLOCK_KEYS).each do |block_type|
-      block = traits.delete block_type
-      @block_library[klass] ||= {class: {}, traits: {}}
-      @block_library[klass][:class][block_type] = block
-    end
+      if alternate_create_method = traits.delete(:create_using)
+        @alternate_create_methods[klass] = [alternate_create_method, traits.delete(:create_using_kwargs) || {}]
+      end
 
-    traits.each do |trait, attributes|
-      (attributes.keys & BLOCK_KEYS).each do |block_type|
-        block = attributes.delete block_type
+      (traits.keys & BLOCK_KEYS).each do |block_type|
+        raise Traitor::Error.new("Callbacks are forbidden!") if Traitor::Config.no_callbacks
+        block = traits.delete block_type
         @block_library[klass] ||= {class: {}, traits: {}}
-        @block_library[klass][:traits][trait] ||= {}
-        @block_library[klass][:traits][trait][block_type] = block
+        @block_library[klass][:class][block_type] = block
       end
-    end
 
-    @trait_library[klass].merge!(traits)
-  end
+      traits.each do |trait, attributes|
+        (attributes.keys & BLOCK_KEYS).each do |block_type|
+          raise Traitor::Error.new("Callbacks are forbidden!") if Traitor::Config.no_callbacks
+          block = attributes.delete block_type
+          @block_library[klass] ||= {class: {}, traits: {}}
+          @block_library[klass][:traits][trait] ||= {}
+          @block_library[klass][:traits][trait][block_type] = block
+        end
+      end
 
-  ##
-  # build an instance of an object using the defined traits and attributes,
-  # and then save it.
-  ##
-  def self.create(klass, *traits, **attributes)
-    create_method = Traitor::Config.create_method
-    raise Traitor::Error.new("Cannot call Traitor.create until you have configured Traitor.create_method .") unless create_method
+      @trait_library[klass].merge!(traits)
+    end
 
-    record = build(klass, *traits, **attributes)
+    ##
+    # build an instance of an object using the defined traits and attributes.
+    ##
+    def build(klass, *traits, **attributes)
+      attributes = get_attributes_from_traits(klass, traits).merge(attributes)
+      build_kwargs = Traitor::Config.build_kwargs || {}
+
+      record = if Traitor::Config.build_with_list
+        convert_to_class(klass).new(**attributes.merge(build_kwargs))
+      elsif build_kwargs.any?
+        convert_to_class(klass).new(attributes, **build_kwargs)
+      else
+        convert_to_class(klass).new(attributes)
+      end
 
-    if (create_kwargs = Traitor::Config.create_kwargs).any? # assignment intentional
-      record.public_send(create_method, **create_kwargs)
-    else
-      record.public_send(create_method)
+      call_blocks(klass, :after_build, record, *traits)
+      record
     end
 
-    call_blocks(klass, :after_create, record, *traits)
-    record
-  end
+    ##
+    # build an instance of an object using the defined traits and attributes,
+    # and then save it.
+    ##
+    def create(klass, *traits, **attributes)
+      create_method, create_kwargs = @alternate_create_methods[klass] ||
+        [Traitor::Config.create_method, Traitor::Config.create_kwargs || {}]
 
-  ##
-  # build an instance of an object using the defined traits and attributes.
-  ##
-  def self.build(klass, *traits, **attributes)
-    attributes = get_attributes_from_traits(klass, traits).merge(attributes)
-    build_kwargs = Traitor::Config.build_kwargs
-
-    record = if Traitor::Config.build_with_list
-      convert_to_class(klass).new(**attributes.merge(build_kwargs))
-    elsif build_kwargs.any?
-      convert_to_class(klass).new(attributes, **build_kwargs)
-    else
-      convert_to_class(klass).new(attributes)
-    end
+      raise Traitor::Error.new("Cannot call Traitor.create until you have configured Traitor.create_method .") unless create_method
 
-    call_blocks(klass, :after_build, record, *traits)
-    record
-  end
+      record = build(klass, *traits, **attributes)
 
-  def self.create_using(klass, create_method, *traits, **attributes)
-    old_create_method = Traitor::Config.create_method
-    old_create_kwargs = Traitor::Config.create_kwargs
+      if create_kwargs.any? # assignment intentional
+        record.public_send(create_method, **create_kwargs)
+      else
+        record.public_send(create_method)
+      end
 
-    new_create_kwargs = attributes.delete :create_kwargs
-    Traitor::Config.create_method = create_method
-    Traitor::Config.create_kwargs = new_create_kwargs if new_create_kwargs
+      call_blocks(klass, :after_create, record, *traits)
+      record
+    end
 
-    create(klass, *traits, **attributes)
-  ensure
-    Traitor::Config.create_method = old_create_method
-    Traitor::Config.create_kwargs = old_create_kwargs
-  end
+    def create_using(klass, create_method, *traits, **attributes)
+      old_create_method_kwargs = @alternate_create_methods[klass]
+      @alternate_create_methods[klass] = [create_method, attributes.delete(:create_kwargs) || {}]
+      create(klass, *traits, **attributes)
+    ensure
+      @alternate_create_methods[klass] = old_create_method_kwargs
+    end
 
-  # private methods
+    private
 
-  def self.call_blocks(klass, trigger, record, *traits)
-    return unless @block_library[klass]
-    [].tap do |blocks|
-      blocks << @block_library[klass][:class][trigger]
-      traits.each do |trait|
-        if @block_library[klass][:traits][trait]
-          blocks << @block_library[klass][:traits][trait][trigger]
+    def call_blocks(klass, trigger, record, *traits)
+      return unless @block_library[klass]
+      [].tap do |blocks|
+        blocks << @block_library[klass][:class][trigger]
+        traits.each do |trait|
+          if @block_library[klass][:traits][trait]
+            blocks << @block_library[klass][:traits][trait][trigger]
+          end
         end
-      end
-    end.compact.each { |block| block.call(record) }
-  end
-  private_class_method :call_blocks
+      end.compact.each { |block| block.call(record) }
+    end
 
-  def self.convert_to_class(klass)
-    @class_cache[klass] ||= Object.const_get(camelize(klass))
-  rescue NameError
-    raise Traitor::Error.new("Tried to create a #{camelize(klass)}, but it does not exist!")
-  end
-  private_class_method :convert_to_class
+    def convert_to_class(klass)
+      @class_cache[klass] ||= Object.const_get(camelize(klass))
+    rescue NameError
+      raise Traitor::Error.new("Tried to create a #{camelize(klass)}, but it does not exist!")
+    end
+
+    def get_attributes_from_traits(klass, traits)
+      # we only call this method when the klass has been converted to a key inside create
+      return {} unless library = @trait_library[klass]
 
-  def self.get_attributes_from_traits(klass, traits)
-    # we only call this method when the klass has been converted to a key inside create
-    return {} unless library = @trait_library[klass]
+      traits = [:default_traits] + traits # always include default_traits as the first thing
 
-    traits = [:default_traits] + traits # always include default_traits as the first thing
+      cache_key = klass.to_s + ':' + traits.join(':')
+      @trait_cache[cache_key] ||= {}.tap do |attributes|
+        traits.each { |trait| attributes.merge!(library[trait] || {}) }
+      end
 
-    cache_key = klass.to_s + ':' + traits.join(':')
-    @trait_cache[cache_key] ||= {}.tap do |attributes|
-      traits.each { |trait| attributes.merge!(library[trait] || {}) }
+      # use late resolution on lambda values by calling them here as part of constructing a new hash
+      Hash[
+        @trait_cache[cache_key].map do |attribute, value|
+          [attribute, value.is_a?(Proc) && ![:after_build, :after_create].include?(attribute) ? value.call : value]
+        end
+      ]
     end
 
-    # use late resolution on lambda values by calling them here as part of constructing a new hash
-    Hash[
-      @trait_cache[cache_key].map do |attribute, value|
-        [attribute, value.is_a?(Proc) && ![:after_build, :after_create].include?(attribute) ? value.call : value]
-      end
-    ]
-  end
-  private_class_method :get_attributes_from_traits
-
-  def self.camelize(term)
-    string = term.to_s
-    string[0] = string[0].upcase
-    string.gsub!(/(?:_|(\/))([a-z\d]*)/i) { "#{$1}#{$2.capitalize}" }
-    string.gsub!('/'.freeze, '::'.freeze)
-    string
+    def camelize(term)
+      string = term.to_s
+      string[0] = string[0].upcase
+      string.gsub!(/(?:_|(\/))([a-z\d]*)/i) { "#{$1}#{$2.capitalize}" }
+      string.gsub!('/'.freeze, '::'.freeze)
+      string
+    end
   end
-  private_class_method :camelize
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: traitr
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Zach Lome
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-07-26 00:00:00.000000000 Z
+date: 2016-07-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -25,35 +25,35 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '3.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '3.0'
 - !ruby/object:Gem::Dependency
-  name: rspec
+  name: rspec-its
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
-  name: rspec-its
+  name: timecop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -67,7 +67,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: timecop
+  name: pry
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -81,7 +81,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: pry
+  name: sqlite3
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -94,6 +94,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4'
 description: a lightweight, simplified, pure-ruby way to handle object model construction,
   similar to FactoryGirl.
 email:
@@ -106,6 +120,8 @@ files:
 - lib/traitor/config.rb
 - lib/traitor/error.rb
 - lib/traitor/find_definitions.rb
+- lib/traitor/helpers/active_record.rb
+- lib/traitor/helpers/rspec.rb
 homepage: https://github.com/kuraiou/traitor
 licenses:
 - MIT