biryani 0.0.1 → 0.0.2

data/lib/biryani/streams_context.rb

@@ -5,11 +5,13 @@ module Biryani
     end
 
     # @param stream_id [Integer]
+    # @param send_initial_window_size [Integer]
+    # @param recv_initial_window_size [Integer]
     # @param proc [Proc]
     #
     # @return [StreamContext]
-    def new_context(stream_id, proc)
-      ctx = StreamContext.new(stream_id, proc)
+    def new_context(stream_id, send_initial_window_size, recv_initial_window_size, proc)
+      ctx = StreamContext.new(stream_id, send_initial_window_size, recv_initial_window_size, proc)
       @h[stream_id] = ctx
       ctx
     end
@@ -54,18 +56,71 @@ module Biryani
     def last_stream_id
       @h.reject { |_, ctx| ctx.idle? }.keys.max || 0
     end
+
+    # @param stream_id [Integer]
+    # @param data [String]
+    # @param send_window [Window]
+    # @param max_frame_size [Integer]
+    #
+    # @return [Array<Object>] frames
+    # @return [String]
+    def sendable_data_frames(stream_id, data, send_window, max_frame_size)
+      len = [data.bytesize, send_window.length, @h[stream_id].send_window.length].min
+
+      payload = data[0...len]
+      remains = data[len..] || ''
+
+      len = (len + max_frame_size - 1) / max_frame_size
+      frames = payload.gsub(/.{1,#{max_frame_size}}/m).with_index.map do |s, index|
+        end_stream = remains.empty? && index == len - 1
+        Frame::Data.new(end_stream, stream_id, s, nil)
+      end
+
+      [frames, remains]
+    end
+
+    # @param data_buffer [DataBuffer]
+    def remove_closed(data_buffer)
+      closed_ids = closed_stream_ids.filter { |id| !data_buffer.has?(id) }
+      closed_ids.each do |id|
+        @h[id].tx.close
+        @h[id].stream.rx << nil
+        @h[id].fragment.close
+        @h[id].content.close
+      end
+    end
+
+    def close_all
+      each do |ctx|
+        ctx.tx.close
+        ctx.fragment.close
+        ctx.content.close
+        ctx.state.close
+      end
+    end
+
+    def clear_all
+      each do |ctx|
+        ctx.tx.close
+        ctx.stream.rx << nil
+        ctx.fragment.close
+        ctx.content.close
+      end
+    end
   end
 
   class StreamContext
     attr_accessor :stream, :tx, :send_window, :recv_window, :fragment, :content, :state
 
     # @param stream_id [Integer]
+    # @param send_initial_window_size [Integer]
+    # @param recv_initial_window_size [Integer]
     # @param proc [Proc]
-    def initialize(stream_id, proc)
+    def initialize(stream_id, send_initial_window_size, recv_initial_window_size, proc)
       @tx = Ractor::Port.new
       @stream = Stream.new(@tx, stream_id, proc)
-      @send_window = Window.new
-      @recv_window = Window.new
+      @send_window = Window.new(send_initial_window_size)
+      @recv_window = Window.new(recv_initial_window_size)
       @fragment = StringIO.new
       @content = StringIO.new
       @state = State.new

data/lib/biryani/utils.rb

@@ -0,0 +1,8 @@
+module Biryani
+  # @param obj [Object]
+  #
+  # @return [Boolean]
+  def self.err?(obj)
+    obj.is_a?(StreamError) || obj.is_a?(ConnectionError)
+  end
+end

data/lib/biryani/version.rb

@@ -1,3 +1,3 @@
 module Biryani
-  VERSION = '0.0.1'.freeze
+  VERSION = '0.0.2'.freeze
 end

data/lib/biryani/window.rb

@@ -1,29 +1,27 @@
 module Biryani
   class Window
-    def initialize
-      @window = 2**16 - 1
-    end
+    attr_reader :length
 
-    # @param length [Integer]
-    #
-    # @return [Boolean]
-    def available?(length)
-      @window > length
+    # @param initial_window_size [Integer]
+    def initialize(initial_window_size)
+      @length = initial_window_size
+      @capacity = initial_window_size
     end
 
     # @param length [Integer]
     def consume!(length)
-      @window -= length
+      @length -= length
     end
 
     # @param length [Integer]
     def increase!(length)
-      @window += length
+      @length += length
     end
 
-    # @return [Integer]
-    def length
-      @window
+    # @param initial_window_size [Integer]
+    def update!(initial_window_size)
+      @length = initial_window_size - @capacity + @length
+      @capacity = initial_window_size
     end
   end
 end

data/lib/biryani.rb

@@ -1,8 +1,8 @@
 require 'stringio'
 require 'uri'
 
-require_relative 'biryani/connection'
 require_relative 'biryani/connection_error'
+require_relative 'biryani/connection'
 require_relative 'biryani/data_buffer'
 require_relative 'biryani/frame'
 require_relative 'biryani/hpack'
@@ -10,8 +10,9 @@ require_relative 'biryani/http_request'
 require_relative 'biryani/http_response'
 require_relative 'biryani/server'
 require_relative 'biryani/state'
-require_relative 'biryani/streams_context'
 require_relative 'biryani/stream_error'
 require_relative 'biryani/stream'
+require_relative 'biryani/streams_context'
+require_relative 'biryani/utils'
 require_relative 'biryani/version'
 require_relative 'biryani/window'

data/spec/connection/handle_connection_window_update_spec.rb

@@ -6,7 +6,7 @@ RSpec.describe Connection do
       Frame::WindowUpdate.new(0, 1000)
     end
     let(:send_window) do
-      Window.new
+      Window.new(65_535)
     end
     it 'should handle' do
       expect { Connection.handle_connection_window_update(window_update, send_window) }.not_to raise_error

data/spec/connection/handle_rst_stream_spec.rb

@@ -7,8 +7,8 @@ RSpec.describe Connection do
     end
     let(:streams_ctx) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should handle' do

data/spec/connection/handle_settings_spec.rb

@@ -8,12 +8,18 @@ RSpec.describe Connection do
     let(:decoder) do
       HPACK::Decoder.new(4_096)
     end
+    let(:streams_ctx) do
+      streams_ctx = StreamsContext.new
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx
+    end
 
     let(:settings1) do
       Frame::Settings.new(false, 0, { SettingsID::SETTINGS_HEADER_TABLE_SIZE => 8_192 })
     end
     it 'should handle' do
-      reply_settings = Connection.handle_settings(settings1, send_settings, decoder)
+      reply_settings = Connection.handle_settings(settings1, send_settings, decoder, streams_ctx)
       expect(reply_settings.ack?).to eq true
       expect(reply_settings.setting.empty?).to eq true
       expect(send_settings[SettingsID::SETTINGS_MAX_CONCURRENT_STREAMS]).to eq 0xffffffff
@@ -25,7 +31,7 @@ RSpec.describe Connection do
       Frame::Settings.new(true, 0, {})
     end
     it 'should handle' do
-      expect(Connection.handle_settings(settings2, send_settings, decoder)).to eq nil
+      expect(Connection.handle_settings(settings2, send_settings, decoder, streams_ctx)).to eq nil
     end
   end
 end

data/spec/connection/handle_stream_window_update_spec.rb

@@ -7,8 +7,8 @@ RSpec.describe Connection do
     end
     let(:streams_ctx) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should handle' do

data/spec/connection/send_spec.rb

@@ -10,29 +10,25 @@ RSpec.describe Connection do
     end
 
     let(:headers1) do
-      Frame::Headers.new(true, false, 2, nil, nil, 'this is dummy', nil)
-    end
-    let(:data1) do
-      Frame::Data.new(true, 2, 'Hello, world!', 'Howdy!')
+      Frame::Headers.new(true, true, 2, nil, nil, 'this is dummy'.b, nil)
     end
     let(:send_window1) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx1) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should send' do
       streams_ctx1[2].state.transition!(headers1, :recv)
-      streams_ctx1[2].state.transition!(data1, :recv)
-      Connection.send(io, headers1, send_window1, streams_ctx1, data_buffer)
-      Connection.send(io, data1, send_window1, streams_ctx1, data_buffer)
-      expect(io.string).to eq "\x00\x00\x0d\x01\x04\x00\x00\x00\x02this is dummy\x00\x00\x14\x00\x09\x00\x00\x00\x02\x06\x48\x65\x6c\x6c\x6f\x2c\x20\x77\x6f\x72\x6c\x64\x21\x48\x6f\x77\x64\x79\x21".b
-      expect(send_window1.length).to eq 2**16 - 21
-      expect(streams_ctx1[1].send_window.length).to eq 2**16 - 1
-      expect(streams_ctx1[2].send_window.length).to eq 2**16 - 21
+      Connection.send_headers(io, 2, "\x88".b, false, 16_384, streams_ctx1)
+      Connection.send_data(io, 2, 'Hello, world!'.b, send_window1, 16_384, streams_ctx1, data_buffer)
+      expect(io.string.force_encoding(Encoding::ASCII_8BIT)).to eq "\x00\x00\x01\x01\x04\x00\x00\x00\x02\x88\x00\x00\x0d\x00\x01\x00\x00\x00\x02Hello, world!".b
+      expect(send_window1.length).to eq 65_535 - 13
+      expect(streams_ctx1[1].send_window.length).to eq 65_535
+      expect(streams_ctx1[2].send_window.length).to eq 65_535 - 13
       expect(data_buffer.length).to eq 0
     end
 
@@ -40,46 +36,43 @@ RSpec.describe Connection do
       Frame::Headers.new(true, true, 1, nil, nil, 'this is dummy', nil)
     end
     let(:send_window2) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx2) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should send' do
       streams_ctx2[1].state.transition!(headers2, :recv)
