yam-db-charmer 1.7.01 → 1.7.4.0

data/lib/tasks/test.rake

@@ -0,0 +1,115 @@
+
+namespace :db_charmer do
+  namespace :test do
+    
+    desc 'Reinitialize all test databases'
+    task :prepare => ['db:abort_if_pending_migrations', 'db:load_config'] do |t|
+      old_rails_env = Rails.env
+      Rails.env = 'test'
+      unless ActiveRecord::Base.schema_format==:sql
+        abort "db_charmer can only reset databases using sql, check config.active_record.schema_format"
+      end
+      Rake::Task['db_charmer:test:clone_structure'].invoke
+      Rails.env = old_rails_env
+    end
+
+    task :purge do
+      Rake::Task['db_charmer:drop'].invoke
+      Rake::Task['db_charmer:create'].invoke
+    end
+
+    task :dump_structure do
+      ::ActiveRecord::Base.configurations['development'].each_key do |name|
+        config = ::ActiveRecord::Base.configurations['development'][name]
+        next unless config.is_a?(Hash) && config['database']
+        # Only connect to local databases
+        local_database?(config) { dump_database_structure(name, config) }
+      end
+
+      config = ::ActiveRecord::Base.configurations['development']
+      local_database?(config) { dump_database_structure('main', config) }
+    end
+
+    # desc "Recreate the test databases from the development structure"
+    task :clone_structure => [ 'db_charmer:test:dump_structure', 'db_charmer:test:purge' ] do
+      ::ActiveRecord::Base.configurations['test'].each_key do |name|
+        config = ::ActiveRecord::Base.configurations['test'][name]
+        next unless config.is_a?(Hash) && config['database']
+        # Only connect to local databases
+        local_database?(config) { clone_database_structure(name, config) }
+      end
+      config = ::ActiveRecord::Base.configurations['test']
+      local_database?(config) { clone_database_structure('main', config) }
+    end
+
+    def dump_database_structure(name, config, origin_env='development')
+      return unless config.is_a?(Hash) && config['database']
+      structure_filename = "#{Rails.root}/db/#{origin_env}_#{name}_structure.sql"
+      case config["adapter"]
+      when /^(jdbc)?mysql/, "oci", "oracle"
+        ActiveRecord::Base.establish_connection(config)
+        File.open(structure_filename, "w+") { |f| f << ActiveRecord::Base.connection.structure_dump }
+      when /^(jdbc)?postgresql$/
+        ENV['PGHOST']     = config["host"] if config["host"]
+        ENV['PGPORT']     = config["port"].to_s if config["port"]
+        search_path = config["schema_search_path"]
+        unless search_path.blank?
+          search_path = search_path.split(",").map{|search_path| "--schema=#{search_path.strip}" }.join(" ")
+        end
+        cmd ="pg_dump -U #{config["username"]} -s -x -O -f db/#{origin_env}_#{name}_structure.sql #{search_path} #{config['database']}"
+        puts cmd                                                                                                                         
+        `#{cmd}`
+        raise "Error dumping database" if $?.exitstatus == 1
+      when /^(jdbc)?sqlite/
+        dbfile = config["database"] || config["dbfile"]
+        `sqlite3 #{dbfile} .schema > #{structure_filename}`
+      when "sqlserver"
+        `scptxfr /s #{config["host"]} /d #{config["database"]} /I /f db\\#{origin_env}_#{name}_structure.sql /q /A /r`
+        `scptxfr /s #{config["host"]} /d #{config["database"]} /I /F db\ /q /A /r`
+      when "firebird"
+        set_firebird_env(config)
+        db_string = firebird_db_string(config)
+        sh "isql -a #{db_string} > #{structure_filename}"
+      else
+        raise "Task not supported by '#{config["adapter"]}'"
+      end
+    end
+
+    def clone_database_structure(name, config, origin_env='development')
+      structure_filename = "#{Rails.root}/db/#{origin_env}_#{name}_structure.sql"
+      case config["adapter"]
+      when /^(jdbc)?mysql/
+        ActiveRecord::Base.establish_connection(:test)
+        ActiveRecord::Base.connection.execute('SET foreign_key_checks = 0')
+        IO.readlines(structure_filename).join.split("\n\n").each do |table|
+          ActiveRecord::Base.connection.execute(table)
+        end
+      when /^(jdbc)?postgresql$/
+        ENV['PGHOST']     = config["host"] if config["host"]
+        ENV['PGPORT']     = config["port"].to_s if config["port"]
+        #ENV['PGPASSWORD'] = config["password"].to_s if config["password"]
+        cmd = "psql -U #{config["username"]} -f #{structure_filename} #{config["database"]}"
+        puts cmd
+        `#{cmd}`
+      when /^(jdbc)?sqlite/
+        dbfile = config["database"] || config["dbfile"]
+        `sqlite3 #{dbfile} < #{structure_filename}`
+      when "sqlserver"
+        `osql -E -S #{config["host"]} -d #{config["database"]} -i #{structure_filename}`
+      when "oci", "oracle"
+        ActiveRecord::Base.establish_connection(:test)
+        IO.readlines(structure_filename).join.split(";\n\n").each do |ddl|
+          ActiveRecord::Base.connection.execute(ddl)
+        end
+      when "firebird"
+        set_firebird_env(abcs["test"])
+        db_string = firebird_db_string(abcs["test"])
+        sh "isql -i #{structure_filename} #{db_string}"
+      else
+        raise "Task not supported by '#{config["adapter"]}'"
+      end
+
+      File.delete(structure_filename) if File.exist?(structure_filename)
+    end
+  end
+end

