From c9e582c24ac4dc1542396989cdfc66c1ce5cff24 Mon Sep 17 00:00:00 2001
From: Brian Cain <brian.cain@puppetlabs.com>
Date: Sat, 24 Sep 2016 13:34:28 -0700
Subject: [PATCH] (maint) Catch TokenError from Auth class

---
 lib/vmfloaty.rb | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/lib/vmfloaty.rb b/lib/vmfloaty.rb
index 3f9ec26..e17e8e5 100644
--- a/lib/vmfloaty.rb
+++ b/lib/vmfloaty.rb
@@ -66,7 +66,7 @@ class Vmfloaty
               pass = password "Enter your password please:", '*'
               begin
                 token = Auth.get_token(verbose, url, user, pass)
-              rescue AuthError => e
+              rescue TokenError => e
                 STDERR.puts e
                 exit 1
               end
@@ -114,7 +114,7 @@ class Vmfloaty
           # list active vms
           begin
             status = Auth.token_status(verbose, url, token)
-          rescue AuthError => e
+          rescue TokenError => e
             STDERR.puts e
             exit 1
           end
@@ -224,7 +224,7 @@ class Vmfloaty
           # get vms with token
           begin
             status = Auth.token_status(verbose, url, token)
-          rescue AuthError => e
+          rescue TokenError => e
             STDERR.puts e
             exit 1
           end
@@ -373,7 +373,7 @@ class Vmfloaty
           pass = password "Enter your password please:", '*'
           begin
             token = Auth.get_token(verbose, url, user, pass)
-          rescue AuthError => e
+          rescue TokenError => e
             STDERR.puts e
             exit 1
           end
@@ -383,7 +383,7 @@ class Vmfloaty
           pass = password "Enter your password please:", '*'
           begin
             result = Auth.delete_token(verbose, url, user, pass, token)
-          rescue AuthError => e
+          rescue TokenError => e
             STDERR.puts e
             exit 1
           end
@@ -392,7 +392,7 @@ class Vmfloaty
         when "status"
           begin
             status = Auth.token_status(verbose, url, token)
-          rescue AuthError => e
+          rescue TokenError => e
             STDERR.puts e
             exit 1
           end
@@ -439,7 +439,7 @@ class Vmfloaty
           pass = password "Enter your password please:", '*'
           begin
             token = Auth.get_token(verbose, url, user, pass)
-          rescue AuthError => e
+          rescue TokenError => e
             STDERR.puts e
             STDERR.puts 'Could not get token...requesting vm without a token anyway...'
           else