Updated generators to make enum inheritance more explicit

beerlington · Jan 29, 2011 · da9f8fd · da9f8fd
1 parent d25dec9
commit da9f8fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/generators/templates/enum.rb b/generators/templates/enum.rb
@@ -2,6 +2,6 @@ class <%= class_name %> < ClassyEnum::Base
   enum_classes <%= args.map {|a| ":#{a}"}.join(", ") %>
 end
 <% args.each do |arg| %>
-class <%= class_name + arg.camelize %>
+class <%= class_name + arg.camelize %> < <%= class_name %>
 end
 <% end %>
diff --git a/lib/generators/classy_enum/templates/enum.rb b/lib/generators/classy_enum/templates/enum.rb
@@ -2,6 +2,6 @@ class <%= class_name %> < ClassyEnum::Base
   enum_classes <%= values.map {|a| ":#{a}"}.join(", ") %>
 end
 <% values.each do |arg| %>
-class <%= class_name + arg.camelize %>
+class <%= class_name + arg.camelize %> < <%= class_name %>
 end
 <% end %>