polyphony 0.81 → 0.83

data/test/test_io.rb

@@ -279,6 +279,129 @@ class IOTest < MiniTest::Test
     sleep 0.01
     assert_equal ['foo', 'bar', 'baz'], receiver.buffer
   end
+
+  def test_splice_from
+    i1, o1 = IO.pipe
+    i2, o2 = IO.pipe
+    len = nil
+
+    spin {
+      len = o2.splice_from(i1, 1000)
+      o2.close
+    }
+
+    o1.write('foobar')
+    result = i2.read
+
+    assert_equal 'foobar', result
+    assert_equal 6, len
+  end
+
+  def test_splice_class_method
+    i1, o1 = IO.pipe
+    i2, o2 = IO.pipe
+    len = nil
+
+    spin {
+      len = IO.splice(i1, o2, 1000)
+      o2.close
+    }
+
+    o1.write('foobar')
+    result = i2.read
+
+    assert_equal 'foobar', result
+    assert_equal 6, len
+  end
+
+  def test_splice_to_eof_from
+    i1, o1 = IO.pipe
+    i2, o2 = IO.pipe
+    len = nil
+
+    f = spin {
+      len = o2.splice_to_eof_from(i1, 1000)
+      o2.close
+    }
+
+    o1.write('foo')
+    result = i2.readpartial(1000)
+    assert_equal 'foo', result
+
+    o1.write('bar')
+    result = i2.readpartial(1000)
+    assert_equal 'bar', result
+    o1.close
+    f.await
+    assert_equal 6, len
+  ensure
+    if f.alive?
+      f.interrupt
+      f.await
+    end
+  end
+
+  def test_splice_to_eof_class_method
+    i1, o1 = IO.pipe
+    i2, o2 = IO.pipe
+    len = nil
+
+    f = spin {
+      len = IO.splice_to_eof(i1, o2, 1000)
+      o2.close
+    }
+
+    o1.write('foo')
+    result = i2.readpartial(1000)
+    assert_equal 'foo', result
+
+    o1.write('bar')
+    result = i2.readpartial(1000)
+    assert_equal 'bar', result
+    o1.close
+    f.await
+    assert_equal 6, len
+  ensure
+    if f.alive?
+      f.interrupt
+      f.await
+    end
+  end
+end
+
+class IOWithRawBufferTest < MiniTest::Test
+  def setup
+    super
+    @i, @o = IO.pipe
+  end
+
+  def test_write_with_raw_buffer
+    Polyphony.__with_raw_buffer__(64) do |b|
+      Polyphony.__raw_buffer_set__(b, 'foobar')
+      @o << b
+      @o.close
+    end
+
+    str = @i.read
+    assert_equal 'foobar', str
+  end
+
+  def test_read_with_raw_buffer
+    @o << '*' * 65
+    @o.close
+    chunks = []
+    Polyphony.__with_raw_buffer__(64) do |b|
+      res = @i.read(64, b)
+      assert_equal 64, res
+      chunks << Polyphony.__raw_buffer_get__(b, res)
+
+      res = @i.read(64, b)
+      assert_equal 1, res
+      assert_equal 64, Polyphony.__raw_buffer_size__(b)
+      chunks << Polyphony.__raw_buffer_get__(b, res)
+    end
+    assert_equal ['*' * 64, '*'], chunks
+  end
 end
 
 class IOClassMethodsTest < MiniTest::Test
@@ -468,3 +591,247 @@ class IOClassMethodsTest < MiniTest::Test
     assert_equal ['foo', 'bar'], buf
   end
 end
