active_record_shards 3.0.0.beta1 → 3.0.0.beta2

data/test/cowardly_migration/20110824010215_cowardly_migration.rb

@@ -1,8 +0,0 @@
-class CowardlyMigration < ActiveRecord::Migration
-  def self.up
-    # shouldn't get here.  must fail.
-  end
-
-  def self.down
-  end
-end

data/test/database.yml

@@ -1,73 +0,0 @@
-test:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test
-  username: root
-  password:
-  host: 127.0.0.1
-  shard_names: ['0', '1']
-
-test_slave:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test_slave
-  username: root
-  password:
-  host: 127.0.0.1
-
-test_shard_0:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test_shard0
-  username: root
-  password:
-  host: 127.0.0.1
-
-test_shard_0_slave:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test_shard0_slave
-  username: root
-  password:
-  host: 127.0.0.1
-
-test_shard_1:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test_shard1
-  username: root
-  password:
-  host: 127.0.0.1
-
-test_shard_1_slave:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test_shard1_slave
-  username: root
-  password:
-  host: 127.0.0.1
-
-test2:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test2
-  username: root
-  password:
-  host: 127.0.0.1
-
-test2_slave:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test2_slave
-  username: root
-  password:
-  host: 127.0.0.1
-
-alternative:
-  adapter: mysql2
-  encoding: utf8
-  database: ars_test_alternative
-  username: root
-  password:
-  host: 127.0.0.1
-  shard_names: ['0', '1']

data/test/database_parse_test.yml

@@ -1,21 +0,0 @@
-test:
-  adapter: mysql
-  encoding: utf8
-  database: ars_test
-  port: 123
-  username: root
-  password:
-  host: main_host
-  slave:
-    host: main_slave_host
-  shards:
-    a:
-      database: ars_test_shard_a
-      host: shard_a_host
-      slave:
-        host: shard_a_slave_host
-    b:
-      database: ars_test_shard_b
-      host: shard_b_host
-      slave:
-        database: ars_test_shard_b_slave

data/test/failure_migration/20110824010215_failure_migration.rb

@@ -1,14 +0,0 @@
-class FailureMigration < ActiveRecord::Migration
-  shard :all
-
-  def self.up
-    $fail_at_two ||= 0
-    $fail_at_two += 1
-    raise "FAIL FAIL FAIL" if $fail_at_two == 2
-    add_column :tickets, :sharded_column, :integer
-  end
-
-  def self.down
-    remove_column :tickets, :sharded_column
-  end
-end

data/test/helper.rb

@@ -1,57 +0,0 @@
-require 'bundler/setup'
-Bundler.require
-
-require 'minitest/autorun'
-MiniTest::Reporters.use! MiniTest::Reporters::DefaultReporter.new
-
-if defined?(Debugger)
-  ::Debugger.start
-  ::Debugger.settings[:autoeval] = true if ::Debugger.respond_to?(:settings)
-end
-
-$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
-$LOAD_PATH.unshift(File.dirname(__FILE__))
-require 'active_record_shards'
-require 'logger'
-
-RAILS_ENV = "test"
-
-ActiveRecord::Base.logger = Logger.new(File.dirname(__FILE__) + "/test.log")
-ActiveRecord::Base.configurations = YAML::load(IO.read(File.dirname(__FILE__) + '/database.yml'))
-
-def init_schema
-  ActiveRecord::Base.configurations.each do |name, conf|
-    `echo "drop DATABASE if exists #{conf['database']}" | mysql --user=#{conf['username']}`
-    `echo "create DATABASE #{conf['database']}" | mysql --user=#{conf['username']}`
-    ActiveRecord::Base.establish_connection(name)
-    load(File.dirname(__FILE__) + "/schema.rb")
-  end
-end
-init_schema
-
-require 'models'
-
-require 'active_support/test_case'
-class Minitest::Spec
-  def clear_databases
-    ActiveRecord::Base.configurations.each do |name, conf|
-      ActiveRecord::Base.establish_connection(name)
-      ActiveRecord::Base.connection.execute("DELETE FROM accounts")
-      ActiveRecord::Base.connection.execute("DELETE FROM tickets")
-    end
-    ActiveRecord::Base.establish_connection(RAILS_ENV)
-  end
-  before { clear_databases }
-
-  def assert_using_master_db
-    assert_using_database('ars_test')
-  end
-
-  def assert_using_slave_db
-    assert_using_database('ars_test_slave')
-  end
-
-  def assert_using_database(db_name, model = ActiveRecord::Base)
-    assert_equal(db_name, model.connection.current_database)
-  end
-end

data/test/migrations/20110824010216_shard_migration.rb

@@ -1,12 +0,0 @@
-#This is to recreate the deleted onetime payment for DealFish
-class ShardMigration < ActiveRecord::Migration
-  shard :all
-
-  def self.up
-    add_column :emails, :sharded_column, :integer
-  end
-
-  def self.down
-    remove_column :emails, :sharded_column
-  end
-end

data/test/migrations/20110829215912_account_migration.rb

@@ -1,13 +0,0 @@
-# See ticket 130214
-# Setting up an odd initial three year term for box.net to be paid by credit card by the normal means
-class AccountMigration < ActiveRecord::Migration
-  shard :none
-
-  def self.up
-    add_column :accounts, :non_sharded_column, :integer
-  end
-
-  def self.down
-    remove_column :accounts, :non_sharded_column
-  end
-end

data/test/migrator_test.rb

