add reset! method to reset a parser's buffer and zlib

igrigorik · Jan 22, 2012 · 6773e2b · 6773e2b
1 parent 98f44ca
commit 6773e2b
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/lib/spdy/parser.rb b/lib/spdy/parser.rb
@@ -38,6 +38,11 @@ def on_reset(&blk)
       @on_reset = blk
     end
 
+    def reset!
+      @buffer = ''
+      @zlib_session.reset
+    end
+
     private
 
       def handle_headers(pckt)

diff --git a/spec/parser_spec.rb b/spec/parser_spec.rb
@@ -193,4 +193,29 @@
 
   end
 
+  context "#reset!" do
+    # because the zlib session is stateful
+    it "should raise when trying to inflate the same thing twice" do
+      s << SYN_STREAM
+      lambda {
+        s << SYN_STREAM
+      }.should raise_error
+    end
+
+    it "should not raise when resetting the parser after each frame" do
+      s << SYN_STREAM
+      s.reset!
+      lambda {
+        s << SYN_STREAM
+      }.should_not raise_error
+    end
+
+    it "should clear the buffer" do
+      s << SYN_STREAM[0..30]
+      s.reset!
+      lambda {
+        s << SYN_STREAM
+      }.should_not raise_error
+    end
+  end
 end