Introduce TIMEOUT_SEC on test

red-data-tools · May 31, 2024 · eb68d14 · eb68d14
1 parent fdd041a
commit eb68d14
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -72,6 +72,8 @@ jobs:
     - run: bundle install --jobs 4 --retry 3 --without "nmatrix python"
 
     - run: bundle exec rake
+      env:
+        TIMEOUT_SEC: 180
 
     - run: rake build
 

diff --git a/.github/workflows/nmatrix.yml b/.github/workflows/nmatrix.yml
@@ -65,6 +65,8 @@ jobs:
     - run: bundle install --jobs 4 --retry 3 --without "numo python"
 
     - run: bundle exec rake
+      env:
+        TIMEOUT_SEC: 180
 
     - run: rake build
 

diff --git a/.github/workflows/pycall.yml b/.github/workflows/pycall.yml
@@ -86,4 +86,5 @@ jobs:
     - run: bundle exec rake
       env:
         PYTHON: python
+        TIMEOUT_SEC: 180
       continue-on-error: ${{ matrix.python == '2.x' }}
diff --git a/test/run.rb b/test/run.rb
@@ -4,6 +4,7 @@
 # $VERBOSE = true
 
 require "pathname"
+require "timeout"
 
 base_dir = Pathname(__dir__).parent.expand_path
 
@@ -16,4 +17,11 @@
 
 require_relative "helper"
 
-exit(Test::Unit::AutoRunner.run(true, test_dir.to_s))
+run_test = lambda { Test::Unit::AutoRunner.run(true, test_dir.to_s) }
+timeout_sec = ENV.fetch("TIMEOUT_SEC", "0").to_i
+status = if timeout_sec > 0
+           Timeout.timeout(timeout_sec) { run_test.() }
+         else
+           run_test.()
+         end
+exit(status)