@@ -1,113 +0,0 @@
-require_relative 'helper'
-
-class CowardlyMigration < ActiveRecord::Migration
-  def self.up
-    "not gonna happen"
-  end
-
-  def self.down
-    "uh uh"
-  end
-end
-
-describe ActiveRecord::Migrator do
-  before do
-    init_schema
-  end
-
-  it "migrates" do
-    migration_path = File.join(File.dirname(__FILE__), "/migrations")
-    ActiveRecord::Migrator.migrate(migration_path)
-    ActiveRecord::Base.on_all_shards do
-      assert ActiveRecord::Base.connection.table_exists?(:schema_migrations), "Schema Migrations doesn't exist"
-      assert ActiveRecord::Base.connection.table_exists?(:accounts)
-      assert ActiveRecord::Base.connection.select_value("select version from schema_migrations where version = '20110824010216'")
-      assert ActiveRecord::Base.connection.select_value("select version from schema_migrations where version = '20110829215912'")
-    end
-
-    ActiveRecord::Base.on_all_shards do
-      assert table_has_column?("emails", "sharded_column")
-      assert !table_has_column?("accounts", "non_sharded_column")
-    end
-
-    ActiveRecord::Base.on_shard(nil) do
-      assert !table_has_column?("emails", "sharded_column")
-      assert table_has_column?("accounts", "non_sharded_column")
-    end
-
-    # now test down/ up
-    ActiveRecord::Migrator.run(:down, migration_path, 20110824010216)
-    ActiveRecord::Base.on_all_shards do
-      assert !table_has_column?("emails", "sharded_column")
-    end
-
-    ActiveRecord::Migrator.run(:down, migration_path, 20110829215912)
-    ActiveRecord::Base.on_shard(nil) do
-      assert !table_has_column?("accounts", "non_sharded_column")
-    end
-
-    ActiveRecord::Migrator.run(:up, migration_path, 20110824010216)
-    ActiveRecord::Base.on_all_shards do
-      assert table_has_column?("emails", "sharded_column")
-    end
-
-    ActiveRecord::Migrator.run(:up, migration_path, 20110829215912)
-    ActiveRecord::Base.on_shard(nil) do
-      assert table_has_column?("accounts", "non_sharded_column")
-    end
-  end
-
-  it "does not migrate bad migrations" do
-    migration_path = File.join(File.dirname(__FILE__), "/cowardly_migration")
-    exception = nil
-    begin
-      ActiveRecord::Migrator.migrate(migration_path)
-    rescue Exception => e
-      exception = e
-    end
-    assert e
-  end
-
-  it "fails with failing migrations" do
-    # like, if you have to break a migration in the middle somewhere.
-    migration_path = File.join(File.dirname(__FILE__), "/failure_migration")
-
-    assert failure_migration_pending?(migration_path)
-    begin
-      ActiveRecord::Migrator.migrate(migration_path)
-    rescue
-      # after first fail, should still be pending
-      assert failure_migration_pending?(migration_path)
-      retry
-    end
-
-    assert !failure_migration_pending?(migration_path)
-    ActiveRecord::Base.on_all_shards do
-      assert table_has_column?("tickets", "sharded_column")
-    end
-  end
-
-  describe "#shard_status" do
-    it "shows nothing if everything is ok" do
-      ActiveRecord::Migrator.shard_status([1]).must_equal([{}, {}])
-    end
-
-    it "shows missing migrations" do
-      ActiveRecord::Migrator.shard_status([]).must_equal([{}, {nil => [1], "0" => [1], "1" => [1]}])
-    end
-
-    it "shows pending migrations" do
-      ActiveRecord::Migrator.shard_status([1, 2]).must_equal([{nil => [2], "0" => [2], "1" => [2]}, {}])
-    end
-  end
-
-  private
-
-  def failure_migration_pending?(migration_path)
-    ActiveRecord::Migrator.new(:up, migration_path).pending_migrations.detect { |f| f.name == "FailureMigration" }
-  end
-
-  def table_has_column?(table, column)
-    !ActiveRecord::Base.connection.select_values("desc #{table}").grep(column).empty?
-  end
-end

data/test/models.rb

@@ -1,38 +0,0 @@
-class Account < ActiveRecord::Base
-  # attributes: id, name, updated_at, created_at
-  not_sharded
-
-  has_many :tickets
-  has_many :account_things
-end
-
-class AccountThing < ActiveRecord::Base
-  not_sharded
-
-  if respond_to?(:where)
-    scope :enabled, where(:enabled => true)
-  else
-    named_scope :enabled, :conditions => {:enabled => true}
-  end
-end
-
-class Email < ActiveRecord::Base
-  not_sharded
-  establish_connection_override :alternative
-end
-
-class AccountInherited < Account
-end
-
-class Ticket < ActiveRecord::Base
-  # attributes: id, title, account_id, updated_at, created_at
-  belongs_to :account
-end
-
-class Person < ActiveRecord::Base
-  not_sharded
-end
-
-class User < Person
-end
-

data/test/schema.rb

@@ -1,36 +0,0 @@
-ActiveRecord::Migration.verbose = false
-
-ActiveRecord::Schema.define(:version => 1) do
-  create_table "accounts", :force => true do |t|
-    t.string   "name"
-    t.datetime "created_at"
-    t.datetime "updated_at"
-  end
-
-  create_table "account_things", :force => true do |t|
-    t.integer "account_id"
-    t.boolean "enabled", :default => true
-  end
-
-  create_table "emails", :force => true do |t|
-    t.string   "from"
-    t.string   "to"
-    t.text     "mail"
-    t.datetime "created_at"
-    t.datetime "updated_at"
-  end
-
-  create_table "tickets", :force => true do |t|
-    t.string   "title"
-    t.integer  "account_id"
-    t.datetime "created_at"
-    t.datetime "updated_at"
-  end
-
-  create_table "people", :force => true do |t|
-    t.string "name"
-    t.string "type"
-    t.datetime "created_at"
-    t.datetime "updated_at"
-  end
-end