From 559005057e9aa6fb3f2788d062701c8bdacacaf8 Mon Sep 17 00:00:00 2001
From: Brian Cain <brian.cain@puppetlabs.com>
Date: Mon, 12 Oct 2015 09:48:17 -0700
Subject: [PATCH] Ensure vm number is specified from cli args

---
 lib/vmfloaty.rb        | 4 ++++
 lib/vmfloaty/pooler.rb | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/lib/vmfloaty.rb b/lib/vmfloaty.rb
index 9b459e7..3e19a52 100644
--- a/lib/vmfloaty.rb
+++ b/lib/vmfloaty.rb
@@ -33,6 +33,10 @@ class Vmfloaty
         user = options.user ||= config['user']
         url = options.url ||= config['url']
 
+        if args.empty?
+          STDERR.puts "You did not provide any vms to grab"
+        end
+
         os_types = {}
         args.each do |arg|
           os_arr = arg.split("=")
diff --git a/lib/vmfloaty/pooler.rb b/lib/vmfloaty/pooler.rb
index 0be14a9..eb91772 100644
--- a/lib/vmfloaty/pooler.rb
+++ b/lib/vmfloaty/pooler.rb
@@ -33,6 +33,11 @@ class Pooler
 
     os_string = os_string.chomp("+")
 
+    if os_string.size == 0
+      STDERR.puts "No request was made, os hash specified no vms #{os_type}"
+      exit 1
+    end
+
     response = conn.post "/vm/#{os_string}"
 
     res_body = JSON.parse(response.body)