make cache buster more robust

lib/middleman/features/cache_buster.rb

@@ -1,14 +1,18 @@
 class Middleman::Base  
   after_feature_init do 
     ::Compass.configuration do |config|
-      config.asset_cache_buster do |path, real_path|
+      if self.respond_to?(:cache_buster?) && self.cache_buster?
-        real_path = real_path.path if real_path.is_a? File
+        config.asset_cache_buster do |path, real_path|
-        real_path = real_path.gsub(File.join(self.root, self.build_dir), self.public)
+          real_path = real_path.path if real_path.is_a? File
-        if File.readable?(real_path)
+          real_path = real_path.gsub(File.join(self.root, self.build_dir), self.public)
-          File.mtime(real_path).strftime("%s") 
+          if File.readable?(real_path)
-        else
+            File.mtime(real_path).strftime("%s") 
-          $stderr.puts "WARNING: '#{File.basename(path)}' was not found (or cannot be read) in #{File.dirname(real_path)}"
+          else
+            $stderr.puts "WARNING: '#{File.basename(path)}' was not found (or cannot be read) in #{File.dirname(real_path)}"
+          end
         end
+      else
+        config.asset_cache_buster { false }
       end
     end
     
@@ -21,6 +25,8 @@ class << Middleman::Base
   def asset_url(path, prefix="", request=nil)
     http_path = pre_cache_buster_asset_url(path, prefix, request)
     
+    return http_path unless self.respond_to?(:cache_buster?) && self.cache_buster?
+    
     if http_path.include?("://") || !%w(.css .png .jpg .js .gif).include?(File.extname(http_path))
       http_path
     else

spec/cache_buster_off_spec.rb

@@ -1,21 +1,20 @@
 require File.join(File.dirname(__FILE__), "spec_helper")
-
-describe "Cache Buster Feature in CSS" do
-  before :each do
-    @base = ::Middleman::Base
-    @base.set :root, File.join(File.dirname(__FILE__), "fixtures", "sample")
-    @base.disable :cache_buster
-  end
   
+describe "Cache Buster Feature in CSS" do
+  before do
+    base = ::Middleman::Base
+    base.set :root, File.join(File.dirname(__FILE__), "fixtures", "sample")
+    base.disable :cache_buster
+    @browser = Rack::Test::Session.new(Rack::MockSession.new(base.new))
+  end
+
   it "should not append query string in CSS if off" do
-    browser = Rack::Test::Session.new(Rack::MockSession.new(@base.new))
+    @browser.get("/stylesheets/relative_assets.css")
-    browser.get("/stylesheets/relative_assets.css")
+    @browser.last_response.body.count("?").should == 0
-    browser.last_response.body.should_not include("?")
   end
   
   it "should not append query string in HTML if off" do
-    browser = Rack::Test::Session.new(Rack::MockSession.new(@base.new))
+    @browser.get("/cache-buster.html")
-    browser.get("/cache-buster.html")
+    @browser.last_response.body.count("?").should == 0
-    browser.last_response.body.count("?").should == 0
   end
 end

spec/cache_buster_on_spec.rb

@@ -1,21 +1,20 @@
 require File.join(File.dirname(__FILE__), "spec_helper")
 
 describe "Cache Buster Feature in CSS" do
-  before :each do
+  before do 
-    @base = ::Middleman::Base
+    base = ::Middleman::Base
-    @base.set :root, File.join(File.dirname(__FILE__), "fixtures", "sample")
+    base.set :root, File.join(File.dirname(__FILE__), "fixtures", "sample")
-    @base.enable :cache_buster
+    base.enable :cache_buster
+    @browser = Rack::Test::Session.new(Rack::MockSession.new(base.new))
   end
   
   it "should append query string in CSS if on" do
-    browser = Rack::Test::Session.new(Rack::MockSession.new(@base.new))
+    @browser.get("/stylesheets/relative_assets.css")
-    browser.get("/stylesheets/relative_assets.css")
+    @browser.last_response.body.should include("?")
-    browser.last_response.body.should include("?")
   end
 
   it "should not append query string in HTML if on IN DEVELOPMENT" do
-    browser = Rack::Test::Session.new(Rack::MockSession.new(@base.new))
+    @browser.get("/cache-buster.html")
-    browser.get("/cache-buster.html")
+    @browser.last_response.body.count("?").should == 0
-    browser.last_response.body.count("?").should == 0
   end
 end