metadata

@@ -1,13 +1,14 @@
 --- !ruby/object:Gem::Specification 
 name: yam-db-charmer
 version: !ruby/object:Gem::Version 
-  hash: 9
+  hash: 119
   prerelease: 
   segments: 
   - 1
   - 7
-  - 1
-  version: 1.7.01
+  - 4
+  - 0
+  version: 1.7.4.0
 platform: ruby
 authors: 
 - Oleksiy Kovyrin
@@ -16,7 +17,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-06-04 00:00:00 -07:00
+date: 2012-09-18 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -25,13 +26,13 @@ dependencies:
   requirement: &id001 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
-    - - <
+    - - ~>
       - !ruby/object:Gem::Version 
-        hash: 5
+        hash: 7
         segments: 
         - 3
-        - 1
-        version: "3.1"
+        - 0
+        version: "3.0"
   type: :runtime
   version_requirements: *id001
 - !ruby/object:Gem::Dependency 
@@ -40,13 +41,13 @@ dependencies:
   requirement: &id002 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
-    - - <
+    - - ~>
       - !ruby/object:Gem::Version 
-        hash: 5
+        hash: 7
         segments: 
         - 3
-        - 1
-        version: "3.1"
+        - 0
+        version: "3.0"
   type: :runtime
   version_requirements: *id002
 - !ruby/object:Gem::Dependency 
@@ -130,16 +131,22 @@ files:
 - lib/db_charmer/rails3/active_record/master_slave_routing.rb
 - lib/db_charmer/rails3/active_record/relation/connection_routing.rb
 - lib/db_charmer/rails3/active_record/relation_method.rb
+- lib/db_charmer/rails31/active_record/migration/command_recorder.rb
+- lib/db_charmer/rails31/active_record/preloader/association.rb
+- lib/db_charmer/rails31/active_record/preloader/has_and_belongs_to_many.rb
 - lib/db_charmer/sharding.rb
 - lib/db_charmer/sharding/connection.rb
 - lib/db_charmer/sharding/method.rb
 - lib/db_charmer/sharding/method/db_block_group_map.rb
+- lib/db_charmer/sharding/method/db_block_group_map_base.rb
 - lib/db_charmer/sharding/method/db_block_map.rb
+- lib/db_charmer/sharding/method/db_block_schema_map.rb
 - lib/db_charmer/sharding/method/hash_map.rb
 - lib/db_charmer/sharding/method/range.rb
 - lib/db_charmer/sharding/stub_connection.rb
 - lib/db_charmer/version.rb
 - lib/tasks/databases.rake
+- lib/tasks/test.rake
 has_rdoc: true
 homepage: http://kovyrin.github.com/db-charmer
 licenses: []