diff --git a/lib/sitemap_generator/application.rb b/lib/sitemap_generator/application.rb index 9febc254..388310f9 100644 --- a/lib/sitemap_generator/application.rb +++ b/lib/sitemap_generator/application.rb @@ -3,7 +3,7 @@ module SitemapGenerator class Application def rails? - defined?(Rails) + defined?(Rails::Railtie) end # Returns a boolean indicating whether this environment is Rails 3 diff --git a/spec/sitemap_generator/application_spec.rb b/spec/sitemap_generator/application_spec.rb index 824d8fc6..4c07ab8d 100644 --- a/spec/sitemap_generator/application_spec.rb +++ b/spec/sitemap_generator/application_spec.rb @@ -3,7 +3,8 @@ describe SitemapGenerator::Application do before :all do SitemapGenerator::Utilities.with_warnings(nil) do - Object.const_set(:Rails, Object.new) + Object.const_set(:Rails, Object) + Object::Rails.const_set(:Railtie, Object) end end