Contents

module Foodtaster
  class RSpecRun
    attr_reader :client

    def initialize
      @client = nil
      @server_process = nil
      @stopped = false
    end

    def start
      setup_signal_handlers
      start_server_and_connect_client

      if @client && @server_process && !@server_process.alive?
        Foodtaster.logger.fatal "Failed to start Foodtaster DRb Server:\n\n#{@server_process.output}"
        exit 1
      elsif !@client || !@server_process
        Foodtaster.logger.fatal "Failed to connect to Foodtaster DRb Server"
        exit 1
      end
    end

    def stop
      return if @stopped

      @stopped = true
      puts "" # newline after rspec output
      Vm.shutdown_running_vms if Foodtaster.config.shutdown_vms
      terminate_server
    end

    def client
      @client
    end

    class << self
      @instance = nil

      def current
        @instance ||= self.new
      end
    end

    private

    def setup_signal_handlers
      terminator = proc {
        self.stop
        exit 1
      }

      trap("INT", &terminator)
      trap("TERM", &terminator)

      at_exit do
        self.stop
      end
    end

    def start_server_and_connect_client
      drb_port = Foodtaster.config.drb_port

      @server_process = start_server(drb_port) if Foodtaster.config.start_server
      @client = connect_client(drb_port)
    end

    def start_server(drb_port)
      Foodtaster::ServerProcess.new(drb_port)
    end

    def terminate_server
      @server_process && @server_process.terminate
    end

    def connect_client(drb_port)
      Foodtaster::Client.connect(drb_port, @server_process)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
foodtaster-0.0.10	lib/foodtaster/rspec_run.rb
foodtaster-0.0.9	lib/foodtaster/rspec_run.rb