vmfloaty 0.7.0 → 0.7.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9bfa904ec8ca9ec10fa9c6c9571d5ebbb3a0647e
-  data.tar.gz: c6d347f9075dd4ecf5056ae39c4ff4f5f2b7e6e7
+  metadata.gz: c5a8aa5f96b46944ccf39b5218791ade53d56834
+  data.tar.gz: 5ba89a4493c1bbe5aa14caaecf58d992af59fdc2
 SHA512:
-  metadata.gz: 231d4ef21dc8ebb75b57511c4e4d33d8c77b2b0566027973c0a9e9cc790bc79a9bca532ab433b42600473e9811e867523a74927988524a11b4330542b639b85c
-  data.tar.gz: 88b32545eae35d0b49e366c5883052b20dcb38445c4ae534987b7de0bbb6246635ca5004c5cd8e095308692e0c817649655cf82aceab02b38a27bb380bb6d26f
+  metadata.gz: 46df128c9da17337d0c4880eedd3288d367b928134d3c931115d27801ddf22a776834926cd0c55896cba13b944915666fe4610f237b44efb179a8088313ca722
+  data.tar.gz: da0865befca262fcb10b32f05ea233a724f5b1c56bbd40f4b015ff455050b738900c6d7da55ac888eeb01dfc2a914f06d700819f58565e15ee4c9044961df733

data/LICENSE

@@ -1,6 +1,6 @@
-   vmpooler
+   vmfloaty
 
-   Copyright (C) 2014 Puppet Labs
+   Copyright (C) 2014-2016 Puppet
 
    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.

data/lib/vmfloaty.rb

@@ -177,7 +177,12 @@ class Vmfloaty
         token = options.token || config['token']
 
         if lifetime || tags
-          modify_req = Pooler.modify(verbose, url, hostname, token, lifetime, tags)
+          begin
+            modify_req = Pooler.modify(verbose, url, hostname, token, lifetime, tags)
+          rescue TokenError => e
+            STDERR.puts e
+            exit 1
+          end
 
           if modify_req["ok"]
             puts "Successfully modified vm #{hostname}."
@@ -189,7 +194,13 @@ class Vmfloaty
         end
 
         if disk
-          disk_req = Pooler.disk(verbose, url, hostname, token, disk)
+          begin
+            disk_req = Pooler.disk(verbose, url, hostname, token, disk)
+          rescue TokenError => e
+            STDERR.puts e
+            exit 1
+          end
+
           if disk_req["ok"]
             puts "Successfully updated disk space of vm #{hostname}."
           else
@@ -290,7 +301,13 @@ class Vmfloaty
         hostname = args[0]
         token = options.token ||= config['token']
 
-        snapshot_req = Pooler.snapshot(verbose, url, hostname, token)
+        begin
+          snapshot_req = Pooler.snapshot(verbose, url, hostname, token)
+        rescue TokenError => e
+          STDERR.puts e
+          exit 1
+        end
+
         pp snapshot_req
       end
     end
@@ -315,7 +332,13 @@ class Vmfloaty
           STDERR.puts "Two snapshot arguments were given....using snapshot #{snapshot_sha}"
         end
 
-        revert_req = Pooler.revert(verbose, url, hostname, token, snapshot_sha)
+        begin
+          revert_req = Pooler.revert(verbose, url, hostname, token, snapshot_sha)
+        rescue TokenError => e
+          STDERR.puts e
+          exit 1
+        end
+
         pp revert_req
       end
     end

data/lib/vmfloaty/pooler.rb

@@ -55,6 +55,10 @@ class Pooler
   end
 
   def self.modify(verbose, url, hostname, token, lifetime, tags)
+    if token.nil?
+      raise TokenError, "Token provided was nil. Request cannot be made to modify vm"
+    end
+
     modify_body = {}
     if lifetime
       modify_body['lifetime'] = lifetime
@@ -76,6 +80,10 @@ class Pooler
   end
 
   def self.disk(verbose, url, hostname, token, disk)