-      Connection.send(io, headers2, send_window2, streams_ctx2, data_buffer)
-      expect(io.string).to eq "\x00\x00\x0d\x01\x05\x00\x00\x00\x01\x74\x68\x69\x73\x20\x69\x73\x20\x64\x75\x6d\x6d\x79".b
-      expect(send_window2.length).to eq 2**16 - 1
-      expect(streams_ctx2[1].send_window.length).to eq 2**16 - 1
-      expect(streams_ctx2[2].send_window.length).to eq 2**16 - 1
+      Connection.send_headers(io, 1, "\x88".b, true, 16_384, streams_ctx2)
+      expect(io.string.force_encoding(Encoding::ASCII_8BIT)).to eq "\x00\x00\x01\x01\x05\x00\x00\x00\x01\x88".b
+      expect(send_window2.length).to eq 65_535
+      expect(streams_ctx2[1].send_window.length).to eq 65_535
+      expect(streams_ctx2[2].send_window.length).to eq 65_535
       expect(data_buffer.length).to eq 0
     end
 
     let(:headers3) do
       Frame::Headers.new(true, true, 2, nil, nil, 'this is dummy', nil)
     end
-    let(:data3) do
-      Frame::Data.new(false, 2, 'Hello, world!', 'Howdy!')
-    end
     let(:send_window3) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx3) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
-      streams_ctx[2].send_window.consume!(2**16 - 1)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx[2].send_window.consume!(65_535)
       streams_ctx
     end
     it 'should send' do
       streams_ctx3[2].state.transition!(headers3, :recv)
-      Connection.send(io, data3, send_window3, streams_ctx3, data_buffer)
-      expect(io.string).to eq ''
-      expect(send_window3.length).to eq 2**16 - 1
-      expect(streams_ctx3[1].send_window.length).to eq 2**16 - 1
+      Connection.send_data(io, 2, 'Hello, world!'.b, send_window3, 16_384, streams_ctx3, data_buffer)
+      expect(io.string.force_encoding(Encoding::ASCII_8BIT)).to eq ''.b
+      expect(send_window3.length).to eq 65_535
+      expect(streams_ctx3[1].send_window.length).to eq 65_535
       expect(streams_ctx3[2].send_window.length).to eq 0
       expect(data_buffer.length).to eq 1
     end
@@ -87,27 +80,24 @@ RSpec.describe Connection do
     let(:headers4) do
       Frame::Headers.new(true, true, 2, nil, nil, 'this is dummy', nil)
     end
-    let(:data4) do
-      Frame::Data.new(false, 2, 'Hello, world!', 'Howdy!')
-    end
     let(:send_window4) do
-      send_window = Window.new
-      send_window.consume!(2**16 - 1)
+      send_window = Window.new(65_535)
+      send_window.consume!(65_535)
       send_window
     end
     let(:streams_ctx4) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should send' do
       streams_ctx4[2].state.transition!(headers4, :recv)
-      Connection.send(io, data4, send_window4, streams_ctx4, data_buffer)
-      expect(io.string).to eq ''
+      Connection.send_data(io, 2, 'Hello, world!'.b, send_window4, 16_384, streams_ctx4, data_buffer)
+      expect(io.string.force_encoding(Encoding::ASCII_8BIT)).to eq ''.b
       expect(send_window4.length).to eq 0
-      expect(streams_ctx4[1].send_window.length).to eq 2**16 - 1
-      expect(streams_ctx4[2].send_window.length).to eq 2**16 - 1
+      expect(streams_ctx4[1].send_window.length).to eq 65_535
+      expect(streams_ctx4[2].send_window.length).to eq 65_535
       expect(data_buffer.length).to eq 1
     end
   end

