dibber 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: df49d47736e3cf6e58a22a2b84d77e76912c8d32
-  data.tar.gz: 03277a6b8d5f532f6ba343bfba8dc71a789785ee
+  metadata.gz: 4e8e76d2b2715019b02b55446e6445a4f12fd16c
+  data.tar.gz: f4fd83aabdc6286b9c3a64b2fa50ce017eff50fe
 SHA512:
-  metadata.gz: 330909b6d4fe39385438b0eda1984c7f1f9571cae69eeba9732bd0927ae9d027ea0f4a6b7fa6324991d446158f89373235bf34c3685f92097cf41e50e0eb4f0f
-  data.tar.gz: 627b1582b890c6721a35f969c15c338372b1f96aa445e48af4ff4f3a12f9d56690a088557a5034c1c2ba44a32c5f0e7da6a59dbb9dbcffe9ce8235f5fcfd7619
+  metadata.gz: d8bfef835c9cf35d2f556e4c731b3e0a396faacc1e7a5e3061b3b9b8d6b9a10efb96ae1d00b1a54aa5c853548eabfa325bd778771ce6d41aa13c97a874ac582b
+  data.tar.gz: 18dabf89066e0cf69f0763ebfee6ac333af2e27a932bb68003bb09f869911d3ec6e28b73a7f480298ba49e151555b44445f7ea86c3a19368504c35cf47823bf9

data/README.rdoc

@@ -48,6 +48,27 @@ You'll then be able to do this:
     thing = Thing.find_by(name: 'foo')
     thing.colour    ---> 'red'
 
+=== Pass the name of the class to seed
+
+All of these are equivalent commands
+
+    Dibber::Seeder.seed Thing
+    Dibber::Seeder.seed :thing
+    Dibber::Seeder.seed 'Thing'
+
+=== Dynamic content
+
+Seed content can also have dynamic content added via ERB. For example, if
+you want the colour of the `foo` thing to be either 'red' or 'yellow', change
+`db/seeds/things.yml` to
+
+    foo:
+      colour: <%= ['red', 'yellow'].sample %>
+      size: large
+
+Then when we run `Dibber::Seeder.seed :thing` a new `Thing` will be created
+with the colour set to either 'red' or 'yellow'.
+
 == Report
 
 Seeder.report outputs a report detailing start and end time, and a log of how

data/lib/dibber/seeder.rb

@@ -1,5 +1,6 @@
 require 'active_support/inflector'
 require 'yaml'
+require 'erb'
 
 module Dibber
   class Seeder
@@ -9,7 +10,7 @@ module Dibber
 
       def seed(klass, args = {})
         if klass.kind_of?(String) || klass.kind_of?(Symbol)
-          name = klass.to_s
+          name = klass.to_s.underscore
           class_name = klass.to_s.strip.classify
           klass = Kernel.const_get(class_name)
         else
@@ -39,7 +40,15 @@ module Dibber
       end
 
       def objects_from(file)
-        YAML.load_file("#{seeds_path}#{file}")
+        YAML.load erb_processed(file_content(file))
+      end
+
+      def erb_processed(content)
+        ERB.new(content).result
+      end
+
+      def file_content(file)
+        File.read File.join(seeds_path, file)
       end
 
       def seeds_path
@@ -130,4 +139,4 @@ module Dibber
     end
 
   end
-end
+end

data/lib/dibber/version.rb

@@ -1,9 +1,12 @@
 module Dibber
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end
 
 # History
 # =======
+# 0.6.0 ERB parse the source files before passing it to YAML. Allows dynamic
+#       content to be used.
+#
 # 0.5.0 Allow the `Seeder.seed` method to take a Class or the string/symbol
 #       representation of the class name.
 #

data/test/dibber/dynamic_thing_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+require_relative 'thing'
+
+module Dibber
+  class DynamicThingTest < Minitest::Test
+
+    def setup
+      Seeder.seeds_path = File.join(File.dirname(__FILE__), 'seeds')
+    end
+
+
+    def test_objects_from
+      hash = Seeder.objects_from('dynamic_things.yml')
+      assert_equal 'something', hash['one']['title']
+    end
+
+  end
+end

data/test/dibber/seeder_test.rb

@@ -38,7 +38,7 @@ module Dibber
     def test_clear_process_log
       test_monitor
       Seeder.clear_process_log
-      assert_equal(nil, Seeder.process_log.raw[:things])
+      assert_nil Seeder.process_log.raw[:things], "Log should be empty"
     end
 
     def test_seeds_path
@@ -179,7 +179,17 @@ module Dibber
       assert_equal({'title' => 'one'}, foo.attributes)
     end
 
-     def test_seed_with_class_with_alternative_name_method
+    def test_seed_with_class_name
+      assert_equal(0, Thing.count)
+      Seeder.seed('Thing')
+      assert_equal(2, Thing.count)
+      foo = Thing.find_or_initialize_by(name: :foo)
+      bar = Thing.find_or_initialize_by(name: :bar)
+      assert_equal([foo, bar], Thing.saved)
+      assert_equal({'title' => 'one'}, foo.attributes)
+    end
+
+    def test_seed_with_class_with_alternative_name_method
       Seeder.seed(Thing, :name_method => 'other_method')
       assert_equal(2, Thing.count)
       foo = Thing.find_or_initialize_by(other_method: :foo)
@@ -244,7 +254,7 @@ module Dibber
     end
 
     def thing_file_path
-      File.join(File.dirname(__FILE__),'seeds', 'things.yml')
+      File.join(File.dirname(__FILE__), 'seeds', 'things.yml')
     end
   end
 

data/test/dibber/seeds/dynamic_things.yml

@@ -0,0 +1,2 @@
+one:
+  title: <%= 'something' %>

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dibber
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Rob Nichols
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-04-29 00:00:00.000000000 Z
+date: 2017-03-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -53,9 +53,11 @@ files:
 - lib/dibber/process_log.rb
 - lib/dibber/seeder.rb
 - lib/dibber/version.rb
+- test/dibber/dynamic_thing_test.rb
 - test/dibber/foo/bar.rb
 - test/dibber/process_log_test.rb
 - test/dibber/seeder_test.rb
+- test/dibber/seeds/dynamic_things.yml
 - test/dibber/seeds/empty.yml
 - test/dibber/seeds/foo/bars.yml
 - test/dibber/seeds/things.yml
@@ -99,24 +101,26 @@ signing_key:
 specification_version: 4
 summary: Tool for seeding database from YAML.
 test_files:
-- test/dibber/foo/bar.rb
-- test/dibber/seeder_test.rb
 - test/dibber/thing.rb
+- test/dibber/seeder_test.rb
+- test/dibber/dynamic_thing_test.rb
+- test/dibber/foo/bar.rb
 - test/dibber/process_log_test.rb
+- test/dibber/thing_test.rb
+- test/dibber/seeds/dynamic_things.yml
+- test/dibber/seeds/empty.yml
 - test/dibber/seeds/foo/bars.yml
 - test/dibber/seeds/things.yml
-- test/dibber/seeds/empty.yml
-- test/dibber/thing_test.rb
-- test/examples/seeds.rb
+- test/examples/process_logs.rb
 - test/examples/models/borough.rb
-- test/examples/models/disclaimer.rb
-- test/examples/models/category.rb
 - test/examples/models/fee.rb
+- test/examples/models/disclaimer.rb
 - test/examples/models/admin_user.rb
+- test/examples/models/category.rb
+- test/examples/seeds.rb
+- test/examples/seeds/boroughs.yml
 - test/examples/seeds/fees.yml
 - test/examples/seeds/categories.yml
 - test/examples/seeds/disclaimer/documents.yml
-- test/examples/seeds/boroughs.yml
-- test/examples/process_logs.rb
-- test/not_quite_active_record.rb
 - test/test_helper.rb
+- test/not_quite_active_record.rb