lib/vines/stream/state.rb in vines-0.1.0 vs lib/vines/stream/state.rb in vines-0.1.1

@@ -7,12 +7,13 @@
     # nodes and advance to their next valid state or fail the stream.
     class State
       include Nokogiri::XML
       include Vines::Log
 
-      attr_reader :stream
+      attr_accessor :stream
 
+      BODY   = 'body'.freeze
       STREAM = 'stream'.freeze
 
       def initialize(stream, success=nil)
         @stream, @success = stream, success
       end
@@ -39,9 +40,13 @@
         stream.advance(@success.new(stream))
       end
 
       def stream?(node)
         node.name == STREAM && namespace(node) == NAMESPACES[:stream]
+      end
+
+      def body?(node)
+        node.name == BODY && namespace(node) == NAMESPACES[:http_bind]
       end
 
       def namespace(node)
         node.namespace ? node.namespace.href : nil
       end
\ No newline at end of file