Contents

module PoolParty
  class PluginSpecHelper
    def self.define_stubs(klass, num=1)
      require File.dirname(__FILE__) + '/../../spec/helpers/ec2_mock'
      
      @klass = klass.send :new
      klass.stub!(:new).and_return @klass
      
      @master = Master.new      
      @instances = define_instances(num)
      
      Master.stub!(:new).and_return @master

      @master.stub!(:execute_tasks).and_return true
      @master.stub!(:launch_minimum_instances).and_return true
      @master.stub!(:number_of_pending_instances).and_return 0
      @master.stub!(:get_node).with(0).and_return @instance0

      @master.stub!(:nodes).and_return @instances

      Kernel.stub!(:system).and_return "true"

      Provider.stub!(:install_poolparty).and_return true

      [@klass, @master, @instances]
    end
    def self.define_instances(num)
      # Too many gross evals
      returning [] do |arr|
        num.times do |i|
          eval <<-EOE
            @instance#{i} = RemoteInstance.new
            @instance#{i}.stub!(:ssh).and_return "true"
            @instance#{i}.stub!(:scp).and_return "true"
            @instance#{i}.stub!(:run).and_return "true"
            @instance#{i}.stub!(:name).and_return "node#{i}"
            @instance#{i}.stub!(:ip).and_return "127.0.0.#{i}"
          EOE
          arr << eval("@instance#{i}")
        end
      end
    end
  end
end

module Spec
  module Mocks
    module Methods
      def should_receive_at_least_once(sym, opts={}, &block)
        begin
          e = __mock_proxy.add_message_expectation(opts[:expected_from] || caller(1)[0], sym.to_sym, opts, &block).at_least(1)
          __mock_proxy.add_message_expectation(opts[:expected_from] || caller(1)[0], sym.to_sym, opts, &block).any_number_of_times
          e
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 2 rubygems

Version	Path
auser-poolparty-0.1.2	lib/helpers/plugin_spec_helper.rb
jtzemp-poolparty-0.1.2	lib/helpers/plugin_spec_helper.rb