From 0f9307914b08a60102d29cee2260358bd2aaac75 Mon Sep 17 00:00:00 2001 From: tenderlove Date: Mon, 29 Oct 2012 01:23:54 +0000 Subject: * ext/psych/lib/psych/handlers/recorder.rb: added a class for recording YAML parse and emit events. * ext/psych/lib/psych/handler.rb: adding a list of events so that handler classes can more easily be meta-programmed. * test/psych/handlers/test_recorder.rb: tests for the change. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@37378 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 10 ++++++++ ext/psych/lib/psych/handler.rb | 13 +++++++++++ ext/psych/lib/psych/handlers/recorder.rb | 39 ++++++++++++++++++++++++++++++++ test/psych/handlers/test_recorder.rb | 25 ++++++++++++++++++++ 4 files changed, 87 insertions(+) create mode 100644 ext/psych/lib/psych/handlers/recorder.rb create mode 100644 test/psych/handlers/test_recorder.rb diff --git a/ChangeLog b/ChangeLog index 5ce5bcbf14..7732083476 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,13 @@ +Mon Oct 29 10:22:00 2012 Aaron Patterson + + * ext/psych/lib/psych/handlers/recorder.rb: added a class for + recording YAML parse and emit events. + + * ext/psych/lib/psych/handler.rb: adding a list of events so that + handler classes can more easily be meta-programmed. + + * test/psych/handlers/test_recorder.rb: tests for the change. + Mon Oct 29 05:48:52 2012 Marc-Andre Lafortune * lib/ostruct.rb: Add [] and []=, base on a patch by Thomas Sawyer. diff --git a/ext/psych/lib/psych/handler.rb b/ext/psych/lib/psych/handler.rb index d3b99636c4..c55afe745f 100644 --- a/ext/psych/lib/psych/handler.rb +++ b/ext/psych/lib/psych/handler.rb @@ -25,6 +25,19 @@ module Psych # Default dumping options OPTIONS = DumperOptions.new + # Events that a Handler should respond to. + EVENTS = [ :alias, + :empty, + :end_document, + :end_mapping, + :end_sequence, + :end_stream, + :scalar, + :start_document, + :start_mapping, + :start_sequence, + :start_stream ] + ### # Called with +encoding+ when the YAML stream starts. This method is # called once per stream. A stream may contain multiple documents. diff --git a/ext/psych/lib/psych/handlers/recorder.rb b/ext/psych/lib/psych/handlers/recorder.rb new file mode 100644 index 0000000000..4eae62e5f9 --- /dev/null +++ b/ext/psych/lib/psych/handlers/recorder.rb @@ -0,0 +1,39 @@ +require 'psych/handler' + +module Psych + module Handlers + ### + # This handler will capture an event and record the event. Recorder events + # are available vial Psych::Handlers::Recorder#events. + # + # For example: + # + # recorder = Psych::Handlers::Recorder.new + # parser = Psych::Parser.new recorder + # parser.parse '--- foo' + # + # recorder.events # => [list of events] + # + # # Replay the events + # + # emitter = Psych::Emitter.new $stdout + # recorder.events.each do |m, args| + # emitter.send m, *args + # end + + class Recorder < Psych::Handler + attr_reader :events + + def initialize + @events = [] + super + end + + EVENTS.each do |event| + define_method event do |*args| + @events << [event, args] + end + end + end + end +end diff --git a/test/psych/handlers/test_recorder.rb b/test/psych/handlers/test_recorder.rb new file mode 100644 index 0000000000..96b8eac1fc --- /dev/null +++ b/test/psych/handlers/test_recorder.rb @@ -0,0 +1,25 @@ +require 'psych/helper' +require 'psych/handlers/recorder' + +module Psych + module Handlers + class TestRecorder < TestCase + def test_replay + yaml = "--- foo\n...\n" + output = StringIO.new + + recorder = Psych::Handlers::Recorder.new + parser = Psych::Parser.new recorder + parser.parse yaml + + assert_equal 5, recorder.events.length + + emitter = Psych::Emitter.new output + recorder.events.each do |m, args| + emitter.send m, *args + end + assert_equal yaml, output.string + end + end + end +end -- cgit v1.2.3