Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions railties/lib/rails/source_annotation_extractor.rb
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,12 @@ def self.directories
@@directories ||= %w(app config db lib test) + (ENV['SOURCE_ANNOTATION_DIRECTORIES'] || '').split(',')
end

# Registers additional directories to be included
# SourceAnnotationExtractor::Annotation.register_directories("spec","another")
def self.register_directories(*dirs)
directories.push(*dirs)
end

def self.extensions
@@extensions ||= {}
end
Expand Down
12 changes: 12 additions & 0 deletions railties/test/application/rake/notes_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,18 @@ def teardown
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 %q{ 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')
Expand Down