+    if token.nil?
+      raise TokenError, "Token provided was nil. Request cannot be made to modify vm"
+    end
+
     conn = Http.get_conn(verbose, url)
     conn.headers['X-AUTH-TOKEN'] = token
 
@@ -129,6 +137,10 @@ class Pooler
   end
 
   def self.snapshot(verbose, url, hostname, token)
+    if token.nil?
+      raise TokenError, "Token provided was nil. Request cannot be made to snapshot vm"
+    end
+
     conn = Http.get_conn(verbose, url)
     conn.headers['X-AUTH-TOKEN'] = token
 
@@ -138,6 +150,10 @@ class Pooler
   end
 
   def self.revert(verbose, url, hostname, token, snapshot_sha)
+    if token.nil?
+      raise TokenError, "Token provided was nil. Request cannot be made to revert vm"
+    end
+
     conn = Http.get_conn(verbose, url)
     conn.headers['X-AUTH-TOKEN'] = token
 

data/lib/vmfloaty/version.rb

@@ -1,6 +1,6 @@
 
 class Version
-  @version = '0.7.0'
+  @version = '0.7.1'
 
   def self.get
     @version

data/spec/vmfloaty/pooler_spec.rb

@@ -90,6 +90,10 @@ describe Pooler do
       @modify_response_body_fail = "{\"ok\":false}"
     end
 
+    it "raises a TokenError if token provided is nil" do
+      expect{ Pooler.modify(false, @vmpooler_url, 'myfakehost', nil, 12, nil) }.to raise_error(TokenError)
+    end
+
     it "modifies the TTL of a vm" do
       stub_request(:put, "#{@vmpooler_url}/vm/fq6qlpjlsskycq6").
         with(:body => {"{\"lifetime\":12}"=>true},
@@ -116,7 +120,7 @@ describe Pooler do
     end
   end
 
-  describe "#staus" do
+  describe "#status" do
     before :each do
       #smaller version
       @status_response_body = "{\"capacity\":{\"current\":716,\"total\":717,\"percent\": 99.9},\"status\":{\"ok\":true,\"message\":\"Battle station fully armed and operational.\"}}"
@@ -188,5 +192,28 @@ describe Pooler do
     it "doesn't make a request to revert a vm if snapshot is not provided" do
       expect{ Pooler.revert(false, @vmpooler_url, 'fq6qlpjlsskycq6', 'mytokenfile', nil) }.to raise_error(RuntimeError, "Snapshot SHA provided was nil, could not revert fq6qlpjlsskycq6")
     end
+
+    it "raises a TokenError if no token was provided" do
+      expect{ Pooler.revert(false, @vmpooler_url, 'myfakehost', nil, 'shaaaaaaa') }.to raise_error(TokenError)
+    end
+  end
+
+  describe "#disk" do
+    before :each do
+      @disk_response_body_success = "{\"ok\":true}"
+      @disk_response_body_fail = "{\"ok\":false}"
+    end
+
+    it "makes a request to extend disk space of a vm" do
+      stub_request(:post, "#{@vmpooler_url}/vm/fq6qlpjlsskycq6/disk/12").
+        with(:headers => {'Accept'=>'*/*', 'Accept-Encoding'=>'gzip;q=1.0,deflate;q=0.6,identity;q=0.3', 'Content-Length'=>'0', 'User-Agent'=>'Faraday v0.9.2', 'X-Auth-Token'=>'mytokenfile'}).  to_return(:status => 200, :body => @disk_response_body_success, :headers => {})
+
+      disk_req = Pooler.disk(false, @vmpooler_url, 'fq6qlpjlsskycq6', 'mytokenfile', 12)
+      expect(disk_req["ok"]).to be true
+    end
+
+    it "raises a TokenError if no token was provided" do
+      expect{ Pooler.disk(false, @vmpooler_url, 'myfakehost', nil, 12) }.to raise_error(TokenError)
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vmfloaty
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.7.1
 platform: ruby
 authors:
 - Brian Cain
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-09-24 00:00:00.000000000 Z
+date: 2016-10-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: commander