1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
module Test
module Unit
class CodeSnippetFetcher
def initialize
@sources = {}
end
def fetch(file, line, options={})
n_context_line = options[:n_context_line] || 3
lines = source(file)
return [] if lines.nil?
min_line = [line - n_context_line, 1].max
max_line = [line + n_context_line, lines.length].min
window = min_line..max_line
window.collect do |n|
attributes = {:target_line? => (n == line)}
[n, lines[n - 1].chomp, attributes]
end
end
def source(file)
@sources[file] ||= read_source(file)
end
private
def read_source(file)
return nil unless File.exist?(file)
File.readlines(file)
end
end
end
end
|
