fixturease 0.2.5 → 0.2.6

data/Rakefile

@@ -7,7 +7,7 @@ desc 'Default: run unit tests.'
 task :default => :spec
 
 
-PKG_VERSION="0.2.5"
+PKG_VERSION="0.2.6"
 PKG_FILES=["bin/fixturease.rb","Rakefile","spec/database.yml","spec/fixturease_spec.rb"]
 
 spec = Gem::Specification.new do |s|

data/bin/fixturease.rb

@@ -11,6 +11,26 @@ else
 
   $reload_fixtures = false
 
+  def reset_sequence!(mod)
+    if (adapter = YAML::load_file(ENV['RAILS_ROOT']+'/config/database.yml')[ENV['RAILS_ENV']]['adapter']) == "mysql"
+      ActiveRecord::Base.connection.rollback_db_transaction
+      ActiveRecord::Base.send :decrement_open_transactions
+      klass = mod.is_a?(Class) ? mod : mod.singularize.camelize.constantize
+      max_id = klass.find(:first,:order => "id DESC").id
+      ActiveRecord::Base.connection.execute "ALTER TABLE #{klass.table_name} AUTO_INCREMENT=#{max_id}"
+      ActiveRecord::Base.send :increment_open_transactions
+      ActiveRecord::Base.connection.begin_db_transaction
+    else
+      puts "Database adapter #{adapter} is not yet supported. Sorry"
+    end
+  end
+  
+  def reset_sequences!
+    Dir[ENV['RAILS_ROOT']+'/'+ENV['FIXTURES_ROOT']+'/*.yml'].each do |fixture_file|
+      reset_sequence! File.basename(fixture_file,".yml")
+    end
+  end
+
   def load_fixtures!
     $reload_fixtures = true
     require 'active_record/fixtures'
@@ -56,6 +76,7 @@ else
   ActiveRecord::Base.send :increment_open_transactions
   ActiveRecord::Base.connection.begin_db_transaction
   load_fixtures! if ENV['LOAD_FIXTURES']=='yes'
+  reset_sequences!
   at_exit do
     vars = {} 
     instance_variables.each {|varname| vars[instance_variable_get(varname)]=varname.gsub(/^@/,"") if instance_variable_get(varname).is_a? ActiveRecord::Base}

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.0
 specification_version: 1
 name: fixturease
 version: !ruby/object:Gem::Version 
-  version: 0.2.5
-date: 2007-01-20 00:00:00 +02:00
+  version: 0.2.6
+date: 2007-01-21 00:00:00 +02:00
 summary: Easy fixture creation tool
 require_paths: 
 - lib