aboutsummaryrefslogblamecommitdiffstats
path: root/railties/test/application/rake/notes_test.rb
blob: 8e9fe9b6b4202213c89a6c9334bf004c486f78a3 (plain) (tree)
1
2
3
4
5
6
7
8
9
10

                             
                                 
                                           


                       
                                                 

                                               
               

                           
             
         
 
                  
             


                    
                                                        
                                                                             

                                                                                 
                                                                                                                        
                                                              


                                                                               
                                                                    
 
                                         
                                             

                                             
                                              
                                                 

                                              
                                              
 
                                    
                                                  
           
         
 
                                                        

                                                                                              
                                                                   
                                                                    

                                                                                                          

                                                                                 
                                         

                                                          
                                                      
                                                       

                                                                 
 
                                    
                                                  
           
         
 
                                                       

                                                                                              
                                                                   
                                                                    



                                                                                                          
                                                                                                          

                                                          
                                                      
                                                       
                                                        
 
                                                              
 
                                    
                                                  
           

         
                                                                             
                                                                                      

                                                                            










                                                                                                          
                                                                  





                                                                 
                                                  


           
                                        
                                                     



                                                                                                                                      
                                         



                                              

         
                                               

                                                                         
                                                                         
 






                                                    
             
 


                                                       
 


                                                            
 

                               
           
 



                                 
 

                                      
 


                                                  


       
# frozen_string_literal: true

require "isolation/abstract_unit"
require "rails/source_annotation_extractor"

module ApplicationTests
  module RakeTests
    class RakeNotesTest < ActiveSupport::TestCase
      include ActiveSupport::Testing::Isolation

      def setup
        build_app
        require "rails/all"
        super
      end

      def teardown
        super
        teardown_app
      end

      test "notes finds notes for certain file_types" do
        app_file "app/views/home/index.html.erb", "<% # TODO: note in erb %>"
        app_file "app/assets/javascripts/application.js", "// TODO: note in js"
        app_file "app/assets/stylesheets/application.css", "// TODO: note in css"
        app_file "app/controllers/application_controller.rb", 1000.times.map { "" }.join("\n") << "# TODO: note in ruby"
        app_file "lib/tasks/task.rake", "# TODO: note in rake"
        app_file "app/views/home/index.html.builder", "# TODO: note in builder"
        app_file "config/locales/en.yml", "# TODO: note in yml"
        app_file "config/locales/en.yaml", "# TODO: note in yaml"
        app_file "app/views/home/index.ruby", "# TODO: note in ruby"

        run_rake_notes do |output, lines|
          assert_match(/note in erb/, output)
          assert_match(/note in js/, output)
          assert_match(/note in css/, output)
          assert_match(/note in rake/, output)
          assert_match(/note in builder/, output)
          assert_match(/note in yml/, output)
          assert_match(/note in yaml/, output)
          assert_match(/note in ruby/, output)

          assert_equal 9, lines.size
          assert_equal [4], lines.map(&:size).uniq
        end
      end

      test "notes finds notes in default directories" do
        app_file "app/controllers/some_controller.rb", "# TODO: note in app directory"
        app_file "config/initializers/some_initializer.rb", "# TODO: note in config directory"
        app_file "db/some_seeds.rb", "# TODO: note in db directory"
        app_file "lib/some_file.rb", "# TODO: note in lib directory"
        app_file "test/some_test.rb", 1000.times.map { "" }.join("\n") << "# TODO: note in test directory"

        app_file "some_other_dir/blah.rb", "# TODO: note in some_other directory"

        run_rake_notes do |output, lines|
          assert_match(/note in app directory/, output)
          assert_match(/note in config directory/, output)
          assert_match(/note in db directory/, output)
          assert_match(/note in lib directory/, output)
          assert_match(/note in test directory/, output)
          assert_no_match(/note in some_other directory/, output)

          assert_equal 5, lines.size
          assert_equal [4], lines.map(&:size).uniq
        end
      end

      test "notes finds notes in custom directories" do
        app_file "app/controllers/some_controller.rb", "# TODO: note in app directory"
        app_file "config/initializers/some_initializer.rb", "# TODO: note in config directory"
        app_file "db/some_seeds.rb", "# TODO: note in db directory"
        app_file "lib/some_file.rb", "# TODO: note in lib directory"
        app_file "test/some_test.rb", 1000.times.map { "" }.join("\n") << "# TODO: note in test directory"

        app_file "some_other_dir/blah.rb", "# TODO: note in some_other directory"

        run_rake_notes "SOURCE_ANNOTATION_DIRECTORIES='some_other_dir' bin/rails notes" do |output, lines|
          assert_match(/note in app directory/, output)
          assert_match(/note in config directory/, output)
          assert_match(/note in db directory/, output)
          assert_match(/note in lib directory/, output)
          assert_match(/note in test directory/, output)

          assert_match(/note in some_other directory/, output)

          assert_equal 6, lines.size
          assert_equal [4], lines.map(&:size).uniq
        end
      end

      test "custom rake task finds specific notes in specific directories" do
        app_file "app/controllers/some_controller.rb", "# TODO: note in app directory"
        app_file "lib/some_file.rb", "# OPTIMIZE: note in lib directory\n" \
          "# FIXME: note in lib directory"
        app_file "test/some_test.rb", 1000.times.map { "" }.join("\n") << "# TODO: note in test directory"

        app_file "lib/tasks/notes_custom.rake", <<-EOS
          require 'rails/source_annotation_extractor'
          task :notes_custom do
            tags = 'TODO|FIXME'
            opts = { dirs: %w(lib test), tag: true }
            SourceAnnotationExtractor.enumerate(tags, opts)
          end
        EOS

        run_rake_notes "bin/rails notes_custom" do |output, lines|
          assert_match(/\[FIXME\] note in lib directory/, output)
          assert_match(/\[TODO\] note in test directory/, output)
          assert_no_match(/OPTIMIZE/, output)
          assert_no_match(/note in app directory/, output)

          assert_equal 2, lines.size
          assert_equal [4], lines.map(&:size).uniq
        end
      end

      test "register a new extension" do
        add_to_config "config.assets.precompile = []"
        add_to_config %q{ config.annotations.register_extensions("scss", "sass") { |annotation| /\/\/\s*(#{annotation}):?\s*(.*)$/ } }
        app_file "app/assets/stylesheets/application.css.scss", "// TODO: note in scss"
        app_file "app/assets/stylesheets/application.css.sass", "// TODO: note in sass"

        run_rake_notes do |output, lines|
          assert_match(/note in scss/, output)
          assert_match(/note in sass/, output)
          assert_equal 2, lines.size
        end
      end

      test "register additional directories" do
        app_file "spec/spec_helper.rb", "# TODO: note in spec"
        app_file "spec/models/user_spec.rb", "# TODO: note in model spec"
        add_to_config ' config.annotations.register_directories("spec") '

        run_rake_notes do |output, lines|
          assert_match(/note in spec/, output)
          assert_match(/note in model spec/, output)
          assert_equal 2, lines.size
        end
      end

      private

        def run_rake_notes(command = "bin/rails notes")
          boot_rails
          load_tasks

          Dir.chdir(app_path) do
            output = `#{command}`
            lines  = output.scan(/\[([0-9\s]+)\]\s/).flatten

            yield output, lines
          end
        end

        def load_tasks
          require "rake"
          require "rdoc/task"
          require "rake/testtask"

          Rails.application.load_tasks
        end

        def boot_rails
          require "#{app_path}/config/environment"
        end
    end
  end
end