mirror of
https://github.com/puppetlabs/vmfloaty.git
synced 2026-01-26 05:28:40 -05:00
Raise TokenError if none provided to Pooler.delete
This commit is contained in:
parent
97e188cf6d
commit
a6ff0515d3
3 changed files with 16 additions and 1 deletions
|
|
@ -336,7 +336,14 @@ class Vmfloaty
|
|||
exit 1
|
||||
else
|
||||
hosts = hostnames.split(',')
|
||||
Pooler.delete(verbose, url, hosts, token)
|
||||
begin
|
||||
Pooler.delete(verbose, url, hosts, token)
|
||||
rescue TokenError => e
|
||||
STDERR.puts e
|
||||
exit 1
|
||||
end
|
||||
|
||||
puts "Schedulered vmpooler to delete vms #{hosts}."
|
||||
exit 0
|
||||
end
|
||||
end
|
||||
|
|
|
|||
|
|
@ -94,6 +94,10 @@ class Pooler
|
|||
end
|
||||
|
||||
def self.delete(verbose, url, hosts, token)
|
||||
if token.nil?
|
||||
raise TokenError, "Token provided was nil. Request cannot be made to delete vm"
|
||||
end
|
||||
|
||||
conn = Http.get_conn(verbose, url)
|
||||
|
||||
if token
|
||||
|
|
|
|||
|
|
@ -118,6 +118,10 @@ describe Pooler do
|
|||
|
||||
expect(Pooler.delete(false, @vmpooler_url, ['fq6qlpjlsskycq6'], 'mytokenfile')).to eq @delete_response
|
||||
end
|
||||
|
||||
it "raises a token error if no token provided" do
|
||||
expect{ Pooler.delete(false, @vmpooler_url, ['myfakehost'], nil) }.to raise_error(TokenError)
|
||||
end
|
||||
end
|
||||
|
||||
describe "#status" do
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue