summaryrefslogtreecommitdiff
diff options
-rw-r--r--lib/rdoc/token_stream.rb2
-rw-r--r--test/rdoc/test_rdoc_token_stream.rb9
2 files changed, 10 insertions, 1 deletions
diff --git a/lib/rdoc/token_stream.rb b/lib/rdoc/token_stream.rb
index 8fc6eadd85..1ff4b0d09e 100644
--- a/lib/rdoc/token_stream.rb
+++ b/lib/rdoc/token_stream.rb
@@ -105,7 +105,7 @@ module RDoc::TokenStream
# Current token stream
def token_stream
- @token_stream
+ @token_stream || []
end
##
diff --git a/test/rdoc/test_rdoc_token_stream.rb b/test/rdoc/test_rdoc_token_stream.rb
index 29c5047fb5..dafbe22323 100644
--- a/test/rdoc/test_rdoc_token_stream.rb
+++ b/test/rdoc/test_rdoc_token_stream.rb
@@ -53,5 +53,14 @@ class TestRDocTokenStream < RDoc::TestCase
end.new
assert_equal "foo 'bar'", foo.tokens_to_s
+
+ foo = Class.new do
+ include RDoc::TokenStream
+
+ def initialize
+ @token_stream = nil
+ end
+ end.new
+ assert_equal "", foo.tokens_to_s
end
end