polyphony 0.47.5.1 → 0.50.0

data/test/test_fiber.rb

@@ -349,6 +349,22 @@ class FiberTest < MiniTest::Test
     assert_equal [:foo, :terminate], buffer
   end
 
+  CMD_TERMINATE_MAIN_FIBER = <<~BASH
+    ruby -rbundler/setup -rpolyphony -e"spin { sleep 0.1; Thread.current.main_fiber.terminate }; begin; sleep; rescue Polyphony::Terminate; STDOUT << 'terminated'; end" 2>&1
+  BASH
+
+  CMD_TERMINATE_CHILD_FIBER = <<~BASH
+    ruby -rbundler/setup -rpolyphony -e"f = spin { sleep }; spin { sleep 0.1; f.terminate }; f.await" 2>&1
+  BASH
+
+  def test_terminate_main_fiber
+    output = `#{CMD_TERMINATE_CHILD_FIBER}`
+    assert_equal '', output
+
+    output = `#{CMD_TERMINATE_MAIN_FIBER}`
+    assert_equal 'terminated', output
+  end
+
   def test_interrupt_timer
     result = []
     f = Fiber.current.spin do
@@ -639,7 +655,6 @@ class FiberTest < MiniTest::Test
         i.close
         f.await
       rescue Exception => e
-        trace e
         o << e.class.name
         o.close
       end
@@ -1038,3 +1053,82 @@ class RestartTest < MiniTest::Test
     assert_equal [f, 'foo', 'bar', :done, f2, 'baz', 42, :done], buffer
   end
 end
+
+class ChildrenTerminationTest < MiniTest::Test
+  def test_shutdown_all_children
+    f = spin do
+      1000.times { spin { suspend } }
+      suspend
+    end
+
+    snooze
+    assert_equal 1000, f.children.size
+
+    f.shutdown_all_children
+    assert_equal 0, f.children.size
+  end
+end
+
+class GracefulTerminationTest < MiniTest::Test
+  def test_graceful_termination
+    buffer = []
+    f = spin do
+      buffer << 1
+      snooze
+      buffer << 2
+      sleep 3
+      buffer << 3
+    ensure
+      buffer << 4 if Fiber.current.graceful_shutdown?
+    end
+
+    3.times { snooze }
+    f.terminate(false)
+    f.await
+    assert_equal [1, 2], buffer
+
+    buffer = []
+    f = spin do
+      buffer << 1
+      snooze
+      buffer << 2
+      sleep 3
+      buffer << 3
+    ensure
+      buffer << 4 if Fiber.current.graceful_shutdown?
+    end
+
+    3.times { snooze }
+    f.terminate(true)
+    f.await
+    assert_equal [1, 2, 4], buffer
+  end
+
+  def test_graceful_child_shutdown
+    buffer = []
+    f0 = spin do
+      f1 = spin do
+        sleep
+      ensure
+        buffer << 1 if Fiber.current.graceful_shutdown?
+      end
+
+      f2 = spin do
+        sleep
+      ensure
+        buffer << 2 if Fiber.current.graceful_shutdown?
+      end
+
+      sleep
+    ensure
+      Fiber.current.terminate_all_children(true) if Fiber.current.graceful_shutdown?
+      Fiber.current.await_all_children
+    end
+
+    3.times { snooze }
+    f0.terminate(true)
+    f0.await
+
+    assert_equal [1, 2], buffer
+  end
+end

data/test/test_global_api.rb

@@ -307,6 +307,36 @@ class SpinLoopTest < MiniTest::Test
     f.stop
     assert_in_range 1..3, counter
   end
+
+  def test_spin_loop_break
+    i = 0
+    f = spin_loop do
+      i += 1
+      snooze
+      break if i >= 5
+    end
+    f.await
+    assert_equal 5, i
+
+    i = 0
+    f = spin_loop do
+      i += 1
+      snooze
+      raise StopIteration if i >= 5
+    end
+    f.await
+    assert_equal 5, i
+  end
+
+  def test_throttled_spin_loop_break
+    i = 0
+    f = spin_loop(rate: 100) do
+      i += 1
+      break if i >= 5
+    end
+    f.await
+    assert_equal 5, i
+  end
 end
 
 class SpinScopeTest < MiniTest::Test

data/test/test_io.rb

@@ -92,6 +92,32 @@ class IOTest < MiniTest::Test
     assert_raises(EOFError) { i.readpartial(1) }    
   end
 
