sql_enum 0.1.10 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 76781aadbc073e6d58e5366276ed05ab1a65a95d
-  data.tar.gz: 4d4f79bb985e42bb01905f564702be489a8b2149
+  metadata.gz: e0391b2647bafa76738ca6e50034303599631547
+  data.tar.gz: 9486b8ef30b7f6c6d10e0519ed5ee5bc0a363e3c
 SHA512:
-  metadata.gz: eed5cce9bc7c9e89422d1c100b9d72b72e2568b077ecc24fcf2f0fd24c088f7f6b54abe630e4f7bd5f07996e4043234be5ee8d8cafd0839bbeb7164eda8f628d
-  data.tar.gz: c5626fcb86a7f85764a89d8102fa122db773ff70b4929867aec9e47f96fa08407daea6bd4a908abdefc5f65e461749e806e71364855e2b18760f6fd467b149ec
+  metadata.gz: 6e7191b04a08c7baa32dbf5810549548375a16fa9bb0bd3a0aeba4ddff5a1c14e21e407420112eea5d800c98b8a04e232a098b97bf770c891580feb6846be43d
+  data.tar.gz: aadfd52951701ac331b88d24c6a9acae220cb19916a3670c5db9cdf336013c1e2979098d39f58fe42c08966cc016eee690f82f5532cbddca731a0e6b1295b04b

data/lib/active_record/fixtures_override.rb

@@ -0,0 +1,79 @@
+require 'active_record/fixtures'
+
+module ActiveRecord
+  class FixtureSet
+    def table_rows
+      now = config.default_timezone == :utc ? Time.now.utc : Time.now
+
+      # allow a standard key to be used for doing defaults in YAML
+      fixtures.delete('DEFAULTS')
+
+      # track any join tables we need to insert later
+      rows = Hash.new { |h,table| h[table] = [] }
+
+      rows[table_name] = fixtures.map do |label, fixture|
+        row = fixture.to_hash
+
+        if model_class
+          # fill in timestamp columns if they aren't specified and the model is set to record_timestamps
+          if model_class.record_timestamps
+            timestamp_column_names.each do |c_name|
+              row[c_name] = now unless row.key?(c_name)
+            end
+          end
+
+          # interpolate the fixture label
+          row.each do |key, value|
+            row[key] = value.gsub("$LABEL", label.to_s) if value.is_a?(String)
+          end
+
+          # generate a primary key if necessary
+          if has_primary_key_column? && !row.include?(primary_key_name)
+            row[primary_key_name] = ActiveRecord::FixtureSet.identify(label, primary_key_type)
+          end
+
+          # Resolve enums
+          # model_class.defined_enums.each do |name, values|
+          #   if row.include?(name)
+          #     row[name] = values.fetch(row[name], row[name])
+          #   end
+          # end
+
+          # If STI is used, find the correct subclass for association reflection
+          reflection_class =
+            if row.include?(inheritance_column_name)
+              row[inheritance_column_name].constantize rescue model_class
+            else
+              model_class
+            end
+
+          reflection_class._reflections.each_value do |association|
+            case association.macro
+            when :belongs_to
+              # Do not replace association name with association foreign key if they are named the same
+              fk_name = (association.options[:foreign_key] || "#{association.name}_id").to_s
+
+              if association.name.to_s != fk_name && value = row.delete(association.name.to_s)
+                if association.polymorphic? && value.sub!(/\s*\(([^\)]*)\)\s*$/, "")
+                  # support polymorphic belongs_to as "label (Type)"
+                  row[association.foreign_type] = $1
+                end
+
+                fk_type = reflection_class.type_for_attribute(fk_name).type
+                row[fk_name] = ActiveRecord::FixtureSet.identify(value, fk_type)
+              end
+            when :has_many
+              if association.options[:through]
+                add_join_records(rows, row, HasManyThroughProxy.new(association))
+              end
+            end
+          end
+        end
+
+        row
+      end
+      rows
+    end
+  end
+end
+

data/lib/sql_enum/version.rb

@@ -1,3 +1,3 @@
 module SqlEnum
-  VERSION = "0.1.10"
+  VERSION = "0.1.11"
 end

data/lib/sql_enum.rb

@@ -23,6 +23,7 @@ end
 require_relative 'active_record/enum/enum_type'
 require_relative 'active_record/type/enum'
 require_relative 'active_record/enum_override'
+require_relative 'active_record/fixtures_override'
 require_relative 'active_record/connection_adapters/mysql2'
 require_relative 'active_record/connection_adapters/abstract_mysql'
 require_relative 'active_record/connection_adapters/mysql/column_methods'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sql_enum
 version: !ruby/object:Gem::Version
-  version: 0.1.10
+  version: 0.1.11
 platform: ruby
 authors:
 - Fletcher Fowler
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-03-01 00:00:00.000000000 Z
+date: 2018-03-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -89,8 +89,8 @@ files:
 - lib/active_record/connection_adapters/mysql2.rb
 - lib/active_record/enum/enum_type.rb
 - lib/active_record/enum_override.rb
+- lib/active_record/fixtures_override.rb
 - lib/active_record/type/enum.rb
-- lib/array.rb
 - lib/sql_enum.rb
 - lib/sql_enum/version.rb
 - sql_enum.gemspec

data/lib/array.rb

@@ -1,5 +0,0 @@
-Array.class_eval do
-  def fetch(key, default)
-    key if self.include?(key&.to_s)
-  end
-end