fixturies 0.0.1 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 02d91937e41679e4804b8d8d7d626c431e988edd
-  data.tar.gz: d1cb12c8c6eb33db75519055c0ff31d72a33a7a4
+SHA256:
+  metadata.gz: 95b9e96f5a11eb6e20390fe38f5a786de44adf85930aa5218840d2e559144e7d
+  data.tar.gz: e83b5b488b24fdbe24f464bdff10d59e916e96a18bdf4e6d319e174f44310747
 SHA512:
-  metadata.gz: aaa46d44632beec2b9f110974ed3eeedbb7b861786818949857db4b34a23f4239ab75960063e94c956d3faa03d05305eae6056c01ee7f4be10b82ec914af80bb
-  data.tar.gz: d6382d8700c90d14612536b1d8ac27265193347fbb38545a14976368daf549d7eb6ea1ed8e449df0327cd0a76df71bbc326efa55d72483849caac271729e68d9
+  metadata.gz: 33c08b5e3804540d938d019024e001c601e8f55f6320518e2573f9500dbe0ad56fe4bc57418d96c922a95e6a70057b19d3d2304e4eb87888414eeebc6fbacdfb
+  data.tar.gz: 4453b116af5b4ed500b78f0dc096c645a6709da2445770e00fe6cc23dbc684336d0279af8773a316140d894ce54b0d9a1ede02e26f5490ed53c5bb6d5e1dc6cd

data/lib/fixturies.rb

@@ -2,14 +2,11 @@ class Fixturies
 
     class << self
 
-        attr_reader :fixtures_directory
+        attr_reader :fixtures_directory, :fixture_class_names
 
-        def build(*things, &proc)
+        def build(&proc)
             meth_name = :"builder_#{rand}"
             define_method meth_name, &proc
-            things.each do |thing|
-                thing.is_a?(String) ? table(thing) : model(thing)
-            end
             builders << meth_name
         end
 
@@ -17,23 +14,32 @@ class Fixturies
             @fixtures_directory = dir
         end
 
+        def set_fixture_class(class_names = {})
+            @fixture_class_names ||= {}
+            @fixture_class_names = fixture_class_names.merge(class_names.stringify_keys)
+        end
+
         def create_fixtures
             self.new.create_fixtures
         end
 
-        def table(table_name)
-            self.table_names << table_name
+        def table_names
+            @all_table_names ||= ActiveRecord::Base.connection.execute("
+                SELECT table_name
+                FROM information_schema.tables
+                WHERE
+                    table_schema = 'public'
+                    AND table_type = 'BASE TABLE'
+            ").to_a.map { |r| r['table_name'] }
+
+            @all_table_names - table_names_to_skip
         end
 
-        def model(model_klass)
-            self.table_names << model_klass.table_name
-        end 
-
-        def table_names
-            unless defined? @table_names
-                @table_names = Set.new 
-            end
-            @table_names
+        # you can add more tables to this if necessary.  For example,
+        # we had to add spatial_ref_sys, which is created by the
+        # postgis extension
+        def table_names_to_skip
+            @table_names_to_skip ||= ['schema_migrations']
         end
 
         def builders
@@ -67,6 +73,17 @@ class Fixturies
         if record.id.nil?
             raise ArgumentError.new("No id for record.  Must be saved before calling identify")
         end
+
+        # ensure that we are not assigning the same name to two
+        # different records
+        @records_by_name ||= {}
+        @records_by_name[record.class.table_name] ||= {}
+        existing_entry = @records_by_name[record.class.table_name][name]
+        if existing_entry && existing_entry != record
+            raise "Cannot assign the name #{name.inspect} to two different #{record.class.table_name}"
+        end
+        @records_by_name[record.class.table_name][name] = record
+
         record_identifiers[record_key(record)] = name
     end
 
@@ -84,12 +101,16 @@ class Fixturies
 
         FileUtils.mkdir_p(self.class.fixtures_directory)
 
-        self.class.table_names.each do |table_name|
-            
+        self.class.table_names.sort.each do |table_name|
+
+            filename = Rails.root.join(self.class.fixtures_directory, "#{table_name}.yml")
+            File.delete(filename) if File.exist?(filename)
+
             # create a simple ActiveRecord klass to connect
             # to the table and handle querying for us
             klass = Class.new(ActiveRecord::Base) {
                 self.table_name = table_name
+                self.inheritance_column = nil # do not blow up if the type column indicates we should be using single-table inheritance
             }
 
             hash = {}
@@ -97,9 +118,17 @@ class Fixturies
                 name = record_identifiers[record_key(record)] || "#{table_name.singularize}_#{i}"
                 hash[name] = record.attributes
             end
-            
-            File.open(Rails.root.join(self.class.fixtures_directory, "#{table_name}.yml"), 'w+') do |f|
-                f.write(hash.to_yaml)
+
+            if hash.any?
+                # Rails allows baking the model_class into the YAML file, which
+                # can be useful if the class cannot be inferred from the table_name.
+                # See https://github.com/rails/rails/pull/20574/files
+                klass = self.class.fixture_class_names&.[](table_name)
+                hash['_fixture'] = {'model_class' => klass.to_s} if klass.present?
+
+                File.open(filename, 'w+') do |f|
+                    f.write(hash.to_yaml)
+                end
             end
 
         end
@@ -108,11 +137,12 @@ class Fixturies
     private
     def clear_db
         self.class.table_names.each do |table_name|
+            # Note: This does not handle foreign key constraints.  See Readme for a solution
             quoted_table_name = ActiveRecord::Base.connection.quote_table_name(table_name)
             sql = "DELETE FROM #{quoted_table_name} "
             ActiveRecord::Base.connection.delete(sql, "#{quoted_table_name} Delete all")
         end
-        
+
     end
 
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fixturies
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 1.0.3
 platform: ruby
 authors:
 - Nate Brustein
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-22 00:00:00.000000000 Z
+date: 2018-04-17 00:00:00.000000000 Z
 dependencies: []
 description: The speed of fixtures meets the maintanability of factories
 email: nate@pedago.com
@@ -27,17 +27,16 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 3.0.8
 signing_key: 
 specification_version: 4
 summary: The speed of fixtures meets the maintanability of factories