+  def test_gets
+    i, o = IO.pipe
+
+    buf = []
+    f = spin do
+      while (l = i.gets)
+        buf << l
+      end
+    end
+
+    snooze
+    assert_equal [], buf
+
+    o << 'fab'
+    snooze
+    assert_equal [], buf
+    
+    o << "ulous\n"
+    10.times { snooze }
+    assert_equal ["fabulous\n"], buf
+
+    o.close
+    f.await
+    assert_equal ["fabulous\n"], buf
+  end
+
   def test_getc
     i, o = IO.pipe
 

data/test/test_signal.rb

@@ -46,8 +46,7 @@ class SignalTrapTest < Minitest::Test
       end.await
     rescue Interrupt
       o.puts "3 - interrupted"
-      Fiber.current.terminate_all_children
-      Fiber.current.await_all_children
+      Fiber.current.shutdown_all_children
     ensure
       o.close
     end

data/test/test_socket.rb

@@ -60,18 +60,18 @@ class SocketTest < MiniTest::Test
 end
 
 class HTTPClientTest < MiniTest::Test
-  require 'httparty'
   require 'json'
 
   def test_http
-    res = HTTParty.get('http://worldtimeapi.org/api/timezone/Europe/Paris')
+    res = HTTParty.get('http://ipinfo.io/')
+
     response = JSON.load(res.body)
-    assert_equal "CET", response['abbreviation']
+    assert_equal 'https://ipinfo.io/missingauth', response['readme']
   end
 
   def test_https
-    res = HTTParty.get('https://worldtimeapi.org/api/timezone/Europe/Paris')
+    res = HTTParty.get('https://ipinfo.io/')
     response = JSON.load(res.body)
-    assert_equal "CET", response['abbreviation']
+    assert_equal 'https://ipinfo.io/missingauth', response['readme']
   end
 end

data/test/test_supervise.rb

@@ -20,7 +20,7 @@ class SuperviseTest < MiniTest::Test
 
     f2 << 'bar'
     f2.await
-    assert_equal :waiting, p.state
+    assert_equal :runnable, p.state
     snooze
 
     assert_equal :dead, p.state

data/test/test_timer.rb

@@ -0,0 +1,157 @@
+# frozen_string_literal: true
+
+require_relative 'helper'
+
+class TimerMoveOnAfterTest < MiniTest::Test
+  def setup
+    @timer = Polyphony::Timer.new(resolution: 0.01)
+  end
+
+  def teardown
+    @timer.stop
+  end
+
+  def test_timer_move_on_after
+    t0 = Time.now
+    v = @timer.move_on_after(0.1) do
+      sleep 1
+      :foo
+    end
+    t1 = Time.now
+
+    assert_in_range 0.1..0.15, t1 - t0
+    assert_nil v
+  end
+
+  def test_timer_move_on_after_with_value
+    t0 = Time.now
+    v = @timer.move_on_after(0.01, with_value: :bar) do
+      sleep 1
+      :foo
+    end
+    t1 = Time.now
+
+    assert_in_range 0.01..0.025, t1 - t0
+    assert_equal :bar, v
+  end
+
+  def test_timer_move_on_after_with_reset
+    t0 = Time.now
+    v = @timer.move_on_after(0.01, with_value: :moved_on) do
+      sleep 0.007
+      @timer.reset
+      sleep 0.007
+      @timer.reset
+      sleep 0.007
+      nil
+    end
+    t1 = Time.now
+
+    assert_nil v
+    assert_in_range 0.015..0.03, t1 - t0
+  end
+end
+
+class TimerCancelAfterTest < MiniTest::Test
+  def setup
+    @timer = Polyphony::Timer.new(resolution: 0.01)
+  end
+
+  def teardown
+    @timer.stop
+  end
+
+  def test_timer_cancel_after
+    t0 = Time.now
+
+    assert_raises Polyphony::Cancel do
+      @timer.cancel_after(0.01) do
+        sleep 1
+        :foo
+      end
+    end
+    t1 = Time.now
+    assert_in_range 0.01..0.03, t1 - t0
+  end
+
+  def test_timer_cancel_after_with_reset
+    t0 = Time.now
+    @timer.cancel_after(0.01) do
+      sleep 0.007
+      @timer.reset
+      sleep 0.007
+    end
+    t1 = Time.now
+    assert_in_range 0.013..0.024, t1 - t0
+  end
+
+  class CustomException < Exception
+  end
+
+  def test_timer_cancel_after_with_custom_exception
+    assert_raises CustomException do
+      @timer.cancel_after(0.01, with_exception: CustomException) do
+        sleep 1
+        :foo
+      end
+    end
+
+    begin
+      err = nil
+      @timer.cancel_after(0.01, with_exception: [CustomException, 'custom message']) do
+        sleep 1
+        :foo
+      end
+    rescue Exception => err
+    ensure
+      assert_kind_of CustomException, err
+      assert_equal 'custom message', err.message
+    end
+
+
+    begin
+      e = nil
+      @timer.cancel_after(0.01, with_exception: 'foo') do
+        sleep 1
+        :foo
+      end
+    rescue => e
+    ensure
+      assert_kind_of RuntimeError, e
+      assert_equal 'foo', e.message
+    end
+  end
+end
+
+class TimerMiscTest < MiniTest::Test
+  def setup
+    @timer = Polyphony::Timer.new(resolution: 0.001)
+    sleep 0
+  end
+  
+  def teardown
+    @timer.stop
+  end
+    
+  def test_timer_after
+    buffer = []
+    f = @timer.after(0.01) { buffer << 2 }
+    assert_kind_of Fiber, f
+    snooze
+    assert_equal [], buffer
+    sleep 0.1
+    p :post_sleep
+    assert_equal [2], buffer
+  end
+
+  def test_timer_every
+    buffer = []
+    t0 = Time.now
+    f = spin do
+      @timer.every(0.01) { buffer << 1 }
+    end
+    sleep 0.05
+    f.stop
+    assert_in_range 4..6, buffer.size
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polyphony
 version: !ruby/object:Gem::Version
