CI fixes

Author: cretz

.github/workflows/ci.yml

@@ -73,4 +73,6 @@ jobs:
 
       - name: Lint, compile, test Ruby
         working-directory: ./temporalio
+        # Timeout just in case there's a hanging part in rake
+        timeout-minutes: 20
         run: bundle exec rake TESTOPTS="--verbose"

temporalio/Gemfile

@@ -17,7 +17,7 @@ group :development do
   gem 'rbs', '~> 3.5.3'
   gem 'rb_sys', '~> 0.9.63'
   gem 'rubocop'
-  gem 'sqlite3', '~> 1.4'
+  gem 'sqlite3'
   gem 'steep', '~> 1.7.1'
   gem 'yard'
 end

temporalio/lib/temporalio/internal/worker/workflow_instance/illegal_call_tracer.rb

@@ -33,26 +33,28 @@ def self.frozen_validated_illegal_calls(illegal_calls)
             end.freeze
           end
 
-          # Illegal calls are Hash[String, Hash[Symbol, Bool]]
+          # Illegal calls are Hash[String, :all | Hash[Symbol, Bool]]
           def initialize(illegal_calls)
             @tracepoint = TracePoint.new(:call, :c_call) do |tp|
               cls = tp.defined_class
               next unless cls.is_a?(Module)
 
-              # We need this to be quick so we don't do extra validations to satisfy type checker
-              # steep:ignore:start
-              if cls.singleton_class?
-                cls = cls.attached_object
+              if cls.singleton_class? && cls.is_a?(Class)
+                cls = begin
+                  cls.attached_object
+                rescue NoMethodError
+                  # In Ruby 3.1, cls can be "Class:Delegator" which does not respond to attached_object
+                  next
+                end
                 next unless cls.is_a?(Module)
               end
-              vals = illegal_calls[cls.name]
-              if vals == :all || vals&.[](tp.callee_id)
+              vals = illegal_calls[cls.name.to_s]
+              if vals == :all || vals&.[](tp.callee_id) # steep:ignore
                 raise Workflow::NondeterminismError,
                       "Cannot access #{cls.name} #{tp.callee_id} from inside a " \
                       'workflow. If this is known to be safe, the code can be run in ' \
                       'a Temporalio::Workflow::Unsafe.illegal_call_tracing_disabled block.'
               end
-              # steep:ignore:end
             end
           end
 

temporalio/rbs_collection.lock.yaml

@@ -37,6 +37,10 @@ gems:
   version: '0'
   source:
     type: stdlib
+- name: benchmark
+  version: '0'
+  source:
+    type: stdlib
 - name: bigdecimal
   version: '0'
   source:
@@ -173,6 +177,10 @@ gems:
     revision: e9bc1bf94c262e79a2d599a9c173342915b29808
     remote: https://github.com/ruby/gem_rbs_collection.git
     repo_dir: gems
+- name: uri
+  version: '0'
+  source:
+    type: stdlib
 - name: yard
   version: '0.9'
   source:

temporalio/test/api/payload_visitor_test.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
 require 'temporalio/api/payload_visitor'
-require 'test_base'
+require 'test'
 
 module Api
-  class PayloadVisitorTest < TestBase
+  class PayloadVisitorTest < Test
     def test_basics
       # Make protos that have:
       # * single payload

temporalio/test/cancellation_test.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
 require 'temporalio/cancellation'
-require 'test_base'
+require 'test'
 
-class CancellationTest < TestBase
+class CancellationTest < Test
   also_run_all_tests_in_fiber
 
   def test_simple_cancellation

temporalio/test/client_schedule_test.rb

@@ -4,9 +4,9 @@
 require 'temporalio/client'
 require 'temporalio/client/schedule'
 require 'temporalio/testing'
-require 'test_base'
+require 'test'
 
-class ClientScheduleTest < TestBase
+class ClientScheduleTest < Test
   also_run_all_tests_in_fiber
 
   def test_basics # rubocop:disable Metrics/AbcSize,Metrics/MethodLength

temporalio/test/client_test.rb

@@ -3,9 +3,9 @@
 require 'async'
 require 'temporalio/client'
 require 'temporalio/testing'
-require 'test_base'
+require 'test'
 
-class ClientTest < TestBase
+class ClientTest < Test
   def test_version_number
     assert !Temporalio::VERSION.nil?
   end

temporalio/test/client_workflow_test.rb

@@ -3,9 +3,9 @@
 require 'async'
 require 'temporalio/client'
 require 'temporalio/testing'
-require 'test_base'
+require 'test'
 
-class ClientWorkflowTest < TestBase
+class ClientWorkflowTest < Test
   also_run_all_tests_in_fiber
 
   def test_start_simple

temporalio/test/converters/data_converter_test.rb

@@ -5,10 +5,10 @@
 require 'temporalio/converters/data_converter'
 require 'temporalio/converters/payload_codec'
 require 'temporalio/testing'
-require 'test_base'
+require 'test'
 
 module Converters
-  class DataConverterTest < TestBase
+  class DataConverterTest < Test
     def test_with_codec
       converter = Temporalio::Converters::DataConverter.new(
         failure_converter: Ractor.make_shareable(