zoo-generators 0.1.2 → 0.1.3

data/features/zoo_scaffold.feature

@@ -21,8 +21,16 @@ Feature: Zoo Scaffold Generator
     And I run "rake db:migrate"
     And I should successfully run "rails g zoo:scaffold Project -f" 
     And I run "bundle install"
-    Then I should successfully run "rake test"
-
+    Then I should successfully run "rake test"  
+       
+  Scenario: Generate scaffold for existing resource
+    Given a new Rails app
+    When I run "rails g model Project some:string" 
+    And I run "rake db:migrate" 
+    And I run "rails g zoo:scaffold Project"
+    Then I should see "input :some" in file "app/views/projects/_form.html.haml" 
+    
+    
   Scenario: Generate scaffold with rspec tests
     Given a new Rails app
     When I run "rails g zoo:scaffold Project name:string --rspec"

data/lib/generators/zoo/scaffold/scaffold_generator.rb

@@ -43,7 +43,7 @@ module Zoo
             @controller_actions << 'update' if arg == 'edit'
           end
         end
-
+        
         @controller_actions.uniq!
         @model_attributes.uniq!
 
@@ -58,10 +58,12 @@ module Zoo
               @model_attributes << Rails::Generators::GeneratedAttribute.new(column.name.to_s, column.type.to_s)
             end
           else
-            klass = class_name.new
-            klass.attributes.each do
+            unless class_name.nil?      
+              class_name.constantize.columns.each do |column|
                 @model_attributes << Rails::Generators::GeneratedAttribute.new(column.name.to_s, column.type.to_s)
+              end
             end
+            @model_attributes << Rails::Generators::GeneratedAttribute.new("name", "string") if @model_attributes.empty?
           end
         end
       end

metadata

@@ -2,7 +2,7 @@
 name: zoo-generators
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors: 
 - Mikael Henriksson