services/api/test/websocket_runner.rb

   1 require 'bundler'
   2
   3 $ARV_API_SERVER_DIR = File.expand_path('../..', __FILE__)
   4 SERVER_PID_PATH = 'tmp/pids/passenger.3002.pid'
   5
   6 class WebsocketTestRunner < MiniTest::Unit
   7   def _system(*cmd)
   8     Bundler.with_clean_env do
   9       if not system({'ARVADOS_WEBSOCKETS' => 'ws-only', 'RAILS_ENV' => 'test'}, *cmd)
  10         raise RuntimeError, "Command failed with exit status #{$?}: #{cmd.inspect}"
  11       end
  12     end
  13   end
  14
  15   def _run(args=[])
  16     server_pid = Dir.chdir($ARV_API_SERVER_DIR) do |apidir|
  17       # Only passenger seems to be able to run the websockets server successfully.
  18       _system('passenger', 'start', '-d', '-p3002')
  19       timeout = Time.now.tv_sec + 10
  20       begin
  21         sleep 0.2
  22         begin
  23           server_pid = IO.read(SERVER_PID_PATH).to_i
  24           good_pid = (server_pid > 0) and (Process.kill(0, pid) rescue false)
  25         rescue Errno::ENOENT
  26           good_pid = false
  27         end
  28       end while (not good_pid) and (Time.now.tv_sec < timeout)
  29       if not good_pid
  30         raise RuntimeError, "could not find API server Rails pid"
  31       end
  32       server_pid
  33     end
  34     begin
  35       super(args)
  36     ensure
  37       Dir.chdir($ARV_API_SERVER_DIR) do
  38         _system('passenger', 'stop', '-p3002')
  39       end
  40       # DatabaseCleaner leaves the database empty. Prefer to leave it full.
  41       dc = DatabaseController.new
  42       dc.define_singleton_method :render do |*args| end
  43       dc.reset
  44     end
  45   end
  46 end
  47
  48 MiniTest::Unit.runner = WebsocketTestRunner.new