76 lines
1.6 KiB
Ruby
76 lines
1.6 KiB
Ruby
($:.unshift '../..'; require 'coderay') unless defined? CodeRay
|
|
module CodeRay
|
|
module Encoders
|
|
|
|
class Filter < Encoder
|
|
|
|
register_for :filter
|
|
|
|
protected
|
|
def setup options
|
|
@out = Tokens.new
|
|
end
|
|
|
|
def text_token text, kind
|
|
[text, kind] if include_text_token? text, kind
|
|
end
|
|
|
|
def include_text_token? text, kind
|
|
true
|
|
end
|
|
|
|
def block_token action, kind
|
|
[action, kind] if include_block_token? action, kind
|
|
end
|
|
|
|
def include_block_token? action, kind
|
|
true
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
|
|
if $0 == __FILE__
|
|
$VERBOSE = true
|
|
$: << File.join(File.dirname(__FILE__), '..')
|
|
eval DATA.read, nil, $0, __LINE__ + 4
|
|
end
|
|
|
|
__END__
|
|
require 'test/unit'
|
|
|
|
class FilterTest < Test::Unit::TestCase
|
|
|
|
def test_creation
|
|
assert CodeRay::Encoders::Filter < CodeRay::Encoders::Encoder
|
|
filter = nil
|
|
assert_nothing_raised do
|
|
filter = CodeRay.encoder :filter
|
|
end
|
|
assert_kind_of CodeRay::Encoders::Encoder, filter
|
|
end
|
|
|
|
def test_filtering_text_tokens
|
|
tokens = CodeRay::Tokens.new
|
|
10.times do |i|
|
|
tokens << [i.to_s, :index]
|
|
end
|
|
assert_equal tokens, CodeRay::Encoders::Filter.new.encode_tokens(tokens)
|
|
assert_equal tokens, tokens.filter
|
|
end
|
|
|
|
def test_filtering_block_tokens
|
|
tokens = CodeRay::Tokens.new
|
|
10.times do |i|
|
|
tokens << [:open, :index]
|
|
tokens << [i.to_s, :content]
|
|
tokens << [:close, :index]
|
|
end
|
|
assert_equal tokens, CodeRay::Encoders::Filter.new.encode_tokens(tokens)
|
|
assert_equal tokens, tokens.filter
|
|
end
|
|
|
|
end
|