+
+class IOExtensionsTest < MiniTest::Test
+  def test_deflate
+    i, o = IO.pipe
+    r, w = IO.pipe
+
+    spin {
+      IO.deflate(i, w)
+      w.close
+    }
+
+    o << 'foobar' * 20
+    o.close
+
+    data = r.read
+    msg = Zlib::Inflate.inflate(data)
+    assert_equal 'foobar' * 20, msg
+  end
+
+  def test_inflate
+    i, o = IO.pipe
+    r, w = IO.pipe
+
+    spin {
+      data = Zlib::Deflate.deflate('foobar', 9)
+      o << data
+      o.close
+    }
+
+    IO.inflate(i, w)
+    w.close
+    msg = r.read
+    assert_equal 'foobar', msg
+  end
+
+  def test_gzip
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+    now = nil
+
+    spin {
+      now = Time.now
+      IO.gzip(src, dest)
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    gz = Zlib::GzipReader.new(dest)
+    data = gz.read
+    assert_equal IO.read(__FILE__), data
+    assert_in_range (now-1)..(now+1), gz.mtime
+    assert_nil gz.orig_name
+    assert_nil gz.comment
+  end
+
+  def test_gzip_with_mtime_int
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+
+    spin {
+      IO.gzip(src, dest, mtime: 42)
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    gz = Zlib::GzipReader.new(dest)
+    data = gz.read
+    assert_equal IO.read(__FILE__), data
+    assert_equal Time.at(42), gz.mtime
+  end
+
+  def test_gzip_with_mtime_false
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+
+    spin {
+      IO.gzip(src, dest, mtime: false)
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    gz = Zlib::GzipReader.new(dest)
+    data = gz.read
+    assert_equal IO.read(__FILE__), data
+    assert_equal Time.at(0), gz.mtime
+  end
+
+  def test_gzip_with_mtime_time
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+    t = Time.at(Time.now.to_i) - rand(300000)
+
+    spin {
+      IO.gzip(src, dest, mtime: t)
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    gz = Zlib::GzipReader.new(dest)
+    data = gz.read
+    assert_equal IO.read(__FILE__), data
+    assert_equal t, gz.mtime
+  end
+
+  def test_gzip_with_orig_name
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+
+    spin {
+      IO.gzip(src, dest, orig_name: '/foo/bar')
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    gz = Zlib::GzipReader.new(dest)
+    data = gz.read
+    assert_equal IO.read(__FILE__), data
+    assert_equal '/foo/bar', gz.orig_name
+  end
+
+  def test_gzip_with_comment
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+
+    spin {
+      IO.gzip(src, dest, comment: 'hello!')
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    gz = Zlib::GzipReader.new(dest)
+    data = gz.read
+    assert_equal IO.read(__FILE__), data
+    assert_equal 'hello!', gz.comment
+  end
+
+  def test_gunzip
+    src = Polyphony.pipe
+    dest = Polyphony.pipe
+
+    spin {
+      IO.gunzip(src, dest)
+      dest.close
+    }
+
+    gz = Zlib::GzipWriter.new(src, 9)
+    gz << 'foobar'#IO.read(__FILE__)
+    gz.close
+
+    data = dest.read
+    # assert_equal IO.read(__FILE__), data
+    assert_equal 'foobar', data
+  end
+
+  def test_gunzip_multi
+    src1 = Polyphony.pipe
+    src2 = Polyphony.pipe
+    dest = Polyphony.pipe
+
+    spin {
+      IO.gunzip(src1, dest)
+      IO.gunzip(src2, dest)
+      dest.close
+    }
+
+    gz1 = Zlib::GzipWriter.new(src1)
+    gz1 << 'foobar'
+    gz1.close
+
+    gz1 = Zlib::GzipWriter.new(src2)
+    gz1 << 'raboof'
+    gz1.close
+
+    data = dest.read
+    assert_equal 'foobarraboof', data
+  end
+
+  def test_gzip_gunzip
+    gzipped = Polyphony.pipe
+    gunzipped = Polyphony.pipe
+
+    spin { File.open(__FILE__, 'r') { |f| IO.gzip(f, gzipped) }; gzipped.close }
+    spin { IO.gunzip(gzipped, gunzipped); gunzipped.close }
+
+    data = gunzipped.read
+    assert_equal IO.read(__FILE__), data
+  end
+
+  def test_gunzip_with_empty_info
+    gzipped = Polyphony.pipe
+    gunzipped = Polyphony.pipe
+    info = {}
+
+    spin {
+      File.open(__FILE__, 'r') { |f| IO.gzip(f, gzipped, mtime: false) }
+      gzipped.close
+    }
+    spin { IO.gunzip(gzipped, gunzipped, info); gunzipped.close }
+
+    data = gunzipped.read
+    assert_equal IO.read(__FILE__), data
+    assert_equal Time.at(0), info[:mtime]
+    assert_nil info[:orig_name]
+    assert_nil info[:comment]
+  end
+
+  def test_gunzip_with_info
+    src = Polyphony.pipe
+    gzipped = Polyphony.pipe
+    gunzipped = Polyphony.pipe
+
+    src_info = {
+      mtime: 42,
+      orig_name: 'foo.bar',
+      comment: 'hello!'
+    }
+
+    dest_info = {}
+
+    spin { IO.gzip(src, gzipped, src_info); gzipped.close }
+    spin { IO.gunzip(gzipped, gunzipped, dest_info); gunzipped.close }
+
+    src << 'foobar'
+    src.close
+
+    data = gunzipped.read
+    assert_equal 'foobar', data
+    assert_equal Time.at(42), dest_info[:mtime]
+    assert_equal 'foo.bar', dest_info[:orig_name]
+    assert_equal 'hello!', dest_info[:comment]
+  end
+end

data/test/test_pipe.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require_relative 'helper'
+
+class PipeTest < MiniTest::Test
+  def test_pipe_creation
+    pipe = Polyphony::Pipe.new
+
+    fds = pipe.fds
+    assert_equal 2, fds.size
+    assert_kind_of Integer, fds[0]
+    assert_kind_of Integer, fds[1]
+    assert_equal false, pipe.closed?
+  end
+
+  def test_polyphony_pipe_method
+    pipe = Polyphony.pipe
+
+    fds = pipe.fds
+    assert_equal 2, fds.size
+    assert_kind_of Integer, fds[0]
+    assert_kind_of Integer, fds[1]
+    assert_equal false, pipe.closed?
+  end
+
+  def test_pipe_splice
+    src = Polyphony::Pipe.new
+    dest = Polyphony::Pipe.new
+
+    spin {
+      IO.splice(src, dest, 8192)
+      dest.close
+    }
+
+    src << IO.read(__FILE__)
+    src.close
+
+    data = dest.read
+    assert_equal IO.read(__FILE__), data
+  end
+end

data/test/test_process_supervision.rb

@@ -65,7 +65,7 @@ class ProcessSupervisionTest < MiniTest::Test
 
     supervisor = spin { supervise(watcher) }
 
-    sleep 0.1
+    sleep 0.2
     supervisor.terminate
     supervisor.await
 

data/test/test_raw_buffer.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require_relative 'helper'
+require 'msgpack'
+
+class RawBufferTest < MiniTest::Test
+  def test_with_raw_buffer
+    result = Polyphony.__with_raw_buffer__(64) do |b|
+      assert_kind_of Integer, b
+      assert_equal 64, Polyphony.__raw_buffer_size__(b)
+      :foo
+    end
+    assert_equal :foo, result
+  end
+
+  def test_raw_buffer_get_set
+    Polyphony.__with_raw_buffer__(64) do |b|
+      # should raise if buffer not big enough
+      assert_raises { Polyphony.__raw_buffer_set__(b, '*' * 65) }
+
+      Polyphony.__raw_buffer_set__(b, 'foobar')
+      assert_equal 6, Polyphony.__raw_buffer_size__(b)
+
+      str = Polyphony.__raw_buffer_get__(b)
+      assert_equal 'foobar', str
+
+      str = Polyphony.__raw_buffer_get__(b, 3)
+      assert_equal 'foo', str
+
+      Polyphony.__raw_buffer_set__(b, '')
+      assert_equal 0, Polyphony.__raw_buffer_size__(b)
+
+      str = Polyphony.__raw_buffer_get__(b)
+      assert_equal '', str
+    end
+  end
+end

data/test/test_socket.rb

@@ -193,6 +193,56 @@ class SocketTest < MiniTest::Test
   end
 end
 
+class SocketWithRawBufferTest < MiniTest::Test
+  def start_tcp_server_on_random_port(host = '127.0.0.1')
+    port = rand(1100..60000)
+    server = TCPServer.new(host, port)
+    [port, server]
+  rescue Errno::EADDRINUSE
+    retry
+  end  
+
+  def setup
+    super
+
+    port, server = start_tcp_server_on_random_port
+    connector = spin { @o = TCPSocket.new('127.0.0.1', port) }
+    @i = server.accept
+    connector.await
+  end
+
+  def test_send_with_raw_buffer
+    Polyphony.__with_raw_buffer__(64) do |b|
+      Polyphony.__raw_buffer_set__(b, 'foobar')
+      @o << b
+      @o.close
+    end
+
+    str = @i.read
+    assert_equal 'foobar', str
+  end
+
+  def test_recv_with_raw_buffer
+    @o << '*' * 65
+    @o.close
+    chunks = []
+    Polyphony.__with_raw_buffer__(64) do |b|
+      res = @i.recv(64, 0, b)
+      assert_equal 64, res
+      chunks << Polyphony.__raw_buffer_get__(b, res)
+
+      res = @i.recv(64, 0, b)
+      assert_equal 1, res
+      assert_equal 64, Polyphony.__raw_buffer_size__(b)
+      chunks << Polyphony.__raw_buffer_get__(b, res)
+
+      res = @i.recv(64, 0, b)
+      assert_nil res
+    end
+    assert_equal ['*' * 64, '*'], chunks
+  end
+end
+
 if IS_LINUX
   class HTTPClientTest < MiniTest::Test
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polyphony
 version: !ruby/object:Gem::Version
-  version: '0.81'
+  version: '0.83'
 platform: ruby
 authors:
 - Sharon Rosner
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-03-03 00:00:00.000000000 Z
+date: 2022-03-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-compiler
@@ -232,6 +232,7 @@ files:
 - examples/core/nested.rb
 - examples/core/pingpong.rb
 - examples/core/queue.rb
+- examples/core/raw_buffer_test.rb
 - examples/core/recurrent-timer.rb
 - examples/core/resource_delegate.rb
 - examples/core/ring.rb
@@ -333,9 +334,11 @@ files:
 - ext/polyphony/event.c
 - ext/polyphony/extconf.rb
 - ext/polyphony/fiber.c
+- ext/polyphony/io_extensions.c
 - ext/polyphony/libev.c
 - ext/polyphony/libev.h
 - ext/polyphony/liburing.c
+- ext/polyphony/pipe.c
 - ext/polyphony/playground.c
 - ext/polyphony/polyphony.c
 - ext/polyphony/polyphony.h
@@ -349,6 +352,7 @@ files:
 - ext/polyphony/runqueue_ring_buffer.h
 - ext/polyphony/socket_extensions.c
 - ext/polyphony/thread.c
+- ext/polyphony/zlib_conf.rb
 - ext/test_eintr.c
 - lib/polyphony.rb
 - lib/polyphony/adapters/fs.rb
@@ -375,6 +379,7 @@ files:
 - lib/polyphony/extensions/kernel.rb
 - lib/polyphony/extensions/object.rb
 - lib/polyphony/extensions/openssl.rb
+- lib/polyphony/extensions/pipe.rb
 - lib/polyphony/extensions/process.rb
 - lib/polyphony/extensions/socket.rb
 - lib/polyphony/extensions/thread.rb
@@ -396,8 +401,10 @@ files:
 - test/test_global_api.rb
 - test/test_io.rb
 - test/test_kernel.rb
+- test/test_pipe.rb
 - test/test_process_supervision.rb
 - test/test_queue.rb
+- test/test_raw_buffer.rb
 - test/test_resource_pool.rb
 - test/test_signal.rb
 - test/test_socket.rb