test: add util methods

2 files changed, 24 insertions, 0 deletions

diff --git a/test/utils/assertions.rb b/test/utils/assertions.rb
index e11a194..71928ed 100644
--- a/test/utils/assertions.rb
+++ b/test/utils/assertions.rb
@@ -18,6 +18,12 @@ module CustomAssertions
     assert_nil(Plum::ConnectionError.last, "No connection error expected but raised: #{Plum::ConnectionError.last}")
   end
 
+  def assert_frame(frame, **args)
+    args.each do |name, value|
+      assert_equal(value, frame.__send__(name))
+    end
+  end
+
   private
   def assert_http_error(klass, type, &blk)
     klass.reset
diff --git a/test/utils/server.rb b/test/utils/server.rb
index cc17fd5..e3b8386 100644
--- a/test/utils/server.rb
+++ b/test/utils/server.rb
@@ -38,6 +38,24 @@ module ServerUtils
     end
     frames
   end
+
+  def capture_frames(con = nil, &blk)
+    io = (con || @_con).io
+    pos = io.string.bytesize
+    blk.call
+    resp = io.string.byteslice(pos, io.string.bytesize - pos)
+    frames = []
+    while f = Frame.parse!(resp)
+      frames << f
+    end
+    frames
+  end
+
+  def capture_frame(con = nil, &blk)
+    frames = capture_frames(con, &blk)
+    assert_equal(1, frames.size, "Supplied block sent no frames or more than 1 frame")
+    frames.first
+  end
 end
 
 class Minitest::Test