-  version: 0.47.5.1
+  version: 0.50.0
 platform: ruby
 authors:
 - Sharon Rosner
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-11-20 00:00:00.000000000 Z
+date: 2021-01-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-compiler
@@ -269,7 +269,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 0.3.0
 description: 
-email: ciconia@gmail.com
+email: sharon@noteflakes.com
 executables: []
 extensions:
 - ext/polyphony/extconf.rb
@@ -353,6 +353,7 @@ files:
 - examples/core/forking.rb
 - examples/core/handling-signals.rb
 - examples/core/interrupt.rb
+- examples/core/nested.rb
 - examples/core/pingpong.rb
 - examples/core/recurrent-timer.rb
 - examples/core/resource_delegate.rb
@@ -360,6 +361,8 @@ files:
 - examples/core/spin_error_backtrace.rb
 - examples/core/supervise-process.rb
 - examples/core/supervisor.rb
+- examples/core/suspend.rb
+- examples/core/terminate_main_fiber.rb
 - examples/core/thread-sleep.rb
 - examples/core/thread_pool.rb
 - examples/core/throttling.rb
@@ -383,6 +386,7 @@ files:
 - examples/io/raw.rb
 - examples/io/reline.rb
 - examples/io/system.rb
+- examples/io/tcp_proxy.rb
 - examples/io/tcpserver.rb
 - examples/io/tcpsocket.rb
 - examples/io/tunnel.rb
@@ -391,6 +395,8 @@ files:
 - examples/performance/fiber_resume.rb
 - examples/performance/fiber_transfer.rb
 - examples/performance/fs_read.rb
+- examples/performance/line_splitting.rb
+- examples/performance/loop.rb
 - examples/performance/mem-usage.rb
 - examples/performance/messaging.rb
 - examples/performance/multi_snooze.rb
@@ -433,7 +439,6 @@ files:
 - ext/liburing/setup.c
 - ext/liburing/syscall.c
 - ext/liburing/syscall.h
-- ext/polyphony/backend.h
 - ext/polyphony/backend_common.h
 - ext/polyphony/backend_io_uring.c
 - ext/polyphony/backend_io_uring_context.c
@@ -474,6 +479,7 @@ files:
 - lib/polyphony/core/sync.rb
 - lib/polyphony/core/thread_pool.rb
 - lib/polyphony/core/throttler.rb
+- lib/polyphony/core/timer.rb
 - lib/polyphony/extensions/core.rb
 - lib/polyphony/extensions/debug.rb
 - lib/polyphony/extensions/fiber.rb
@@ -508,6 +514,7 @@ files:
 - test/test_thread.rb
 - test/test_thread_pool.rb
 - test/test_throttler.rb
+- test/test_timer.rb
 - test/test_trace.rb
 homepage: https://digital-fabric.github.io/polyphony
 licenses: