diff --git a/test/rdoc/test_rdoc_token_stream.rb b/test/rdoc/test_rdoc_token_stream.rb index 4a0ddb9fecd0fe..8fcd3d8f68d339 100644 --- a/test/rdoc/test_rdoc_token_stream.rb +++ b/test/rdoc/test_rdoc_token_stream.rb @@ -39,6 +39,42 @@ def test_class_to_html_empty assert_equal '', RDoc::TokenStream.to_html([]) end + def test_add_tokens + foo = Class.new do + include RDoc::TokenStream + end.new + foo.collect_tokens + foo.add_tokens([:token]) + assert_equal [:token], foo.token_stream + end + + def test_add_token + foo = Class.new do + include RDoc::TokenStream + end.new + foo.collect_tokens + foo.add_token(:token) + assert_equal [:token], foo.token_stream + end + + def test_collect_tokens + foo = Class.new do + include RDoc::TokenStream + end.new + foo.collect_tokens + assert_equal [], foo.token_stream + end + + def test_pop_token + foo = Class.new do + include RDoc::TokenStream + end.new + foo.collect_tokens + foo.add_token(:token) + foo.pop_token + assert_equal [], foo.token_stream + end + def test_token_stream foo = Class.new do include RDoc::TokenStream