query_diet 0.1.0

data/.gitignore

@@ -0,0 +1,5 @@
+doc
+pkg
+*.gem
+.idea
+

data/.idea/encodings.xml

@@ -0,0 +1,5 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Encoding" useUTFGuessing="true" native2AsciiForPropertiesFiles="false" />
+</project>
+

data/.idea/misc.xml

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="DependencyValidationManager">
+    <option name="SKIP_IMPORT_STATEMENTS" value="false" />
+  </component>
+  <component name="ProjectDetails">
+    <option name="projectName" value="query_diet" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Ruby SDK 1.8.7 (/usr/bin/ruby)" project-jdk-type="RUBY_SDK" />
+  <component name="SvnConfiguration">
+    <option name="USER" value="" />
+    <option name="PASSWORD" value="" />
+    <option name="LAST_MERGED_REVISION" />
+    <option name="UPDATE_RUN_STATUS" value="false" />
+    <option name="MERGE_DRY_RUN" value="false" />
+    <option name="MERGE_DIFF_USE_ANCESTRY" value="true" />
+    <option name="UPDATE_LOCK_ON_DEMAND" value="false" />
+    <option name="IGNORE_SPACES_IN_MERGE" value="false" />
+    <option name="DETECT_NESTED_COPIES" value="false" />
+    <option name="IGNORE_SPACES_IN_ANNOTATE" value="true" />
+    <option name="SHOW_MERGE_SOURCES_IN_ANNOTATE" value="true" />
+    <myIsUseDefaultProxy>false</myIsUseDefaultProxy>
+  </component>
+</project>
+

data/.idea/modules.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/query_diet.iml" filepath="$PROJECT_DIR$/.idea/query_diet.iml" />
+    </modules>
+  </component>
+</project>
+

data/.idea/query_diet.iml

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="RUBY_MODULE" version="4">
+  <component name="GemRequirementsHolder" version="2">
+    <requirement>
+      <requirement>
+        <dependency name="rails" version="0" bound="GREATER_OR_EQUAL" git="false" />
+      </requirement>
+      <source from="spec/app_root/config/boot.rb" />
+    </requirement>
+  </component>
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] rails (v2.3.5, /usr/lib/ruby/gems/1.8/gems/rails-2.3.5)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] activerecord (v2.3.5, /usr/lib/ruby/gems/1.8/gems/activerecord-2.3.5)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] rack (v1.0.1, /usr/lib/ruby/gems/1.8/gems/rack-1.0.1)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] rake (v0.8.7, /usr/lib/ruby/gems/1.8/gems/rake-0.8.7)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] activesupport (v2.3.5, /usr/lib/ruby/gems/1.8/gems/activesupport-2.3.5)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] actionpack (v2.3.5, /usr/lib/ruby/gems/1.8/gems/actionpack-2.3.5)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] actionmailer (v2.3.5, /usr/lib/ruby/gems/1.8/gems/actionmailer-2.3.5)" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="[gem] activeresource (v2.3.5, /usr/lib/ruby/gems/1.8/gems/activeresource-2.3.5)" level="application" />
+  </component>
+</module>
+

data/.idea/vcs.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="" />
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>
+

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Henning Koch
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,34 @@
+= query_diet
+
+query_diet counts the number of database queries for the last request and *subtly* displays it in the upper right corner of your screen.
+The display turns red if too many queries are run, or if they take too long.
+This is useful to prevent {N + 1 queries}[http://guides.rubyonrails.org/active_record_querying.html#eager-loading-associations] from creeping into your code.
+
+== Installation
+
+Install the gem with
+    sudo gem install query_diet
+
+There is no further setup necessary after you add the <tt>query_diet</tt> dependency to your Rails application.
+
+We recommend you only use the gem with the development environment.
+
+== Changing warning thresholds
+
+You can define when the counter turns into a red warning. The default threshold is 8 queries and 5000 miliseconds.
+To change the default, add the following to <tt>config/initializers/query_diet.rb</tt>:
+
+    if defined?(QueryDiet)
+      QueryDiet::Logger.bad_count = 8
+      QueryDiet::Logger.bad_time = 5000
+    end
+
+== Rails 3 compatibility
+
+We cannot guarantee Rails 3 compatibility at this point, but we will upgrade the gem when Rails 3 is released.
+
+=== Credits
+
+Henning Koch
+
+{www.makandra.de}[http://www.makandra.de/]

data/Rakefile

@@ -0,0 +1,37 @@
+require 'rake'
+# require 'rake/testtask'
+require 'rake/rdoctask'
+require 'spec/rake/spectask'
+
+desc 'Default: Run all specs.'
+task :default => :spec
+
+desc "Run all specs"
+Spec::Rake::SpecTask.new() do |t|
+  t.spec_opts = ['--options', "\"spec/support/spec.opts\""]
+  t.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+desc 'Generate documentation for the query_diet gem.'
+Rake::RDocTask.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'query_diet'
+  rdoc.options << '--line-numbers' << '--inline-source'
+  rdoc.rdoc_files.include('README')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "query_diet"
+    gemspec.summary = "Rails database query counter that stays out of your way"
+    gemspec.email = "github@makandra.de"
+    gemspec.homepage = "http://github.com/makandra/query_diet"
+    gemspec.description = "Rails database query counter that stays out of your way"
+    gemspec.authors = ["Henning Koch"]
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install jeweler"
+end
+

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/lib/query_diet/action_controller_ext.rb

@@ -0,0 +1,15 @@
+require_dependency 'application_controller'
+
+ApplicationController.class_eval do
+
+  around_filter :query_diet_logging
+
+  private
+
+  def query_diet_logging
+    QueryDiet::Logger.reset
+    yield
+    QueryDiet::Widget.render(response)
+  end
+
+end

data/lib/query_diet/active_record_ext.rb

@@ -0,0 +1,11 @@
+ActiveRecord::Base.connection.class.class_eval do
+
+  def execute_with_query_diet(query, name = nil)
+    QueryDiet::Logger.log(query) do
+      execute_without_query_diet(query, name)
+    end
+  end
+
+  alias_method_chain :execute, :query_diet
+
+end

data/lib/query_diet/logger.rb

@@ -0,0 +1,41 @@
+module QueryDiet
+  class Logger
+    class << self
+
+      attr_accessor :count, :time, :bad_count, :bad_time
+
+      def reset
+        self.count = 0
+        self.time = 0
+      end
+
+      def log(query, &execution)
+        result = nil
+        seconds = Benchmark.realtime do
+          result = execution.call
+        end
+        if log_query?(query)
+          self.time += (seconds * 1000).to_i
+          self.count += 1
+        end
+        result
+      end
+
+      def bad?
+        count >= bad_count or time >= bad_time
+      end
+
+      private
+
+      def log_query?(query)
+        query =~ /^(select|create|update|delete|insert)\b/i
+      end
+
+    end
+
+    reset
+    self.bad_count = 8
+    self.bad_time = 5000
+
+  end
+end

data/lib/query_diet/widget.rb

@@ -0,0 +1,58 @@
+module QueryDiet
+  class Widget
+    class << self
+
+      def render(response)
+        body = response.body
+        if response.content_type == "text/html" && position = body.index('</body>')
+          body.insert(position, css)
+          body.insert(position, html)
+        end
+      end
+
+      private
+
+      def css
+        <<-EOF
+        <style type="text/css"><!--
+          div#query_diet {
+            position: absolute;
+            top: 0px;
+            right: 0px;
+            background-color: black;
+            color: white;
+            z-index: 999;
+            padding: 4px 6px;
+            font-family; arial, sans-serif;
+            font-size: 12px;
+            line-height: 12px;
+            font-weight: bold;
+            cursor: pointer;
+          }
+          div#query_diet.good {
+            xbackground-color: #160;
+            filter:alpha(opacity=30);
+            -moz-opacity:0.3;
+            -khtml-opacity: 0.3;
+            opacity: 0.3;
+          }
+          div#query_diet.bad {
+            background-color: red;
+            font-size: 16px;
+            line-height: 16px;
+          }
+        --></style>
+        EOF
+      end
+
+      def html
+        <<-EOF
+        <div id="query_diet" class="#{QueryDiet::Logger.bad? ? 'bad' : 'good' }" onclick="this.parentNode.removeChild(this);">
+          #{QueryDiet::Logger.count} / #{QueryDiet::Logger.time}ms
+        </div>
+        EOF
+      end
+
+    end
+  end
+end

data/lib/query_diet.rb