data/spec/connection/transition_state_send_spec.rb

@@ -4,8 +4,8 @@ RSpec.describe Connection do
   context 'transition_state_send' do
     let(:streams_ctx) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
 
@@ -21,7 +21,7 @@ RSpec.describe Connection do
     it 'should transition' do
       streams_ctx[1].state.transition!(headers, :recv)
       streams_ctx[2].state.transition!(headers, :recv)
-      Connection.close_all_streams(streams_ctx)
+      streams_ctx.close_all
       expect { streams_ctx[1].tx << nil }.to raise_error Ractor::ClosedError
       expect { streams_ctx[2].tx << nil }.to raise_error Ractor::ClosedError
     end

data/spec/data_buffer_spec.rb

@@ -6,125 +6,125 @@ RSpec.describe DataBuffer do
       DataBuffer.new
     end
     let(:send_window1) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx1) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should take' do
-      expect(data_buffer1.take!(send_window1, streams_ctx1).length).to eq 0
-      expect(send_window1.length).to eq 2**16 - 1
-      expect(streams_ctx1[1].send_window.length).to eq 2**16 - 1
-      expect(streams_ctx1[2].send_window.length).to eq 2**16 - 1
+      expect(data_buffer1.take!(send_window1, streams_ctx1, 16_384).length).to eq 0
+      expect(send_window1.length).to eq 65_535
+      expect(streams_ctx1[1].send_window.length).to eq 65_535
+      expect(streams_ctx1[2].send_window.length).to eq 65_535
     end
 
     let(:data_buffer2) do
       data_buffer = DataBuffer.new
-      data_buffer << Frame::Data.new(false, 1, 'one', nil)
-      data_buffer << Frame::Data.new(false, 2, 'two', nil)
+      data_buffer.store(1, 'one')
+      data_buffer.store(2, 'two')
       data_buffer
     end
     let(:send_window2) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx2) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should take' do
-      datas = data_buffer2.take!(send_window2, streams_ctx2)
+      datas = data_buffer2.take!(send_window2, streams_ctx2, 16_384)
       expect(datas.length).to eq 2
       expect(datas.map(&:stream_id)).to eq [1, 2]
       expect(datas.map(&:data)).to eq %w[one two]
-      expect(send_window2.length).to eq 2**16 - 7
-      expect(streams_ctx2[1].send_window.length).to eq 2**16 - 4
-      expect(streams_ctx2[2].send_window.length).to eq 2**16 - 4
+      expect(send_window2.length).to eq 65_535 - 6
+      expect(streams_ctx2[1].send_window.length).to eq 65_535 - 3
+      expect(streams_ctx2[2].send_window.length).to eq 65_535 - 3
     end
 
     let(:data_buffer3) do
       data_buffer = DataBuffer.new
-      data_buffer << Frame::Data.new(false, 2, 'two', nil)
+      data_buffer.store(2, 'two')
       data_buffer
     end
     let(:send_window3) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx3) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should take' do
-      datas = data_buffer3.take!(send_window3, streams_ctx3)
+      datas = data_buffer3.take!(send_window3, streams_ctx3, 16_384)
       expect(datas.length).to eq 1
       expect(datas.first.stream_id).to eq 2
       expect(datas.first.data).to eq 'two'
-      expect(send_window3.length).to eq 2**16 - 4
-      expect(streams_ctx3[1].send_window.length).to eq 2**16 - 1
-      expect(streams_ctx3[2].send_window.length).to eq 2**16 - 4
+      expect(send_window3.length).to eq 65_535 - 3
+      expect(streams_ctx3[1].send_window.length).to eq 65_535
+      expect(streams_ctx3[2].send_window.length).to eq 65_535 - 3
     end
 
     let(:data_buffer4) do
       data_buffer = DataBuffer.new
-      data_buffer << Frame::Data.new(false, 1, 'one', nil)
-      data_buffer << Frame::Data.new(false, 2, 'two', nil)
+      data_buffer.store(1, 'one')
+      data_buffer.store(2, 'two')
       data_buffer
     end
     let(:send_window4) do
-      Window.new
+      Window.new(65_535)
     end
     let(:streams_ctx4) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx[1].send_window.consume!(2**16 - 1)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx[1].send_window.consume!(65_535)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should take' do
