Sutto-reversible_data 0.0.5 → 0.1.0

data/lib/reversible_data/table.rb

@@ -62,7 +62,7 @@ module ReversibleData
       return if Object.const_defined?(@model_name)
       @model = Class.new(ActiveRecord::Base)
       @model.class_eval(&@model_definition) unless @model_definition.nil?
-      @model.blueprint(&@blueprint) unless @blueprint.nil?
+      @model.blueprint(&@blueprint) unless @blueprint.nil? || !@model.respond_to?(:blueprint)
       Object.const_set(@model_name, @model)
     end
     

data/test/table_test.rb

@@ -76,9 +76,25 @@ class TableTest < Test::Unit::TestCase
       assert !defined?(::User)
     end
     
-    should 'let you call up! to create the table and the model'
+    should 'let you call up! to create the table and the model' do
+      assert !defined?(::User)
+      assert !ActiveRecord::Base.connection.table_exists?(:users)
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.up!
+      assert defined?(::User)
+      assert ActiveRecord::Base.connection.table_exists?(:users)
+      assert User < ActiveRecord::Base
+    end
     
-    should 'let you call down! to clean up'
+    should 'let you call down! to clean up' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.up!
+      assert ActiveRecord::Base.connection.table_exists?(:users)
+      assert User < ActiveRecord::Base
+      table.down!
+      assert !ActiveRecord::Base.connection.table_exists?(:users)
+      assert !defined?(::User)
+    end
   
     should 'default to skipping model if the constant exists' do
       ::Awesome = true
@@ -107,7 +123,18 @@ class TableTest < Test::Unit::TestCase
       table.remove_model
     end
   
-    should 'let you specify a blueprint for machinist'
+    should 'let not call the blueprint if none is specified' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.up!
+      assert !User.blueprint_called
+    end
+    
+    should 'call the blueprint if present' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.blueprint {}
+      table.up!
+      assert User.blueprint_called
+    end
   
   end
 end

data/test/test_helper.rb

@@ -7,4 +7,19 @@ require 'reversible_data'
 
 class Test::Unit::TestCase
   include RR::Adapters::TestUnit
+end
+
+module MockBlueprints
+  
+  def blueprint(&blk)
+    # Use this as a kind of hacky way to ensure it is called.
+    self.blueprint_called = true
+  end
+  
+end
+
+ActiveRecord::Base.class_eval do
+  class_inheritable_accessor :blueprint_called
+  self.blueprint_called = false
+  extend MockBlueprints
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: Sutto-reversible_data
 version: !ruby/object:Gem::Version 
-  version: 0.0.5
+  version: 0.1.0
 platform: ruby
 authors: 
 - Darcy Laycock
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-08-06 00:00:00 -07:00
+date: 2009-09-12 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -33,7 +33,6 @@ files:
 - test/test_helper.rb
 has_rdoc: false
 homepage: http://sutto.net/
-licenses: 
 post_install_message: 
 rdoc_options: []
 
@@ -54,7 +53,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.5
+rubygems_version: 1.2.0
 signing_key: 
 specification_version: 3
 summary: Reversible Data provides migration-like functionality for tests etc - All with temporary models.