lib/lotus/generators/application/container.rb in lotusrb-0.2.0 vs lib/lotus/generators/application/container.rb in lotusrb-0.2.1

@@ -8,19 +8,22 @@
         def initialize(command)
           super
 
           @slice_generator     = Slice.new(command)
           @lotus_head          = options.fetch(:lotus_head)
+          @test                = options[:test]
           @lotus_model_version = '~> 0.2'
 
           cli.class.source_root(source)
         end
 
         def start
+
           opts      = {
             app_name:            app_name,
             lotus_head:          @lotus_head,
+            test:                @test,
             lotus_model_version: @lotus_model_version
           }
 
           templates = {
             'Gemfile.tt'                 => 'Gemfile',
@@ -28,31 +31,40 @@
             'config/environment.rb.tt'   => 'config/environment.rb',
             'config/.env.tt'             => 'config/.env',
             'config/.env.development.tt' => 'config/.env.development',
             'config/.env.test.tt'        => 'config/.env.test',
             'lib/app_name.rb.tt'         => "lib/#{ app_name }.rb",
+            'lib/config/mapping.rb.tt'   => 'lib/config/mapping.rb',
           }
 
           empty_directories = [
             "db",
             "lib/#{ app_name }/entities",
             "lib/#{ app_name }/repositories"
           ]
 
           case options[:test]
           when 'rspec'
+            templates.merge!(
+              'Rakefile.rspec.tt'           => 'Rakefile',
+              'rspec.rspec.tt'              => '.rspec',
+              'spec_helper.rb.rspec.tt'     => 'spec/spec_helper.rb',
+              'features_helper.rb.rspec.tt' => 'spec/features_helper.rb',
+              'capybara.rb.rspec.tt'        => 'spec/support/capybara.rb'
+            )
           else # minitest (default)
             templates.merge!(
-              'Rakefile.minitest.tt'  => 'Rakefile',
-              'spec_helper.rb.tt'     => 'spec/spec_helper.rb',
-              'features_helper.rb.tt' => 'spec/features_helper.rb'
+              'Rakefile.minitest.tt'           => 'Rakefile',
+              'spec_helper.rb.minitest.tt'     => 'spec/spec_helper.rb',
+              'features_helper.rb.minitest.tt' => 'spec/features_helper.rb'
             )
-
-            empty_directories << [
-              "spec/#{ app_name }/entities",
-              "spec/#{ app_name }/repositories"
-            ]
           end
+
+          empty_directories << [
+            "spec/#{ app_name }/entities",
+            "spec/#{ app_name }/repositories",
+            "spec/support"
+          ]
 
           templates.each do |src, dst|
             cli.template(source.join(src), target.join(dst), opts)
           end