79 lines
2.2 KiB
Ruby
79 lines
2.2 KiB
Ruby
# Copyright 2016 Mirantis, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
|
|
require File.join(File.dirname(__FILE__), '../spec_helper')
|
|
|
|
describe Astute::TaskCluster do
|
|
include SpecHelpers
|
|
|
|
subject { Astute::TaskCluster.new }
|
|
|
|
let(:node) { Astute::TaskNode.new('node_name', subject) }
|
|
|
|
before(:each) do
|
|
subject.stubs(:validate_fault_tolerance)
|
|
end
|
|
|
|
describe "#hook_internal_post_node_poll" do
|
|
it 'should call gracefully_stop with node' do
|
|
subject.expects(:gracefully_stop).with(node)
|
|
subject.hook_internal_post_node_poll(node)
|
|
end
|
|
end
|
|
|
|
describe "#noop_run" do
|
|
it 'should set cluster noop mode' do
|
|
subject.noop_run = true
|
|
expect(subject.noop_run).to eq true
|
|
end
|
|
end
|
|
|
|
describe "#gracefully_stop" do
|
|
it 'should check if node should be stopped' do
|
|
subject.expects(:gracefully_stop?).returns(false)
|
|
subject.hook_internal_post_node_poll(node)
|
|
end
|
|
|
|
it 'should check if node ready' do
|
|
subject.stop_condition { true }
|
|
node.expects(:ready?).returns(false)
|
|
subject.hook_internal_post_node_poll(node)
|
|
end
|
|
|
|
it 'should set node status as skipped if stopped' do
|
|
subject.stop_condition { true }
|
|
node.stubs(:ready?).returns(true)
|
|
node.stubs(:report_node_status)
|
|
|
|
node.expects(:set_status_skipped).once
|
|
subject.hook_internal_post_node_poll(node)
|
|
end
|
|
|
|
it 'should report new node status if stopped' do
|
|
subject.stop_condition { true }
|
|
node.stubs(:ready?).returns(true)
|
|
node.stubs(:set_status_skipped).once
|
|
|
|
node.expects(:report_node_status)
|
|
subject.hook_internal_post_node_poll(node)
|
|
end
|
|
end
|
|
|
|
it "should able to setup stop_condition" do
|
|
expect(subject).to respond_to(:stop_condition)
|
|
end
|
|
|
|
end
|