RubyGems - bullet - Versions diffs - 2.2.1 → 2.3.0 - Mend

bullet 2.2.1 → 2.3.0

Files changed (80) hide show

data/.gitignore +1 -0
data/.travis.yml +4 -1
data/Gemfile +3 -2
data/Gemfile.lock +85 -69
data/Guardfile +8 -0
data/lib/bullet.rb +13 -8
data/lib/bullet/action_controller2.rb +4 -4
data/lib/bullet/active_record2.rb +5 -6
data/lib/bullet/active_record3.rb +2 -3
data/lib/bullet/active_record31.rb +6 -8
data/lib/bullet/detector/association.rb +27 -53
data/lib/bullet/detector/counter.rb +34 -29
data/lib/bullet/detector/n_plus_one_query.rb +47 -28
data/lib/bullet/detector/unused_eager_association.rb +31 -30
data/lib/bullet/notification/base.rb +14 -12
data/lib/bullet/notification/n_plus_one_query.rb +6 -10
data/lib/bullet/notification/unused_eager_loading.rb +1 -2
data/lib/bullet/notification_collector.rb +1 -2
data/lib/bullet/rack.rb +6 -3
data/lib/bullet/registry/association.rb +4 -6
data/lib/bullet/registry/base.rb +10 -7
data/lib/bullet/registry/object.rb +6 -6
data/lib/bullet/version.rb +1 -1
data/perf/benchmark.rb +30 -12
data/spec/bullet/detector/association_spec.rb +90 -0
data/spec/bullet/detector/base_spec.rb +14 -0
data/spec/bullet/detector/counter_spec.rb +65 -0
data/spec/bullet/detector/n_plus_one_query_spec.rb +94 -0
data/spec/bullet/detector/unused_eager_association_spec.rb +62 -0
data/spec/bullet/notification/base_spec.rb +67 -0
data/spec/bullet/notification/counter_cache_spec.rb +12 -0
data/spec/bullet/notification/n_plus_one_query_spec.rb +13 -0
data/spec/bullet/notification/unused_eager_loading_spec.rb +12 -0
data/spec/bullet/notification_collector_spec.rb +32 -0
data/spec/bullet/rack_spec.rb +80 -24
data/spec/bullet/registry/association_spec.rb +26 -0
data/spec/bullet/registry/base_spec.rb +44 -0
data/spec/bullet/registry/object_spec.rb +25 -0
data/spec/integration/association_for_chris_spec.rb +37 -0
data/spec/integration/association_for_peschkaj_spec.rb +26 -0
data/spec/{bullet → integration}/association_spec.rb +1 -359
data/spec/integration/counter_spec.rb +37 -0
data/spec/models/address.rb +3 -0
data/spec/models/author.rb +3 -0
data/spec/models/base_user.rb +5 -0
data/spec/models/category.rb +7 -0
data/spec/models/city.rb +3 -0
data/spec/models/client.rb +4 -0
data/spec/models/comment.rb +4 -0
data/spec/models/company.rb +3 -0
data/spec/models/contact.rb +3 -0
data/spec/models/country.rb +3 -0
data/spec/models/deal.rb +4 -0
data/spec/models/document.rb +5 -0
data/spec/models/email.rb +3 -0
data/spec/models/entry.rb +3 -0
data/spec/models/firm.rb +4 -0
data/spec/models/folder.rb +2 -0
data/spec/models/hotel.rb +4 -0
data/spec/models/location.rb +3 -0
data/spec/models/newspaper.rb +3 -0
data/spec/models/page.rb +2 -0
data/spec/models/person.rb +3 -0
data/spec/models/pet.rb +3 -0
data/spec/models/post.rb +11 -0
data/spec/models/relationship.rb +4 -0
data/spec/models/student.rb +3 -0
data/spec/models/submission.rb +4 -0
data/spec/models/teacher.rb +3 -0
data/spec/models/user.rb +4 -0
data/spec/models/writer.rb +2 -0
data/spec/spec_helper.rb +16 -106
data/spec/support/bullet_ext.rb +55 -0
data/spec/support/rack_double.rb +55 -0
data/spec/support/seed.rb +256 -0
metadata +104 -14
data/.watchr +0 -24
data/spec/bullet/association_for_chris_spec.rb +0 -96
data/spec/bullet/association_for_peschkaj_spec.rb +0 -86
data/spec/bullet/counter_spec.rb +0 -136

data/spec/bullet/notification/counter_cache_spec.rb ADDED