@@ -0,0 +1,5 @@
+require 'query_diet/logger'
+require 'query_diet/widget'
+require 'query_diet/active_record_ext'
+require 'query_diet/action_controller_ext'
+

data/query_diet.gemspec

@@ -0,0 +1,93 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{query_diet}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Henning Koch"]
+  s.date = %q{2010-06-23}
+  s.description = %q{Rails database query counter that stays out of your way}
+  s.email = %q{github@makandra.de}
+  s.extra_rdoc_files = [
+    "README.rdoc"
+  ]
+  s.files = [
+    ".gitignore",
+     ".idea/encodings.xml",
+     ".idea/misc.xml",
+     ".idea/modules.xml",
+     ".idea/query_diet.iml",
+     ".idea/vcs.xml",
+     "MIT-LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "lib/query_diet.rb",
+     "lib/query_diet/action_controller_ext.rb",
+     "lib/query_diet/active_record_ext.rb",
+     "lib/query_diet/logger.rb",
+     "lib/query_diet/widget.rb",
+     "query_diet.gemspec",
+     "spec/app_root/app/controllers/application_controller.rb",
+     "spec/app_root/app/controllers/query_diet_controller.rb",
+     "spec/app_root/app/models/movie.rb",
+     "spec/app_root/app/views/layouts/screen.html.erb",
+     "spec/app_root/app/views/query_diet/no_query.html.erb",
+     "spec/app_root/app/views/query_diet/two_queries.html.erb",
+     "spec/app_root/config/boot.rb",
+     "spec/app_root/config/database.yml",
+     "spec/app_root/config/environment.rb",
+     "spec/app_root/config/environments/in_memory.rb",
+     "spec/app_root/config/environments/mysql.rb",
+     "spec/app_root/config/environments/postgresql.rb",
+     "spec/app_root/config/environments/sqlite.rb",
+     "spec/app_root/config/environments/sqlite3.rb",
+     "spec/app_root/config/routes.rb",
+     "spec/app_root/db/migrate/001_create_movies.rb",
+     "spec/app_root/lib/console_with_fixtures.rb",
+     "spec/app_root/log/.gitignore",
+     "spec/app_root/script/console",
+     "spec/controllers/query_diet_controller_spec.rb",
+     "spec/logger_spec.rb",
+     "spec/spec_helper.rb",
+     "spec/support/rcov.opts",
+     "spec/support/spec.opts"
+  ]
+  s.homepage = %q{http://github.com/makandra/query_diet}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{Rails database query counter that stays out of your way}
+  s.test_files = [
+    "spec/spec_helper.rb",
+     "spec/app_root/db/migrate/001_create_movies.rb",
+     "spec/app_root/config/boot.rb",
+     "spec/app_root/config/environment.rb",
+     "spec/app_root/config/routes.rb",
+     "spec/app_root/config/environments/in_memory.rb",
+     "spec/app_root/config/environments/mysql.rb",
+     "spec/app_root/config/environments/postgresql.rb",
+     "spec/app_root/config/environments/sqlite.rb",
+     "spec/app_root/config/environments/sqlite3.rb",
+     "spec/app_root/lib/console_with_fixtures.rb",
+     "spec/app_root/app/controllers/application_controller.rb",
+     "spec/app_root/app/controllers/query_diet_controller.rb",
+     "spec/app_root/app/models/movie.rb",
+     "spec/controllers/query_diet_controller_spec.rb",
+     "spec/logger_spec.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

data/spec/app_root/app/controllers/application_controller.rb

@@ -0,0 +1,2 @@
+class ApplicationController < ActionController::Base
+end

data/spec/app_root/app/controllers/query_diet_controller.rb

@@ -0,0 +1,21 @@
+class QueryDietController < ApplicationController
+
+  layout 'screen'
+
+  def no_query
+  end
+
+  def two_queries
+    Movie.create!
+    Movie.all
+  end
+
+  def no_body_tag
+    render :text => "no body tag", :content_type => "text/html"
+  end
+
+  def yaml
+    render :text => { :foo => '<body></body>' }.to_yaml, :content_type => 'text/yaml'
+  end
+
+end

data/spec/app_root/app/models/movie.rb

@@ -0,0 +1,3 @@
+class Movie < ActiveRecord::Base
+
+end

data/spec/app_root/app/views/layouts/screen.html.erb

@@ -0,0 +1,6 @@
+<html>
+  <body>
+    <h1>Title</h1>
+    <%= yield %>
+  </body>
+</html>

data/spec/app_root/app/views/query_diet/no_query.html.erb

@@ -0,0 +1,3 @@
+<p>
+  No query.
+</p>

data/spec/app_root/app/views/query_diet/two_queries.html.erb

@@ -0,0 +1,3 @@
+<p>
+  Two queries.
+</p>

data/spec/app_root/config/boot.rb

@@ -0,0 +1,114 @@
+# Allow customization of the rails framework path
+RAILS_FRAMEWORK_ROOT = (ENV['RAILS_FRAMEWORK_ROOT'] || "#{File.dirname(__FILE__)}/../../../../../../vendor/rails") unless defined?(RAILS_FRAMEWORK_ROOT) 
+
+# Don't change this file!
+# Configure your app in config/environment.rb and config/environments/*.rb
+
+RAILS_ROOT = "#{File.dirname(__FILE__)}/.." unless defined?(RAILS_ROOT)
+
+module Rails
+  class << self
+    def boot!
+      unless booted?
+        preinitialize
+        pick_boot.run
+      end
+    end
+
+    def booted?
+      defined? Rails::Initializer
+    end
+
+    def pick_boot
+      (vendor_rails? ? VendorBoot : GemBoot).new
+    end
+
+    def vendor_rails?
+      File.exist?(RAILS_FRAMEWORK_ROOT)
+    end
+
+    def preinitialize
+      load(preinitializer_path) if File.exist?(preinitializer_path)
+    end
+
+    def preinitializer_path
+      "#{RAILS_ROOT}/config/preinitializer.rb"
+    end
+  end
+
+  class Boot
+    def run
+      load_initializer
+      Rails::Initializer.run(:set_load_path)
+    end
+  end
+
+  class VendorBoot < Boot
+    def load_initializer
+      require "#{RAILS_FRAMEWORK_ROOT}/railties/lib/initializer"
+      Rails::Initializer.run(:install_gem_spec_stubs)
+    end
+  end
+
+  class GemBoot < Boot
+    def load_initializer
+      self.class.load_rubygems
+      load_rails_gem
+      require 'initializer'
+    end
+
+    def load_rails_gem
+      if version = self.class.gem_version
+        gem 'rails', version
+      else
+        gem 'rails'
+      end
+    rescue Gem::LoadError => load_error
+      $stderr.puts %(Missing the Rails #{version} gem. Please `gem install -v=#{version} rails`, update your RAILS_GEM_VERSION setting in config/environment.rb for the Rails version you do have installed, or comment out RAILS_GEM_VERSION to use the latest version installed.)
+      exit 1
+    end
+
+    class << self
+      def rubygems_version
+        Gem::RubyGemsVersion rescue nil
+      end
+
+      def gem_version
+        if defined? RAILS_GEM_VERSION
+          RAILS_GEM_VERSION
+        elsif ENV.include?('RAILS_GEM_VERSION')
+          ENV['RAILS_GEM_VERSION']
+        else
+          parse_gem_version(read_environment_rb)
+        end
+      end
+
+      def load_rubygems
+        require 'rubygems'
+        min_version = '1.1.1'
+        unless rubygems_version >= min_version
+          $stderr.puts %Q(Rails requires RubyGems >= #{min_version} (you have #{rubygems_version}). Please `gem update --system` and try again.)
+          exit 1
+        end
+ 
+      rescue LoadError
+        $stderr.puts %Q(Rails requires RubyGems >= #{min_version}. Please install RubyGems and try again: http://rubygems.rubyforge.org)
+        exit 1
+      end
+
+      def parse_gem_version(text)
+        $1 if text =~ /^[^#]*RAILS_GEM_VERSION\s*=\s*["']([!~<>=]*\s*[\d.]+)["']/
+      end
+
+      private
+        def read_environment_rb
+          environment_rb = "#{RAILS_ROOT}/config/environment.rb"
+          environment_rb = "#{HELPER_RAILS_ROOT}/config/environment.rb" unless File.exists?(environment_rb)
+          File.read(environment_rb)
+        end
+    end
+  end
+end
+
+# All that for this:
+Rails.boot!

data/spec/app_root/config/database.yml

@@ -0,0 +1,21 @@
+in_memory:
+  adapter: sqlite3
+  database: ":memory:"
+  verbosity: quiet
+sqlite:
+  adapter: sqlite
+  dbfile: plugin_test.sqlite.db
+sqlite3:
+  adapter: sqlite3
+  dbfile: plugin_test.sqlite3.db
+postgresql:
+  adapter: postgresql
+  username: postgres
+  password: postgres
+  database: plugin_test
+mysql:
+  adapter: mysql
+  host: localhost
+  username: root
+  password:
+  database: plugin_test

data/spec/app_root/config/environment.rb

@@ -0,0 +1,14 @@
+require File.join(File.dirname(__FILE__), 'boot')
+
+Rails::Initializer.run do |config|
+  config.cache_classes = false
+  config.whiny_nils = true
+  config.action_controller.session = { :key => "_myapp_session", :secret => "gwirofjweroijger8924rt2zfwehfuiwehb1378rifowenfoqwphf23" }
+  config.plugin_locators.unshift(
+    Class.new(Rails::Plugin::Locator) do
+      def plugins
+        [Rails::Plugin.new(File.expand_path('.'))]
+      end
+    end
+  ) unless defined?(PluginTestHelper::PluginLocator)
+end

data/spec/app_root/config/routes.rb

@@ -0,0 +1,4 @@
+ActionController::Routing::Routes.draw do |map|
+  map.connect ':controller/:action/:id'
+  map.connect ':controller/:action/:id.:format'
+end

data/spec/app_root/db/migrate/001_create_movies.rb

@@ -0,0 +1,14 @@
+class CreateMovies < ActiveRecord::Migration
+
+  def self.up
+    create_table :movies do |t|
+      t.string :title
+      t.integer :year
+    end
+  end
+
+  def self.down
+    drop_table :movies
+  end
+  
+end

data/spec/app_root/lib/console_with_fixtures.rb

@@ -0,0 +1,4 @@
+# Loads fixtures into the database when running the test app via the console
+(ENV['FIXTURES'] ? ENV['FIXTURES'].split(/,/) : Dir.glob(File.join(Rails.root, '../fixtures/*.{yml,csv}'))).each do |fixture_file|
+  Fixtures.create_fixtures(File.join(Rails.root, '../fixtures'), File.basename(fixture_file, '.*'))
+end

data/spec/app_root/log/.gitignore

@@ -0,0 +1 @@
+*.log

data/spec/app_root/script/console

@@ -0,0 +1,7 @@
+irb = RUBY_PLATFORM =~ /(:?mswin|mingw)/ ? 'irb.bat' : 'irb'
+libs =  " -r irb/completion"
+libs << " -r test/test_helper"
+libs << " -r console_app"
+libs << " -r console_with_helpers"
+libs << " -r console_with_fixtures"
+exec "#{irb} #{libs} --simple-prompt"

data/spec/controllers/query_diet_controller_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+
+describe QueryDietController do
+  integrate_views
+
+  describe 'widget' do
+
+    it "should be automatically included in a HTML page" do
+      get :two_queries
+      response.body.should have_tag('div#query_diet')
+    end
+
+    it "should not be included in pages that have no body tag" do
+      get :no_body_tag
+      response.body.should_not have_tag('div#query_diet')
+    end
+
+    it "should not be included in pages that are not text/html" do
+      get :yaml
+      response.body.should_not have_tag('div#query_diet')
+    end
+
+    it "should be highlighted if the request was too intimate with the database" do
+      QueryDiet::Logger.stub :bad? => true
+      get :no_query
+      response.body.should have_tag('div#query_diet.bad')
+    end
+
+    it "should not be highlighted if the request spent little time in the database" do
+      QueryDiet::Logger.stub :bad? => false
+      get :no_query
+      response.body.should have_tag('div#query_diet.good')
+    end
+
+    it "should contain the number of requests and the duration spent running queries" do
+      QueryDiet::Logger.stub :count => 78, :time => 43500
+      get :no_query
+      response.body.should have_tag('div#query_diet', :text => "78 / 43500ms")
+    end
+
+  end
+
+end

data/spec/logger_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe QueryDiet::Logger do
+
+  before :each do
+    QueryDiet::Logger.reset
+  end
+
+  describe 'count' do
+
+    it "should return the number of queries since the last reset" do
+      Movie.create
+      Movie.create
+      QueryDiet::Logger.count.should == 2
+    end
+
+  end
+
+  describe 'time' do
+
+    it "should return the number of miliseconds spent running database queries since the last reset" do
+      Benchmark.should_receive(:realtime).and_return(5)
+      Movie.create
+      QueryDiet::Logger.time.should == 5000
+    end
+
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+$: << File.join(File.dirname(__FILE__), "/../lib" )
+
+# Set the default environment to sqlite3's in_memory database
+ENV['RAILS_ENV'] ||= 'in_memory'
+
+# Load the Rails environment and testing framework
+require "#{File.dirname(__FILE__)}/app_root/config/environment"
+require "#{File.dirname(__FILE__)}/../lib/query_diet"
+require 'spec/rails'
+
+# Undo changes to RAILS_ENV
+silence_warnings {RAILS_ENV = ENV['RAILS_ENV']}
+
+# Run the migrations
+ActiveRecord::Migrator.migrate("#{Rails.root}/db/migrate")
+
+Spec::Runner.configure do |config|
+  config.use_transactional_fixtures = true
+  config.use_instantiated_fixtures  = false
+end
+

data/spec/support/rcov.opts

@@ -0,0 +1,2 @@
+--exclude "spec/*,gems/*"
+--rails

data/spec/support/spec.opts

@@ -0,0 +1,4 @@
+--colour
+--format progress
+--loadby mtime
+--reverse

metadata

@@ -0,0 +1,116 @@
+--- !ruby/object:Gem::Specification 
+name: query_diet
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Henning Koch
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-06-23 00:00:00 +02:00
+default_executable: 
+dependencies: []
+
+description: Rails database query counter that stays out of your way
+email: github@makandra.de
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- .gitignore
+- .idea/encodings.xml
+- .idea/misc.xml
+- .idea/modules.xml
+- .idea/query_diet.iml
+- .idea/vcs.xml
+- MIT-LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/query_diet.rb
+- lib/query_diet/action_controller_ext.rb
+- lib/query_diet/active_record_ext.rb
+- lib/query_diet/logger.rb
+- lib/query_diet/widget.rb
+- query_diet.gemspec
+- spec/app_root/app/controllers/application_controller.rb
+- spec/app_root/app/controllers/query_diet_controller.rb
+- spec/app_root/app/models/movie.rb
+- spec/app_root/app/views/layouts/screen.html.erb
+- spec/app_root/app/views/query_diet/no_query.html.erb
+- spec/app_root/app/views/query_diet/two_queries.html.erb
+- spec/app_root/config/boot.rb
+- spec/app_root/config/database.yml
+- spec/app_root/config/environment.rb
+- spec/app_root/config/environments/in_memory.rb
+- spec/app_root/config/environments/mysql.rb
+- spec/app_root/config/environments/postgresql.rb
+- spec/app_root/config/environments/sqlite.rb
+- spec/app_root/config/environments/sqlite3.rb
+- spec/app_root/config/routes.rb
+- spec/app_root/db/migrate/001_create_movies.rb
+- spec/app_root/lib/console_with_fixtures.rb
+- spec/app_root/log/.gitignore
+- spec/app_root/script/console
+- spec/controllers/query_diet_controller_spec.rb
+- spec/logger_spec.rb
+- spec/spec_helper.rb
+- spec/support/rcov.opts
+- spec/support/spec.opts
+has_rdoc: true
+homepage: http://github.com/makandra/query_diet
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Rails database query counter that stays out of your way
+test_files: 
+- spec/spec_helper.rb
+- spec/app_root/db/migrate/001_create_movies.rb
+- spec/app_root/config/boot.rb
+- spec/app_root/config/environment.rb
+- spec/app_root/config/routes.rb
+- spec/app_root/config/environments/in_memory.rb
+- spec/app_root/config/environments/mysql.rb
+- spec/app_root/config/environments/postgresql.rb
+- spec/app_root/config/environments/sqlite.rb
+- spec/app_root/config/environments/sqlite3.rb
+- spec/app_root/lib/console_with_fixtures.rb
+- spec/app_root/app/controllers/application_controller.rb
+- spec/app_root/app/controllers/query_diet_controller.rb
+- spec/app_root/app/models/movie.rb
+- spec/controllers/query_diet_controller_spec.rb
+- spec/logger_spec.rb