Merge pull request #58 from stitchfix/fix-bug-with-generator

make sure the include goes on its own line
stitchfix · Jun 15, 2018 · c457ed6 · c457ed6
2 parents b56890c + 928e94a
commit c457ed6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/lib/stitches/add_deprecation_generator.rb b/lib/stitches/add_deprecation_generator.rb
@@ -7,6 +7,7 @@ class AddDeprecationGenerator < Rails::Generators::Base
     desc "Adds deprecation support to an app creates with an older version of stitches"
     def add_deprecation
       inject_into_file "app/controllers/api/api_controller.rb", after: /^class.*$/ do<<-CODE
+
   include Stitches::Deprecation
       CODE
       end

diff --git a/spec/integration/add_to_rails_app_spec.rb b/spec/integration/add_to_rails_app_spec.rb
@@ -95,9 +95,12 @@ def run(command)
 
     run "bin/rails generate stitches:add_deprecation"
 
-    aggregate_failures do
-      expect(File.read(api_controller)).to include("include Stitches::Deprecation")
-    end
+    lines =  File.read(api_controller).split(/\n/)
+    include_line = lines.detect { |line|
+      line =~ /^\s+include Stitches::Deprecation$/
+    }
+
+    expect(include_line).to_not be_nil,lines.inspect
   end
 
   class RoutesFileAnalysis