@@ -0,0 +1,12 @@
+require 'spec_helper'
+module Bullet
+  module Notification
+    describe CounterCache do
+      subject { CounterCache.new(Post, [:comments, :votes]) }
+      its(:body) { should == "  Post => [:comments, :votes]" }
+      its(:title) { should == "Need Counter Cache" }
+    end
+  end
+end

data/spec/bullet/notification/n_plus_one_query_spec.rb ADDED

@@ -0,0 +1,13 @@
+require 'spec_helper'
+module Bullet
+  module Notification
+    describe NPlusOneQuery do
+      subject { NPlusOneQuery.new([["caller1", "caller2"]], Post, [:comments, :votes], "path") }
+      its(:body_with_caller) { should == "  Post => [:comments, :votes]\n  Add to your finder: :include => [:comments, :votes]\nN+1 Query method call stack\n  caller1\n  caller2" }
+      its(:body) { should == "  Post => [:comments, :votes]\n  Add to your finder: :include => [:comments, :votes]" }
+      its(:title) { should == "N+1 Query in path" }
+    end
+  end
+end

data/spec/bullet/notification/unused_eager_loading_spec.rb ADDED

@@ -0,0 +1,12 @@
+require 'spec_helper'
+module Bullet
+  module Notification
+    describe UnusedEagerLoading do
+      subject { UnusedEagerLoading.new(Post, [:comments, :votes], "path") }
+      its(:body) { should == "  Post => [:comments, :votes]\n  Remove from your finder: :include => [:comments, :votes]" }
+      its(:title) { should == "Unused Eager Loading in path" }
+    end
+  end
+end

data/spec/bullet/notification_collector_spec.rb ADDED

@@ -0,0 +1,32 @@
+require 'spec_helper'
+module Bullet
+  describe NotificationCollector do
+    subject { NotificationCollector.new.tap { |collector| collector.add("value") } }
+    context "#add" do
+      it "should add a value" do
+        subject.add("value1")
+        subject.collection.should be_include("value1")
+      end
+    end
+    context "#reset" do
+      it "should reset collector" do
+        subject.reset
+        subject.collection.should be_empty
+      end
+    end
+    context "#notifications_present?" do
+      it "should be true if collection is not empty" do
+        subject.should be_notifications_present
+      end
+      it "should be false if collection is empty" do
+        subject.reset
+        subject.should_not be_notifications_present
+      end
+    end
+  end
+end

data/spec/bullet/rack_spec.rb CHANGED

@@ -1,39 +1,95 @@
-require File.dirname(__FILE__) + '/../spec_helper'
+require 'spec_helper'
+module Bullet
+  describe Rack do
+    let(:middleware) { Bullet::Rack.new app }
+    let(:app) { Support::AppDouble.new }
-describe Bullet::Rack do
-  let(:middleware) { Bullet::Rack.new app }
-  let(:app) { AppDouble.new }
+    context "#no_browser_cache" do
+      it "should add no cache meta in http headers" do
+        headers = {}
+        middleware.no_browser_cache(headers)
+        headers["Cache-Control"].should == "no-cache, no-store, max-age=0, must-revalidate"
+        headers["Pragma"].should == "no-cache"
+        headers["Expires"].should == "Wed, 09 Sep 2009 09:09:09 GMT"
+      end
+    end
+    context "#html_request?" do
+      it "should be true if Content-Type is text/html and http body contains html tag" do
+        headers = {"Content-Type" => "text/html"}
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should be_html_request(headers, response)
+      end
+      it "should be false if there is no Content-Type header" do
+        headers = {}
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should_not be_html_request(headers, response)
+      end
+      it "should be false if Content-Type is javascript" do
+        headers = {"Content-Type" => "text/javascript"}
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should_not be_html_request(headers, response)
+      end
-  describe "#call" do
-    context "when Bullet is enabled" do
-      before(:each) { Bullet.enable = true }
+      it "should be false if response body doesn't contain html tag" do
+        headers = {"Content-Type" => "text/html"}
+        response = stub(:body => "<div>Partial</div>")
+        middleware.should_not be_html_request(headers, response)
+      end
+    end
-      it "should invoke Bullet.start_request" do
-        Bullet.should_receive(:start_request)
-        middleware.call([])
+    context "empty?" do
+      it "should be false if response is a string and not empty" do
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should_not be_empty(response)
       end
-      it "should invoke Bullet.end_request" do
-        Bullet.should_receive(:end_request)
-        middleware.call([])
+      it "should be tru if response is not found" do
+        response = ["Not Found"]
+        middleware.should be_empty(response)
       end
-      it "should return original response body" do
-        expected_response = ResponseDouble.new "Actual body"
-        app.response = expected_response
-        status, headers, response = middleware.call([])
-        response.should eq expected_response
+      it "should be true if response body is empty" do
+        response = stub(:body => "")
+        middleware.should be_empty(response)
       end
     end
-    context "when Bullet is disabled" do
-      before(:each) { Bullet.enable = false }
-      after(:each) { Bullet.enable = true }
+    context "#call" do
+      context "when Bullet is enabled" do
+        it "should invoke Bullet.start_request and Bullet.end_request" do
+          Bullet.should_receive(:start_request)
+          Bullet.should_receive(:end_request)
+          middleware.call([])
+        end
+        it "should return original response body" do
+          expected_response = Support::ResponseDouble.new "Actual body"
+          app.response = expected_response
+          status, headers, response = middleware.call([])
+          response.should == expected_response
+        end
+        it "should change response body if notification is active" do
+          Bullet.should_receive(:notification?).and_return(true)
+          Bullet.should_receive(:gather_inline_notifications).and_return("<bullet></bullet>")
+          Bullet.should_receive(:perform_out_of_channel_notifications)
+          status, headers, response = middleware.call([200, {"Content-Type" => "text/html"}])
+          headers["Content-Length"].should == "56"
+          response.should == ["<html><head></head><body></body></html><bullet></bullet>"]
+        end
+      end
+      context "when Bullet is disabled" do
+        before(:each) { Bullet.stub(:enable?, false) }
-      it "should not call Bullet.start_request" do
-        Bullet.should_not_receive(:start_request)
-        middleware.call([])
+        it "should not call Bullet.start_request" do
+          Bullet.should_not_receive(:start_request)
+          middleware.call([])
+        end
       end
     end
   end

data/spec/bullet/registry/association_spec.rb ADDED

@@ -0,0 +1,26 @@
+require 'spec_helper'
+module Bullet
+  module Registry
+    describe Association do
+      subject { Association.new.tap { |association| association.add(["key1", "key2"], "value") } }
+      context "#merge" do
+        it "should merge key/value" do
+          subject.merge("key0", "value0")
+          subject["key0"].should be_include("value0")
+        end
+      end
+      context "#similarly_associated" do
+        it "should return similarly associated keys" do
+          subject.similarly_associated("key1", Set.new(["value"])).should == ["key1", "key2"]
+        end
+        it "should return empty if key does not exist" do
+          subject.similarly_associated("key3", Set.new(["value"])).should be_empty
+        end
+      end
+    end
+  end
+end

data/spec/bullet/registry/base_spec.rb ADDED

@@ -0,0 +1,44 @@
+require 'spec_helper'
+module Bullet
+  module Registry
+    describe Base do
+      subject { Base.new.tap { |base| base.add("key", "value") } }
+      context "#[]" do
+        it "should get value by key" do
+          subject["key"].should == Set.new(["value"])
+        end
+      end
+      context "#delete" do
+        it "should delete key" do
+          subject.delete("key")
+          subject["key"].should be_nil
+        end
+      end
+      context "#add" do
+        it "should add value with string" do
+          subject.add("key", "new_value")
+          subject["key"].should == Set.new(["value", "new_value"])
+        end
+        it "should add value with array" do
+          subject.add("key", ["value1", "value2"])
+          subject["key"].should == Set.new(["value", "value1", "value2"])
+        end
+      end
+      context "#include?" do
+        it "should include key/value" do
+          subject.include?("key", "value").should be_true
+        end
+        it "should not include wrong key/value" do
+          subject.include?("key", "val").should be_false
+        end
+      end
+    end
+  end
+end

data/spec/bullet/registry/object_spec.rb ADDED

@@ -0,0 +1,25 @@
+require 'spec_helper'
+module Bullet
+  module Registry
+    describe Object do
+      let(:post) { Post.first }
+      let(:another_post) { Post.last }
+      subject { Object.new.tap { |object| object.add(post.ar_key) } }
+      context "#include?" do
+        it "should include the object" do
+          subject.should be_include(post.ar_key)
+        end
+      end
+      context "#add" do
+        it "should add an object" do
+          subject.add(another_post.ar_key)
+          subject.should be_include(another_post.ar_key)
+        end
+      end
+    end
+  end
+end

data/spec/integration/association_for_chris_spec.rb ADDED

@@ -0,0 +1,37 @@
+require 'spec_helper'
+# This test is just used for http://github.com/flyerhzm/bullet/issues/#issue/14
+describe Bullet::Detector::Association do
+  before(:each) do
+    Bullet.clear
+    Bullet.start_request
+  end
+  after(:each) do
+    Bullet.end_request
+  end
+  describe "for chris" do
+    it "should detect unpreload association from deal to hotel" do
+      Deal.all.each do |deal|
+        deal.hotel.location.name
+      end
+      Bullet::Detector::Association.should be_detecting_unpreloaded_association_for(Deal, :hotel)
+    end
+    it "should detect unpreload association from hotel to location" do
+      Deal.includes(:hotel).each do |deal|
+        deal.hotel.location.name
+      end
+      Bullet::Detector::Association.should be_detecting_unpreloaded_association_for(Hotel, :location)
+    end
+    it "should not detect unpreload association" do
+      Deal.includes({:hotel => :location}).each do |deal|
+        deal.hotel.location.name
+      end
+      Bullet::Detector::UnusedEagerAssociation.check_unused_preload_associations
+      Bullet::Detector::Association.should_not be_has_unused_preload_associations
+    end
+  end
+end

data/spec/integration/association_for_peschkaj_spec.rb ADDED

@@ -0,0 +1,26 @@
+require 'spec_helper'
+# This test is just used for http://github.com/flyerhzm/bullet/issues#issue/20
+describe Bullet::Detector::Association do
+  before(:each) do
+    Bullet.clear
+    Bullet.start_request
+  end
+  after(:each) do
+    Bullet.end_request
+  end
+  describe "for peschkaj" do
+    it "should not detect unused preload associations" do
+      category = Category.includes({:submissions => :user}).order("id DESC").find_by_name('first')
+      category.submissions.map do |submission|
+        submission.name
+        submission.user.name
+      end
+      Bullet::Detector::UnusedEagerAssociation.check_unused_preload_associations
+      Bullet::Detector::Association.should_not be_unused_preload_associations_for(Category, :submissions)
+      Bullet::Detector::Association.should_not be_unused_preload_associations_for(Submission, :user)
+    end
+  end
+end

data/spec/{bullet → integration}/association_spec.rb RENAMED

@@ -1,125 +1,6 @@
-require File.dirname(__FILE__) + '/../spec_helper'
-ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
+require 'spec_helper'
 describe Bullet::Detector::Association, 'has_many' do
-  def setup_db
-    ActiveRecord::Schema.define(:version => 1) do
-      create_table :categories do |t|
-        t.column :name, :string
-      end
-      create_table :posts do |t|
-        t.column :name, :string
-        t.column :category_id, :integer
-        t.column :writer_id, :integer
-      end
-      create_table :comments do |t|
-        t.column :name, :string
-        t.column :post_id, :integer
-        t.column :author_id, :integer
-      end
-      create_table :entries do |t|
-        t.column :name, :string
-        t.column :category_id, :integer
-      end
-      create_table :base_users do |t|
-        t.column :name, :string
-        t.column :type, :string
-        t.column :newspaper_id, :integer
-      end
-      create_table :newspapers do |t|
-        t.column :name, :string
-      end
-    end
-  end
-  def teardown_db
-    ActiveRecord::Base.connection.tables.each do |table|
-      ActiveRecord::Base.connection.drop_table(table)
-    end
-  end
-  class Category < ActiveRecord::Base
-    has_many :posts
-    has_many :entries
-  end
-  class Post < ActiveRecord::Base
-    belongs_to :category
-    has_many :comments
-    belongs_to :writer
-    scope :preload_posts, lambda { includes(:comments) }
-    scope :in_category_name, lambda { |name|
-      where(['categories.name = ?', name]).includes(:category)
-    }
-  end
-  class Entry < ActiveRecord::Base
-    belongs_to :category
-  end
-  class Comment < ActiveRecord::Base
-    belongs_to :post
-    belongs_to :author, :class_name => "BaseUser"
-  end
-  class BaseUser < ActiveRecord::Base
-    has_many :comments
-    has_many :posts
-    belongs_to :newspaper
-  end
-  class Newspaper < ActiveRecord::Base
-    has_many :writers, :class_name => "BaseUser"
-  end
-  class Writer < BaseUser
-  end
-  before(:all) do
-    setup_db
-    newspaper1 = Newspaper.create(:name => "First Newspaper")
-    newspaper2 = Newspaper.create(:name => "Second Newspaper")
-    writer1 = Writer.create(:name => 'first', :newspaper => newspaper1)
-    writer2 = Writer.create(:name => 'second', :newspaper => newspaper2)
-    user1 = BaseUser.create(:name => 'third', :newspaper => newspaper1)
-    user2 = BaseUser.create(:name => 'fourth', :newspaper => newspaper2)
-    category1 = Category.create(:name => 'first')
-    category2 = Category.create(:name => 'second')
-    post1 = category1.posts.create(:name => 'first', :writer => writer1)
-    post1a = category1.posts.create(:name => 'like first', :writer => writer2)
-    post2 = category2.posts.create(:name => 'second', :writer => writer2)
-    comment1 = post1.comments.create(:name => 'first', :author => writer1)
-    comment2 = post1.comments.create(:name => 'first2', :author => writer1)
-    comment3 = post1.comments.create(:name => 'first3', :author => writer1)
-    comment4 = post1.comments.create(:name => 'second', :author => writer2)
-    comment8 = post1a.comments.create(:name => "like first 1", :author => writer1)
-    comment9 = post1a.comments.create(:name => "like first 2", :author => writer2)
-    comment5 = post2.comments.create(:name => 'third', :author => user1)
-    comment6 = post2.comments.create(:name => 'fourth', :author => user2)
-    comment7 = post2.comments.create(:name => 'fourth', :author => writer1)
-    entry1 = category1.entries.create(:name => 'first')
-    entry2 = category1.entries.create(:name => 'second')
-  end
-  after(:all) do
-    teardown_db
-  end
   before(:each) do
     Bullet.start_request
   end
@@ -482,54 +363,6 @@ describe Bullet::Detector::Association, 'has_many' do
 end
 describe Bullet::Detector::Association, 'has_and_belongs_to_many' do
-  def setup_db
-    ActiveRecord::Schema.define(:version => 1) do
-      create_table :students do |t|
-        t.column :name, :string
-      end
-      create_table :teachers do |t|
-        t.column :name, :string
-      end
-      create_table :students_teachers, :id => false do |t|
-        t.column :student_id, :integer
-        t.column :teacher_id, :integer
-      end
-    end
-  end
-  def teardown_db
-    ActiveRecord::Base.connection.tables.each do |table|
-      ActiveRecord::Base.connection.drop_table(table)
-    end
-  end
-  class Student < ActiveRecord::Base
-    has_and_belongs_to_many :teachers
-  end
-  class Teacher < ActiveRecord::Base
-    has_and_belongs_to_many :students
-  end
-  before(:all) do
-    setup_db
-    student1 = Student.create(:name => 'first')
-    student2 = Student.create(:name => 'second')
-    teacher1 = Teacher.create(:name => 'first')
-    teacher2 = Teacher.create(:name => 'second')
-    student1.teachers = [teacher1, teacher2]
-    student2.teachers = [teacher1, teacher2]
-    teacher1.students << student1
-    teacher2.students << student2
-  end
-  after(:all) do
-    teardown_db
-  end
   before(:each) do
     Bullet.start_request
   end
@@ -566,61 +399,6 @@ describe Bullet::Detector::Association, 'has_and_belongs_to_many' do
 end
 describe Bullet::Detector::Association, 'has_many :through' do
-  def setup_db
-    ActiveRecord::Schema.define(:version => 1) do
-      create_table :firms do |t|
-        t.column :name, :string
-      end
-      create_table :clients do |t|
-        t.column :name, :string
-      end
-      create_table :relationships do |t|
-        t.column :firm_id, :integer
-        t.column :client_id, :integer
-      end
-    end
-  end
-  def teardown_db
-    ActiveRecord::Base.connection.tables.each do |table|
-      ActiveRecord::Base.connection.drop_table(table)
-    end
-  end
-  class Firm < ActiveRecord::Base
-    has_many :relationships
-    has_many :clients, :through => :relationships
-  end
-  class Client < ActiveRecord::Base
-    has_many :relationships
-    has_many :firms, :through => :relationships
-  end
-  class Relationship < ActiveRecord::Base
-    belongs_to :firm
-    belongs_to :client
-  end
-  before(:all) do
-    setup_db
-    firm1 = Firm.create(:name => 'first')
-    firm2 = Firm.create(:name => 'second')
-    client1 = Client.create(:name => 'first')
-    client2 = Client.create(:name => 'second')
-    firm1.clients = [client1, client2]
-    firm2.clients = [client1, client2]
-    client1.firms << firm1
-    client2.firms << firm2
-  end
-  after(:all) do
-    teardown_db
-  end
   before(:each) do
     Bullet.start_request
   end
@@ -659,48 +437,6 @@ end
 describe Bullet::Detector::Association, "has_one" do
-  def setup_db
-    ActiveRecord::Schema.define(:version => 1) do
-      create_table :companies do |t|
-        t.column :name, :string
-      end
-      create_table :addresses do |t|
-        t.column :name, :string
-        t.column :company_id, :integer
-      end
-    end
-  end
-  def teardown_db
-    ActiveRecord::Base.connection.tables.each do |table|
-      ActiveRecord::Base.connection.drop_table(table)
-    end
-  end
-  class Company < ActiveRecord::Base
-    has_one :address
-  end
-  class Address < ActiveRecord::Base
-    belongs_to :company
-  end
-  before(:all) do
-    setup_db
-    company1 = Company.create(:name => 'first')
-    company2 = Company.create(:name => 'second')
-    Address.create(:name => 'first', :company => company1)
-    Address.create(:name => 'second', :company => company2)
-  end
-  after(:all) do
-    teardown_db
-  end
   before(:each) do
     Bullet.start_request
   end
@@ -737,50 +473,6 @@ describe Bullet::Detector::Association, "has_one" do
 end
 describe Bullet::Detector::Association, "call one association that in possible objects" do
-  def setup_db
-    ActiveRecord::Schema.define(:version => 1) do
-      create_table :contacts do |t|
-        t.column :name, :string
-      end
-      create_table :emails do |t|
-        t.column :name, :string
-        t.column :contact_id, :integer
-      end
-    end
-  end
-  def teardown_db
-    ActiveRecord::Base.connection.tables.each do |table|
-      ActiveRecord::Base.connection.drop_table(table)
-    end
-  end
-  class Contact < ActiveRecord::Base
-    has_many :emails
-  end
-  class Email < ActiveRecord::Base
-    belongs_to :contact
-  end
-  before(:all) do
-    setup_db
-    contact1 = Contact.create(:name => 'first')
-    contact2 = Contact.create(:name => 'second')
-    email1 = contact1.emails.create(:name => 'first')
-    email2 = contact1.emails.create(:name => 'second')
-    email3 = contact2.emails.create(:name => 'third')
-    email4 = contact2.emails.create(:name => 'fourth')
-  end
-  after(:all) do
-    teardown_db
-  end
   before(:each) do
     Bullet.start_request
   end
@@ -797,56 +489,6 @@ describe Bullet::Detector::Association, "call one association that in possible o
 end
 describe Bullet::Detector::Association, "STI" do
-  def setup_db
-    ActiveRecord::Schema.define(:version => 1) do
-      create_table :documents do |t|
-        t.string :name
-        t.string :type
-        t.integer :parent_id
-        t.integer :author_id
-      end
-      create_table :authors do |t|
-        t.string :name
-      end
-    end
-  end
-  def teardown_db
-    ActiveRecord::Base.connection.tables.each do |table|
-      ActiveRecord::Base.connection.drop_table(table)
-    end
-  end
-  class Document < ActiveRecord::Base
-    has_many :children, :class_name => "Document", :foreign_key => 'parent_id'
-    belongs_to :parent, :class_name => "Document", :foreign_key => 'parent_id'
-    belongs_to :author
-  end
-  class Page < Document
-  end
-  class Folder < Document
-  end
-  class Author < ActiveRecord::Base
-    has_many :documents
-  end
-  before(:all) do
-    setup_db
-    author1 = Author.create(:name => 'author1')
-    author2 = Author.create(:name => 'author2')
-    folder1 = Folder.create(:name => 'folder1', :author_id => author1.id)
-    folder2 = Folder.create(:name => 'folder2', :author_id => author2.id)
-    page1 = Page.create(:name => 'page1', :parent_id => folder1.id, :author_id => author1.id)
-    page2 = Page.create(:name => 'page2', :parent_id => folder1.id, :author_id => author1.id)
-    page3 = Page.create(:name => 'page3', :parent_id => folder2.id, :author_id => author2.id)
-    page4 = Page.create(:name => 'page4', :parent_id => folder2.id, :author_id => author2.id)
-  end
   before(:each) do
     Bullet.start_request
   end