-      datas = data_buffer4.take!(send_window4, streams_ctx4)
+      datas = data_buffer4.take!(send_window4, streams_ctx4, 16_384)
       expect(datas.length).to eq 1
       expect(datas.first.stream_id).to eq 2
       expect(datas.first.data).to eq 'two'
-      expect(send_window4.length).to eq 2**16 - 4
+      expect(send_window4.length).to eq 65_535 - 3
       expect(streams_ctx4[1].send_window.length).to eq 0
-      expect(streams_ctx4[2].send_window.length).to eq 2**16 - 4
+      expect(streams_ctx4[2].send_window.length).to eq 65_535 - 3
     end
 
     let(:data_buffer5) do
       data_buffer = DataBuffer.new
-      data_buffer << Frame::Data.new(false, 1, 'one', nil)
-      data_buffer << Frame::Data.new(false, 2, 'two', nil)
+      data_buffer.store(1, 'one')
+      data_buffer.store(2, 'two')
       data_buffer
     end
     let(:send_window5) do
-      send_window = Window.new
-      send_window.consume!(2**16 - 1)
+      send_window = Window.new(65_535)
+      send_window.consume!(65_535)
       send_window
     end
     let(:streams_ctx5) do
       streams_ctx = StreamsContext.new
-      streams_ctx.new_context(1, do_nothing_proc)
-      streams_ctx.new_context(2, do_nothing_proc)
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
       streams_ctx
     end
     it 'should take' do
-      expect(data_buffer5.take!(send_window5, streams_ctx5).length).to eq 0
+      expect(data_buffer5.take!(send_window5, streams_ctx5, 16_384).length).to eq 0
       expect(send_window5.length).to eq 0
-      expect(streams_ctx5[1].send_window.length).to eq 2**16 - 1
-      expect(streams_ctx5[2].send_window.length).to eq 2**16 - 1
+      expect(streams_ctx5[1].send_window.length).to eq 65_535
+      expect(streams_ctx5[2].send_window.length).to eq 65_535
     end
   end
 
   context 'has?' do
     let(:data_buffer6) do
       data_buffer = DataBuffer.new
-      data_buffer << Frame::Data.new(false, 1, 'one', nil)
+      data_buffer.store(1, 'one')
       data_buffer
     end
     it 'should take' do

data/spec/streams_context_spec.rb

@@ -0,0 +1,79 @@
+require_relative 'spec_helper'
+
+RSpec.describe StreamsContext do
+  context 'close_all' do
+    let(:streams_ctx) do
+      streams_ctx = StreamsContext.new
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx
+    end
+    it 'should close' do
+      streams_ctx.close_all
+      expect { streams_ctx[1].tx << nil }.to raise_error Ractor::ClosedError
+      expect { streams_ctx[2].tx << nil }.to raise_error Ractor::ClosedError
+    end
+  end
+
+  context 'close_all' do
+    let(:streams_ctx) do
+      streams_ctx = StreamsContext.new
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx
+    end
+    it 'should close' do
+      streams_ctx.close_all
+      expect { streams_ctx[1].tx << nil }.to raise_error Ractor::ClosedError
+      expect { streams_ctx[2].tx << nil }.to raise_error Ractor::ClosedError
+    end
+  end
+
+  context 'remove_closed' do
+    let(:streams_ctx1) do
+      streams_ctx = StreamsContext.new
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx
+    end
+    let(:data_buffer1) do
+      DataBuffer.new
+    end
+    it 'should remove' do
+      streams_ctx1.remove_closed(data_buffer1)
+      expect(streams_ctx1.length).to eq 2
+    end
+
+    let(:streams_ctx2) do
+      streams_ctx = StreamsContext.new
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx[2].state.close
+      streams_ctx
+    end
+    let(:data_buffer2) do
+      DataBuffer.new
+    end
+    it 'should remove' do
+      streams_ctx2.remove_closed(data_buffer2)
+      expect(streams_ctx2.length).to eq 2 # remain stream_id
+    end
+
+    let(:streams_ctx3) do
+      streams_ctx = StreamsContext.new
+      streams_ctx.new_context(1, 65_535, 65_535, do_nothing_proc)
+      streams_ctx.new_context(2, 65_535, 65_535, do_nothing_proc)
+      streams_ctx[2].state.close
+      streams_ctx
+    end
+    let(:data_buffer3) do
+      data_buffer = DataBuffer.new
+      data_buffer.store(2, 'two')
+      data_buffer
+    end
+    it 'should remove' do
+      streams_ctx3.remove_closed(data_buffer3)
+      expect(streams_ctx3.length).to eq 2
+    end
+  end
+end