RubyGems - dr-apartment - Versions diffs - 0.14.1 - Mend

dr-apartment 0.14.1

Files changed (78) hide show

data/.gitignore +6 -0
data/.rspec +2 -0
data/.rvmrc +1 -0
data/Gemfile +22 -0
data/HISTORY.md +133 -0
data/README.md +152 -0
data/Rakefile +79 -0
data/apartment.gemspec +32 -0
data/lib/apartment.rb +69 -0
data/lib/apartment/adapters/abstract_adapter.rb +176 -0
data/lib/apartment/adapters/jdbcpostgresql_adapter.rb +115 -0
data/lib/apartment/adapters/mysql_adapter.rb +18 -0
data/lib/apartment/adapters/postgresql_adapter.rb +114 -0
data/lib/apartment/console.rb +12 -0
data/lib/apartment/database.rb +57 -0
data/lib/apartment/delayed_job/active_record.rb +20 -0
data/lib/apartment/delayed_job/enqueue.rb +20 -0
data/lib/apartment/delayed_job/hooks.rb +25 -0
data/lib/apartment/delayed_job/requirements.rb +23 -0
data/lib/apartment/elevators/subdomain.rb +27 -0
data/lib/apartment/migrator.rb +23 -0
data/lib/apartment/railtie.rb +54 -0
data/lib/apartment/reloader.rb +24 -0
data/lib/apartment/version.rb +3 -0
data/lib/tasks/apartment.rake +70 -0
data/spec/adapters/mysql_adapter_spec.rb +36 -0
data/spec/adapters/postgresql_adapter_spec.rb +137 -0
data/spec/apartment_spec.rb +11 -0
data/spec/config/database.yml +13 -0
data/spec/dummy/Rakefile +7 -0
data/spec/dummy/app/controllers/application_controller.rb +6 -0
data/spec/dummy/app/helpers/application_helper.rb +2 -0
data/spec/dummy/app/models/company.rb +3 -0
data/spec/dummy/app/models/user.rb +3 -0
data/spec/dummy/app/views/application/index.html.erb +1 -0
data/spec/dummy/app/views/layouts/application.html.erb +14 -0
data/spec/dummy/config.ru +4 -0
data/spec/dummy/config/application.rb +47 -0
data/spec/dummy/config/boot.rb +10 -0
data/spec/dummy/config/database.yml +16 -0
data/spec/dummy/config/environment.rb +5 -0
data/spec/dummy/config/environments/development.rb +26 -0
data/spec/dummy/config/environments/production.rb +49 -0
data/spec/dummy/config/environments/test.rb +35 -0
data/spec/dummy/config/initializers/apartment.rb +4 -0
data/spec/dummy/config/initializers/backtrace_silencers.rb +7 -0
data/spec/dummy/config/initializers/inflections.rb +10 -0
data/spec/dummy/config/initializers/mime_types.rb +5 -0
data/spec/dummy/config/initializers/secret_token.rb +7 -0
data/spec/dummy/config/initializers/session_store.rb +8 -0
data/spec/dummy/config/locales/en.yml +5 -0
data/spec/dummy/config/routes.rb +3 -0
data/spec/dummy/db/migrate/20110613152810_create_dummy_models.rb +37 -0
data/spec/dummy/db/migrate/20111202022214_create_table_books.rb +13 -0
data/spec/dummy/db/schema.rb +48 -0
data/spec/dummy/db/seeds.rb +8 -0
data/spec/dummy/db/test.sqlite3 +0 -0
data/spec/dummy/lib/fake_dj_class.rb +6 -0
data/spec/dummy/public/404.html +26 -0
data/spec/dummy/public/422.html +26 -0
data/spec/dummy/public/500.html +26 -0
data/spec/dummy/public/favicon.ico +0 -0
data/spec/dummy/public/stylesheets/.gitkeep +0 -0
data/spec/dummy/script/rails +6 -0
data/spec/integration/apartment_rake_integration_spec.rb +74 -0
data/spec/integration/database_integration_spec.rb +200 -0
data/spec/integration/delayed_job_integration_spec.rb +100 -0
data/spec/integration/middleware/subdomain_elevator_spec.rb +63 -0
data/spec/spec_helper.rb +31 -0
data/spec/support/apartment_helpers.rb +32 -0
data/spec/support/capybara_sessions.rb +15 -0
data/spec/support/config.rb +11 -0
data/spec/tasks/apartment_rake_spec.rb +118 -0
data/spec/unit/config_spec.rb +78 -0
data/spec/unit/middleware/subdomain_elevator_spec.rb +20 -0
data/spec/unit/migrator_spec.rb +87 -0
data/spec/unit/reloader_spec.rb +22 -0
metadata +144 -0

@@ -0,0 +1,8 @@
+def create_users
+  3.times do |x|
+    user = User.find_or_initialize_by_name "Some User #{x}"
+    user.save
+  end
+end
+create_users

data/spec/dummy/db/test.sqlite3 ADDED

Binary file

data/spec/dummy/lib/fake_dj_class.rb ADDED

@@ -0,0 +1,6 @@
+class FakeDjClass
+  def perform
+  end
+end

data/spec/dummy/public/404.html ADDED

@@ -0,0 +1,26 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>The page you were looking for doesn't exist (404)</title>
+  <style type="text/css">
+    body { background-color: #fff; color: #666; text-align: center; font-family: arial, sans-serif; }
+    div.dialog {
+      width: 25em;
+      padding: 0 4em;
+      margin: 4em auto 0 auto;
+      border: 1px solid #ccc;
+      border-right-color: #999;
+      border-bottom-color: #999;
+    }
+    h1 { font-size: 100%; color: #f00; line-height: 1.5em; }
+  </style>
+</head>
+<body>
+  <!-- This file lives in public/404.html -->
+  <div class="dialog">
+    <h1>The page you were looking for doesn't exist.</h1>
+    <p>You may have mistyped the address or the page may have moved.</p>
+  </div>
+</body>
+</html>

data/spec/dummy/public/422.html ADDED

@@ -0,0 +1,26 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>The change you wanted was rejected (422)</title>
+  <style type="text/css">
+    body { background-color: #fff; color: #666; text-align: center; font-family: arial, sans-serif; }
+    div.dialog {
+      width: 25em;
+      padding: 0 4em;
+      margin: 4em auto 0 auto;
+      border: 1px solid #ccc;
+      border-right-color: #999;
+      border-bottom-color: #999;
+    }
+    h1 { font-size: 100%; color: #f00; line-height: 1.5em; }
+  </style>
+</head>
+<body>
+  <!-- This file lives in public/422.html -->
+  <div class="dialog">
+    <h1>The change you wanted was rejected.</h1>
+    <p>Maybe you tried to change something you didn't have access to.</p>
+  </div>
+</body>
+</html>

data/spec/dummy/public/500.html ADDED

@@ -0,0 +1,26 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>We're sorry, but something went wrong (500)</title>
+  <style type="text/css">
+    body { background-color: #fff; color: #666; text-align: center; font-family: arial, sans-serif; }
+    div.dialog {
+      width: 25em;
+      padding: 0 4em;
+      margin: 4em auto 0 auto;
+      border: 1px solid #ccc;
+      border-right-color: #999;
+      border-bottom-color: #999;
+    }
+    h1 { font-size: 100%; color: #f00; line-height: 1.5em; }
+  </style>
+</head>
+<body>
+  <!-- This file lives in public/500.html -->
+  <div class="dialog">
+    <h1>We're sorry, but something went wrong.</h1>
+    <p>We've been notified about this issue and we'll take a look at it shortly.</p>
+  </div>
+</body>
+</html>

data/spec/dummy/public/favicon.ico ADDED

File without changes

data/spec/dummy/public/stylesheets/.gitkeep ADDED

File without changes

data/spec/dummy/script/rails ADDED

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+# This command will automatically be run when you run "rails" with Rails 3 gems installed from the root of your application.
+APP_PATH = File.expand_path('../../config/application',  __FILE__)
+require File.expand_path('../../config/boot',  __FILE__)
+require 'rails/commands'

data/spec/integration/apartment_rake_integration_spec.rb ADDED

@@ -0,0 +1,74 @@
+require 'spec_helper'
+require 'rake'
+describe "apartment rake tasks" do
+  before do
+    @rake = Rake::Application.new
+    Rake.application = @rake
+    Dummy::Application.load_tasks
+    # somehow this misc.rake file gets lost in the shuffle
+    # it defines a `rails_env` task that our db:migrate depends on
+    # No idea why, but during the tests, we somehow lose this tasks, so we get an error when testing migrations
+    # This is STUPID!
+    load "rails/tasks/misc.rake"
+  end
+  after do
+    Rake.application = nil
+  end
+  before do
+    Apartment.configure do |config|
+      config.excluded_models = ["Company"]
+      config.database_names = lambda{ Company.scoped.collect(&:database) }
+    end
+  end
+  context "with x number of databases" do
+    let(:x){ 1 + rand(5) }    # random number of dbs to create
+    let(:db_names){ x.times.map{|y| "database_#{y}" } }
+    before do
+      db_names.collect do |db_name|
+        Apartment::Database.create(db_name)
+        Company.create :database => db_name
+      end
+    end
+    after do
+      db_names.each{ |db| Apartment::Database.drop(db) }
+      Company.delete_all
+    end
+    describe "#migrate" do
+      it "should migrate all databases" do
+        Apartment::Migrator.should_receive(:migrate).exactly(db_names.length).times
+        @rake['apartment:migrate'].invoke
+      end
+    end
+    describe "#rollback" do
+      it "should rollback all dbs" do
+        db_names.each do |name|
+          Apartment::Migrator.should_receive(:rollback).with(name, anything)
+        end
+        @rake['apartment:rollback'].invoke
+        @rake['apartment:migrate'].invoke   # migrate again so that our next test 'seed' can run (requires migrations to be complete)
+      end
+    end
+    describe "apartment:seed" do
+      it "should seed all databases" do
+        Apartment::Database.should_receive(:seed).exactly(db_names.length).times
+        @rake['apartment:seed'].invoke
+      end
+    end
+  end
+end

data/spec/integration/database_integration_spec.rb ADDED

@@ -0,0 +1,200 @@
+require 'spec_helper'
+describe Apartment::Database do
+  context "using postgresql" do
+    # See apartment.yml file in dummy app config
+    let(:config){ Apartment::Test.config['connections']['postgresql'].symbolize_keys }
+    let(:database){ "some_new_database" }
+    let(:database2){ "yet_another_database" }
+    before do
+      Apartment.use_postgres_schemas = true
+      ActiveRecord::Base.establish_connection config
+      Apartment::Test.load_schema   # load the Rails schema in the public db schema
+      Apartment::Database.stub(:config).and_return config   # Use postgresql database config for this test
+      @schema_search_path = ActiveRecord::Base.connection.schema_search_path
+    end
+    after do
+      Apartment::Test.reset
+    end
+    describe "#init" do
+      it "should process model exclusions" do
+        Apartment.configure do |config|
+          config.excluded_models = ["Company"]
+        end
+        Apartment::Database.init
+        Company.table_name.should == "public.companies"
+      end
+    end
+    describe "#adapter" do
+      before do
+        Apartment::Database.reload!
+      end
+      it "should load postgresql adapter" do
+        Apartment::Database.adapter
+        Apartment::Adapters::PostgresqlAdapter.should be_a(Class)
+      end
+      it "should raise exception with invalid adapter specified" do
+        Apartment::Database.stub(:config).and_return config.merge(:adapter => 'unkown')
+        expect {
+          Apartment::Database.adapter
+        }.to raise_error
+      end
+    end
+    context "with schemas" do
+      before do
+        Apartment.configure do |config|
+          config.excluded_models = []
+          config.use_postgres_schemas = true
+          config.seed_after_create = true
+        end
+        Apartment::Database.create database
+      end
+      after do
+        Apartment::Test.drop_schema database
+      end
+      describe "#process" do
+        before do
+          Apartment::Database.create database2
+        end
+        after do
+          Apartment::Test.drop_schema database2
+        end
+        it "should connect to new schema" do
+          Apartment::Database.process(database) do
+            Apartment::Database.current_database.should == database
+          end
+        end
+        it "should reset connection to the previous db" do
+          Apartment::Database.switch(database2)
+          Apartment::Database.process(database)
+          Apartment::Database.current_database.should == database2
+        end
+        it "should reset to previous schema if database is nil" do
+          Apartment::Database.switch(database)
+          Apartment::Database.process
+          Apartment::Database.current_database.should == database
+        end
+        it "should set to public schema if database is nil" do
+          Apartment::Database.process do
+            Apartment::Database.current_database.should == @schema_search_path
+          end
+        end
+      end
+      describe "#create" do
+        it "should create new postgres schema" do
+          ActiveRecord::Base.connection.execute("SELECT nspname FROM pg_namespace;").collect{|row| row['nspname']}.should include(database)
+        end
+        it "should seed data" do
+          Apartment::Database.switch database
+          User.count.should be > 0
+        end
+      end
+      describe "#switch" do
+        let(:x){ rand(3) }
+        it "should connect to new schema" do
+          Apartment::Database.switch database
+          ActiveRecord::Base.connection.schema_search_path.should == database
+        end
+        it "should fail with invalid schema" do
+          expect {
+            Apartment::Database.switch('some_nonexistent_schema')
+          }.to raise_error Apartment::SchemaNotFound
+        end
+        context "creating models" do
+          before do
+            Apartment::Database.create database2
+          end
+          after do
+            Apartment::Test.drop_schema database2
+          end
+          it "should create a model instance in the current schema" do
+            Apartment::Database.switch database2
+            db2_count = User.count + x.times{ User.create }
+            Apartment::Database.switch database
+            db_count = User.count + x.times{ User.create }
+            Apartment::Database.switch database2
+            User.count.should == db2_count
+            Apartment::Database.switch database
+            User.count.should == db_count
+          end
+        end
+        context "with excluded models" do
+          before do
+            Apartment.configure do |config|
+              config.excluded_models = ["Company"]
+            end
+            Apartment::Database.init
+          end
+          it "should ignore excluded models" do
+            Apartment::Database.switch database
+            Company.table_name.should include('public')
+          end
+          it "should create excluded models in public schema" do
+            Apartment::Database.reset # ensure we're on public schema
+            count = Company.count + x.times{ Company.create }
+            Apartment::Database.switch database
+            x.times{ Company.create }
+            Company.count.should == count + x
+            Apartment::Database.reset
+            Company.count.should == count + x
+          end
+        end
+      end
+      describe "#current_database" do
+        it "should return the current schema search path" do
+          Apartment::Database.switch database
+          Apartment::Database.current_database.should == database
+        end
+      end
+    end
+  end
+end

data/spec/integration/delayed_job_integration_spec.rb ADDED

@@ -0,0 +1,100 @@
+require 'spec_helper'
+require 'delayed_job'
+Delayed::Worker.guess_backend
+describe Apartment::Delayed do
+  # See apartment.yml file in dummy app config
+  let(:config){ Apartment::Test.config['connections']['postgresql'].symbolize_keys }
+  let(:database){ "some_new_database" }
+  let(:database2){ "another_db" }
+  before do
+    ActiveRecord::Base.establish_connection config
+    Apartment::Test.load_schema   # load the Rails schema in the public db schema
+    Apartment::Database.stub(:config).and_return config   # Use postgresql database config for this test
+    @schema_search_path = ActiveRecord::Base.connection.schema_search_path
+    Apartment.configure do |config|
+      config.use_postgres_schemas = true
+    end
+    Apartment::Database.create database
+    Apartment::Database.create database2
+  end
+  after do
+    Apartment::Test.drop_schema database
+    Apartment::Test.drop_schema database2
+    Apartment::Test.reset
+  end
+  describe Apartment::Delayed::Job do
+    context "#enqueue" do
+      before do
+        Apartment::Database.reset
+      end
+      it "should queue up jobs in the public schema" do
+        dj_count = Delayed::Job.count
+        Apartment::Database.switch database
+        Apartment::Delayed::Job.enqueue FakeDjClass.new
+        Apartment::Database.reset
+        Delayed::Job.count.should == dj_count + 1
+      end
+      it "should not queue jobs in the current schema" do
+        Apartment::Database.switch database
+        expect {
+          Apartment::Delayed::Job.enqueue FakeDjClass.new
+        }.to_not change(Delayed::Job, :count)        # because we will still be on the `database` schema, not public
+      end
+    end
+  end
+  describe Apartment::Delayed::Requirements do
+    before do
+      Apartment::Database.switch database
+      User.send(:include, Apartment::Delayed::Requirements)
+      User.create
+    end
+    it "should initialize a database attribute on a class" do
+      user = User.first
+      user.database.should == database
+    end
+    it "should not overwrite any previous after_initialize declarations" do
+      User.class_eval do
+        after_find :set_name
+        def set_name
+          self.name = "Some Name"
+        end
+      end
+      user = User.first
+      user.database.should == database
+      user.name.should == "Some Name"
+    end
+    it "should set the db on a new record before it saves" do
+      user = User.create
+      user.database.should == database
+    end
+    context "serialization" do
+      it "should serialize the proper database attribute" do
+        user_yaml = User.first.to_yaml
+        Apartment::Database.switch database2
+        user = YAML.load user_yaml
+        user.database.should == database
+      end
+    end
+  end
+end