timeskipping server works

Makefile

@@ -2,8 +2,14 @@ PROTO_ROOT := proto
 PROTO_FILES = $(shell find $(PROTO_ROOT) -name "*.proto")
 PROTO_DIRS = $(sort $(dir $(PROTO_FILES)))
 PROTO_OUT := lib/gen
+PROTO_TESTING_ROOT := lib/temporal/testing/proto
+PROTO_TESTING_FILES = $(shell find $(PROTO_TESTING_ROOT) -name "*.proto")
+PROTO_TESTING_DIRS = $(sort $(dir $(PROTO_TESTING_FILES)))
 
 proto:
 	$(foreach PROTO_DIR,$(PROTO_DIRS),bundle exec grpc_tools_ruby_protoc -Iproto --ruby_out=$(PROTO_OUT) --grpc_out=$(PROTO_OUT) $(PROTO_DIR)*.proto;)
 
-.PHONY: proto
+proto-testing:
+	$(foreach PROTO_DIR,$(PROTO_TESTING_DIRS),bundle exec grpc_tools_ruby_protoc --proto_path="lib/temporal/testing/proto" --ruby_out=$(PROTO_OUT) --grpc_out=$(PROTO_OUT) $(PROTO_DIR)*.proto;)
+
+.PHONY: proto, proto-testing

lib/temporal/testing/client.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+module Temporal
+  module Testing
+    # TestingClient expects the java `temporal-test-server` to be running and to have been
+    # enabled with timeskipping with
+    # `./temporal-test-server {port} --enable-time-skipping`
+    class Client < Temporal::Client
+      def initialize(config = {})
+        super(config)
+        @timeskipping_connection = Connection.new(config.host, config.port)
+      end
+
+      def sleep(duration)
+        @timeskipping_connection.unlock_time_skipping_with_sleep(duration)
+      end
+
+      def await_workflow_result(workflow, workflow_id:, run_id: nil, timeout: nil, namespace: nil)
+        begin
+          # WorkflowTaskTimeout seems to happen with this SDK so sleep as workaround
+          sleep(0.25)
+          @timeskipping_connection.unlock_time_skipping
+          super
+        ensure
+          @timeskipping_connection.lock_time_skipping
+        end
+      end
+    end
+  end
+
+end