Always look before you leap into hashes.

Signed-off-by: Robin H. Johnson <robbat2@gentoo.org>

1 files changed, 9 insertions, 1 deletions

diff --git a/lib/service-dsl.rb b/lib/service-dsl.rb
index ec80f24..f6eaace 100644
--- a/lib/service-dsl.rb
+++ b/lib/service-dsl.rb
@@ -5,26 +5,34 @@ module HelperMethods
 	# Checks if all hosts are up
 	def all_hosts_up?(hosts)
 		hosts.each do |host|
-			return false unless status_data['hosts'][host]['current_state'] == 0
+			return false unless 
+				status_data['hosts'].has_key?(host) and 
+				status_data['hosts'][host]['current_state'] == 0
 		end
 
 		true
 	end
 
 	def host_up?(host)
+		status_data['hosts'].has_key?(host) and 
 		status_data['hosts'][host]['current_state'] == 0
 	end
 
 	def host_flapping?(host)
+		status_data['hosts'].has_key?(host) and 
 		status_data['hosts'][host]['is_flapping'] != 0
 	end
 
 	# Checks if the service is up
 	def service_up?(host, service)
+		status_data['services'].has_key?(host) and 
+		status_data['services'][host].has_key?(service) and 
 		status_data['services'][host][service]['current_state'] == 0
 	end
 
 	def service_flapping?(host, service)
+		status_data['services'].has_key?(host) and 
+		status_data['services'][host].has_key?(service) and 
 		status_data['services'][host][service]['is_flapping'] != 0
 	end