From b2485bcfb7aa5718ec541baa0b44445b76af7b04 Mon Sep 17 00:00:00 2001
From: Steve Ellis <sellis6688@gmail.com>
Date: Sat, 31 Mar 2012 11:36:27 -0400
Subject: [PATCH] fix generator naming for CamelCased models

---
 .../draper/decorator/decorator_generator.rb   |  2 +-
 .../draper/decorator/templates/decorator.rb   |  2 +-
 .../decorator/decorator_generator_spec.rb     | 41 ++++++++++---------
 3 files changed, 23 insertions(+), 22 deletions(-)

diff --git a/lib/generators/draper/decorator/decorator_generator.rb b/lib/generators/draper/decorator/decorator_generator.rb
index 5f8ab2d..ff4face 100644
--- a/lib/generators/draper/decorator/decorator_generator.rb
+++ b/lib/generators/draper/decorator/decorator_generator.rb
@@ -40,7 +40,7 @@ module Draper
     end
 
     def decorator_name
-      resource_name.downcase.singularize
+      resource_name.underscore.singularize
     end
   end
 end
diff --git a/lib/generators/draper/decorator/templates/decorator.rb b/lib/generators/draper/decorator/templates/decorator.rb
index beeeb88..2c91919 100644
--- a/lib/generators/draper/decorator/templates/decorator.rb
+++ b/lib/generators/draper/decorator/templates/decorator.rb
@@ -1,5 +1,5 @@
 class <%= resource_name.singularize.camelize %>Decorator < ApplicationDecorator
-  decorates :<%= resource_name.singularize.to_sym %>
+  decorates :<%= resource_name.singularize.underscore.to_sym %>
 
   # Accessing Helpers
   #   You can access any helper via a proxy
diff --git a/spec/generators/draper/decorator/decorator_generator_spec.rb b/spec/generators/draper/decorator/decorator_generator_spec.rb
index f5b933f..1d6ffb2 100644
--- a/spec/generators/draper/decorator/decorator_generator_spec.rb
+++ b/spec/generators/draper/decorator/decorator_generator_spec.rb
@@ -10,52 +10,53 @@ describe Draper::DecoratorGenerator do
   before { prepare_destination }
 
   context 'decorator context' do
-    before { run_generator ["product"]  }
+    before { run_generator ["YourModel"]  }
 
-    describe 'app/decorators/product_decorator.rb' do
-      subject { file('app/decorators/product_decorator.rb') }
+    describe 'app/decorators/your_model_decorator.rb' do
+      subject { file('app/decorators/your_model_decorator.rb') }
       it { should exist }
-      it { should contain "class ProductDecorator < ApplicationDecorator" }
+      it { should contain "class YourModelDecorator < ApplicationDecorator" }
+      it { should contain "decorates :your_model" }
     end
   end
 
   context 'decorator name' do
-    before { run_generator ["Product"]  }
+    before { run_generator ["YourModel"]  }
 
-    describe 'spec/decorators/product_decorator_spec.rb' do
-      subject { file('spec/decorators/product_decorator_spec.rb') }
+    describe 'spec/decorators/your_model_decorator_spec.rb' do
+      subject { file('spec/decorators/your_model_decorator_spec.rb') }
       it { should exist }
-      it { should contain "describe ProductDecorator" }
+      it { should contain "describe YourModelDecorator" }
     end
   end
 
   context 'default test framework' do
-    before { run_generator ["product"]  }
+    before { run_generator ["YourModel"]  }
 
-    describe 'spec/decorators/product_decorator_spec.rb' do
-      subject { file('spec/decorators/product_decorator_spec.rb') }
+    describe 'spec/decorators/your_model_decorator_spec.rb' do
+      subject { file('spec/decorators/your_model_decorator_spec.rb') }
       it { should exist }
-      it { should contain "describe ProductDecorator" }
+      it { should contain "describe YourModelDecorator" }
     end
   end
 
   context 'using rspec' do
-    before { run_generator ["product", "-t=rspec"]  }
+    before { run_generator ["YourModel", "-t=rspec"]  }
 
-    describe 'spec/decorators/product_decorator_spec.rb' do
-      subject { file('spec/decorators/product_decorator_spec.rb') }
+    describe 'spec/decorators/your_model_decorator_spec.rb' do
+      subject { file('spec/decorators/your_model_decorator_spec.rb') }
       it { should exist }
-      it { should contain "describe ProductDecorator" }
+      it { should contain "describe YourModelDecorator" }
     end
   end
 
   context 'using rspec' do
-    before { run_generator ["product", "-t=test_unit"]  }
+    before { run_generator ["YourModel", "-t=test_unit"]  }
 
-    describe 'test/decorators/product_decorator_test.rb' do
-      subject { file('test/decorators/product_decorator_test.rb') }
+    describe 'test/decorators/YourModel_decorator_test.rb' do
+      subject { file('test/decorators/your_model_decorator_test.rb') }
       it { should exist }
-      it { should contain "class ProductDecoratorTest < ActiveSupport::TestCase" }
+      it { should contain "class YourModelDecoratorTest < ActiveSupport::TestCase" }
     end
   end
 end