apps/workbench/test/integration_helper.rb

   1 require 'test_helper'
   2 require 'capybara/rails'
   3 require 'capybara/poltergeist'
   4 require 'uri'
   5 require 'yaml'
   6
   7 $ARV_API_SERVER_DIR = File.expand_path('../../../../services/api', __FILE__)
   8
   9 class ActionDispatch::IntegrationTest
  10   # Make the Capybara DSL available in all integration tests
  11   include Capybara::DSL
  12
  13   def self.api_fixture(name)
  14     # Returns the data structure from the named API server test fixture.
  15     path = File.join($ARV_API_SERVER_DIR, 'test', 'fixtures', "#{name}.yml")
  16     YAML.load(IO.read(path))
  17   end
  18
  19   @@API_AUTHS = api_fixture('api_client_authorizations')
  20
  21   def page_with_token(token, path='/')
  22     # Generate a page path with an embedded API token.
  23     # Typical usage: visit page_with_token('token_name', page)
  24     # The token can be specified by the name of an api_client_authorizations
  25     # fixture, or passed as a raw string.
  26     api_token = ((@@API_AUTHS.include? token) ?
  27                  @@API_AUTHS[token]['api_token'] : token)
  28     sep = (path.include? '?') ? '&' : '?'
  29     q_string = URI.encode_www_form('api_token' => api_token)
  30     "#{path}#{sep}#{q_string}"
  31   end
  32 end
  33
  34 class IntegrationTestRunner < MiniTest::Unit
  35   # Don't try to re-use the current Bundle environment when we launch the
  36   # API server.
  37   @@APIENV = ENV.map { |item| (item[0] =~ /^BUNDLE_/) ? [item[0], nil] : nil }.
  38     compact.to_h
  39
  40   def _system(*cmd)
  41     if not system(@@APIENV, *cmd)
  42       raise RuntimeError, "#{cmd[0]} returned exit code #{$?.exitstatus}"
  43     end
  44   end
  45
  46   def _run(args=[])
  47     Capybara.javascript_driver = :poltergeist
  48     server_pid = Dir.chdir($ARV_API_SERVER_DIR) do |apidir|
  49       _system('bundle', 'exec', 'rake', 'db:test:load')
  50       _system('bundle', 'exec', 'rake', 'db:fixtures:load')
  51       _system('bundle', 'exec', 'rails', 'server', '-d')
  52       pid_path = 'tmp/pids/server.pid'
  53       timeout = Time.now.tv_sec + 5
  54       while (not File.exists? pid_path) and (Time.now.tv_sec < timeout)
  55         sleep 0.2
  56       end
  57       IO.read(pid_path).to_i
  58     end
  59     begin
  60       super(args)
  61     ensure
  62       Process.kill('TERM', server_pid)
  63     end
  64   end
  65 end
  66
  67 MiniTest::Unit.runner = IntegrationTestRunner.new