soveran-spawner 0.0.7 → 0.0.8

data/lib/spawner.rb

@@ -1,4 +1,4 @@
-require 'ostruct'
+require "ostruct"
 
 module Spawner
   def spawner &default
@@ -8,6 +8,7 @@ module Spawner
 
   def spawn attrs = {}
     @@spawn[self.name].call(model = OpenStruct.new)
-    create!(model.send(:table).merge(attrs))
+    factory_method = respond_to?(:create!) ? :create! : :create
+    send(factory_method, model.send(:table).merge(attrs))
   end
 end

data/spawner.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = "spawner"
-  s.version = "0.0.7"
+  s.version = "0.0.8"
   s.date = "2008-12-18"
   s.summary = "Simple fixtures replacement for Sequel, ActiveRecord and probably many other ORMs"
   s.email = "michel@soveran.com"
@@ -15,5 +15,7 @@ Gem::Specification.new do |s|
     "lib/spawner.rb",
     "rails/init.rb",
     "extras/samples.rake",
-    "test/all_test.rb"]
+    "test/all_test.rb",
+    "test/sequel_test.rb",
+    "test/active_record_test.rb"]
 end

data/test/active_record_test.rb

@@ -0,0 +1,44 @@
+require "rubygems"
+require "active_record"
+require "contest"
+require File.dirname(__FILE__) + "/../lib/spawner"
+require "faker"
+
+ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :dbfile  => ":memory:")
+ActiveRecord::Schema.define do
+  create_table :active_record_users do |table|
+      table.column :name, :string
+      table.column :email, :string
+  end
+end
+
+class ActiveRecordUser < ActiveRecord::Base
+  extend Spawner
+
+  validates_presence_of :name
+
+  spawner do |user|
+    user.name = Faker::Name.name
+    user.email = Faker::Internet.email
+  end
+end
+
+class TestSpawnerWithActiveRecord < Test::Unit::TestCase
+  setup do
+    @user = ActiveRecordUser.spawn :name => "John"
+  end
+
+  context "spawned user" do
+    should "have John as name" do
+      assert_equal "John", @user.name
+    end
+
+    context "with invalid attributes" do
+      should "raise an error" do
+        assert_raise ActiveRecord::RecordInvalid do
+          ActiveRecordUser.spawn :name => nil
+        end
+      end
+    end
+  end
+end

data/test/all_test.rb

@@ -2,14 +2,14 @@ require 'rubygems'
 require 'contest'
 require File.dirname(__FILE__) + "/../lib/spawner"
 
-class Foo
+class Base
   attr_accessor :attributes
 
   def initialize(attrs = {})
     @attributes = attrs
   end
 
-  def self.create!(attrs = {})
+  def self.create(attrs = {})
     new(attrs)
   end
 
@@ -19,9 +19,15 @@ class Foo
 
   extend Spawner
 
-  spawner do |foo|
-    foo.bar = 7
-    foo.baz = 8
+  spawner do |object|
+    object.bar = 7
+    object.baz = 8
+  end
+end
+
+class Foo < Base
+  class << self
+    alias_method :create!, :create
   end
 end
 
@@ -34,6 +40,9 @@ class Bar < Foo
   end
 end
 
+class Baz < Base
+end
+
 class TestFoo < Test::Unit::TestCase
   should "have a name class method" do
     assert Foo.respond_to?(:name)
@@ -97,5 +106,12 @@ class TestFoo < Test::Unit::TestCase
         end
       end
     end
+
+    context "and it doesn't understand #create! (only #create)" do
+      should "work anyway" do
+        baz = Baz.spawn :bar => 1
+        assert_equal 1, baz.bar
+      end
+    end
   end
 end

data/test/sequel_test.rb

@@ -0,0 +1,41 @@
+require "rubygems"
+require "sequel"
+require "contest"
+require File.dirname(__FILE__) + "/../lib/spawner"
+require "faker"
+
+DB = Sequel.sqlite
+DB << "CREATE TABLE sequel_users (name VARCHAR(255) NOT NULL, email VARCHAR(255) NOT NULL)"
+
+class SequelUser < Sequel::Model
+  extend Spawner
+
+  validates do
+    presence_of :name
+  end
+
+  spawner do |user|
+    user.name = Faker::Name.name
+    user.email = Faker::Internet.email
+  end
+end
+
+class TestSpawnerWithSequel < Test::Unit::TestCase
+  setup do
+    @user = SequelUser.spawn :name => "John"
+  end
+
+  context "spawned user" do
+    should "have John as name" do
+      assert_equal "John", @user.name
+    end
+
+    context "with invalid attributes" do
+      should "raise an error" do
+        assert_raise Sequel::Error::InvalidValue do
+          SequelUser.spawn :name => nil
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: soveran-spawner
 version: !ruby/object:Gem::Version 
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors: 
 - Michel Martens
@@ -29,6 +29,8 @@ files:
 - rails/init.rb
 - extras/samples.rake
 - test/all_test.rb
+- test/sequel_test.rb
+- test/active_record_test.rb
 has_rdoc: false
 homepage: http://github.com/soveran/spawner
 post_install_message: