bullet 1.6.0

data/spec/bullet_counter_spec.rb

@@ -0,0 +1,136 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
+
+describe Bullet::Counter do
+  def setup_db
+    ActiveRecord::Schema.define(:version => 1) do
+      create_table :countries do |t|
+        t.string :name
+      end
+
+      create_table :cities do |t|
+        t.string :name
+        t.integer :country_id
+      end
+    end
+  end
+
+  def teardown_db
+    ActiveRecord::Base.connection.tables.each do |table|
+      ActiveRecord::Base.connection.drop_table(table)
+    end
+  end
+
+  class Country < ActiveRecord::Base
+    has_many :cities
+  end
+
+  class City < ActiveRecord::Base
+    belongs_to :country
+  end
+
+  before(:all) do
+    setup_db
+
+    country1 = Country.create(:name => 'first')
+    country2 = Country.create(:name => 'second')
+    
+    country1.cities.create(:name => 'first')
+    country1.cities.create(:name => 'second')
+    country2.cities.create(:name => 'third')
+    country2.cities.create(:name => 'fourth')
+  end
+
+  after(:all) do
+    teardown_db
+  end
+
+  before(:each) do
+    Bullet::Counter.start_request
+  end
+  
+  after(:each) do
+    Bullet::Counter.end_request
+  end
+
+  it "should need counter cache with all cities" do
+    Country.all.each do |country|
+      country.cities.size
+    end
+    Bullet::Counter.should be_need_counter_caches
+  end
+  
+  it "should not need coounter cache with only one object" do
+    Country.first.cities.size
+    Bullet::Counter.should_not be_need_counter_caches
+  end
+
+  it "should not need counter cache with part of cities" do
+    Country.all.each do |country|
+      country.cities(:conditions => ["name = ?", 'first']).size
+    end
+    Bullet::Counter.should_not be_need_counter_caches
+  end
+end
+
+describe Bullet::Counter do
+  def setup_db
+    ActiveRecord::Schema.define(:version => 1) do
+      create_table :people do |t|
+        t.string :name
+        t.integer :pets_count
+      end
+
+      create_table :pets do |t|
+        t.string :name
+        t.integer :person_id
+      end
+    end
+  end
+
+  def teardown_db
+    ActiveRecord::Base.connection.tables.each do |table|
+      ActiveRecord::Base.connection.drop_table(table)
+    end
+  end
+
+  class Person < ActiveRecord::Base
+    has_many :pets
+  end
+
+  class Pet < ActiveRecord::Base
+    belongs_to :person, :counter_cache => true
+  end
+
+  before(:all) do
+    setup_db
+
+    person1 = Person.create(:name => 'first')
+    person2 = Person.create(:name => 'second')
+    
+    person1.pets.create(:name => 'first')
+    person1.pets.create(:name => 'second')
+    person2.pets.create(:name => 'third')
+    person2.pets.create(:name => 'fourth')
+  end
+
+  after(:all) do
+    teardown_db
+  end
+
+  before(:each) do
+    Bullet::Counter.start_request
+  end
+  
+  after(:each) do
+    Bullet::Counter.end_request
+  end
+
+  it "should not need counter cache" do
+    Person.all.each do |person|
+      person.pets.size
+    end
+    Bullet::Counter.should_not be_need_counter_caches
+  end
+end

data/spec/spec.opts

@@ -0,0 +1,8 @@
+--colour
+--format
+specdoc
+--reverse
+--timeout
+20
+--loadby
+mtime

data/spec/spec_helper.rb

@@ -0,0 +1,50 @@
+require 'rubygems'
+require 'spec/autorun'
+require 'active_record'
+require 'action_controller'
+
+RAILS_ROOT = File.expand_path(__FILE__).split('/')[0..-3].join('/') unless defined? RAILS_ROOT
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet/notification'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet/logger'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet/active_record'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet/action_controller'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet/association'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet/counter'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bullet'))
+require File.expand_path(File.join(File.dirname(__FILE__), '../lib/bulletware'))
+Bullet.enable = true
+
+module BulletTestHelper
+  def silence_logger(&block)
+    orig_stdout = $stdout
+    $stdout = StringIO.new
+    block.call
+    $stdout = orig_stdout
+  end
+end
+
+module Bullet
+  class Association
+    class <<self
+      # returns true if all associations are preloaded
+      def completely_preloading_associations?
+        !has_unpreload_associations?
+      end
+
+      # returns true if a given object has a specific association
+      def creating_object_association_for?(object, association)
+        object_associations[object].present? && object_associations[object].include?(association)
+      end
+
+      # returns true if a given class includes the specific unpreloaded association
+      def detecting_unpreloaded_association_for?(klazz, association)
+        unpreload_associations[klazz].present? && unpreload_associations[klazz].include?(association)
+      end
+
+      # returns true if the given class includes the specific unused preloaded association
+      def unused_preload_associations_for?(klazz, association)
+         unused_preload_associations[klazz].present? && unused_preload_associations[klazz].include?(association)
+      end
+    end
+  end
+end

data/tasks/bullet_tasks.rake

@@ -0,0 +1,9 @@
+namespace :bullet do
+  namespace :log do
+    desc "Truncates the bullet log file to zero bytes"
+    task :clear do
+      f = File.open("log/bullet.log", "w")
+      f.close
+    end
+  end
+end

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification 
+name: bullet
+version: !ruby/object:Gem::Version 
+  version: 1.6.0
+platform: ruby
+authors: 
+- Richard Huang
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-08 00:00:00 +08:00
+default_executable: 
+dependencies: []
+
+description: The Bullet plugin is designed to help you increase your application's performance by reducing the number of queries it makes. It will watch your queries while you develop your application and notify you when you should add eager loading (N+1 queries) or when you're using eager loading that isn't necessary.
+email: flyerhzm@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.textile
+files: 
+- MIT-LICENSE
+- README.textile
+- Rakefile
+- VERSION
+- bullet.gemspec
+- lib/bullet.rb
+- lib/bullet/action_controller.rb
+- lib/bullet/active_record.rb
+- lib/bullet/association.rb
+- lib/bullet/counter.rb
+- lib/bullet/logger.rb
+- lib/bullet/notification.rb
+- lib/bulletware.rb
+- rails/init.rb
+- spec/bullet_association_spec.rb
+- spec/bullet_counter_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+- tasks/bullet_tasks.rake
+has_rdoc: true
+homepage: http://github.com/flyerhzm/bullet
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: A plugin to kill N+1 queries and unused eager loading
+test_files: 
+- spec/bullet_counter_spec.rb
+- spec/spec_helper.rb
+- spec/bullet_association_spec.rb