hq-tools 0.4.1 → 0.4.2

data/lib/hq/tools/check-script.rb

@@ -55,12 +55,16 @@ class CheckScript
 
 		if @critical
 			str.print "CRITICAL"
+			@status = 2
 		elsif @warning
 			str.print "WARNING"
+			@status = 1
 		elsif @unknown
 			str.print "UNKNOWN"
+			@status = 3
 		else
 			str.print "OK"
+			@status = 0
 		end
 
 		str.print ": "

data/spec/hq/tools/check-script-spec.rb

@@ -11,6 +11,56 @@ describe CheckScript do
 		subject.stdout = StringIO.new
 	end
 
+	context "status" do
+
+		it "critical" do
+			subject.stub(:perform_checks) do
+				subject.critical "critical"
+				subject.warning "warning"
+				subject.unknown "unknown"
+				subject.message "ok"
+			end
+			subject.main
+			subject.stdout.string.should ==
+				"Unnamed CRITICAL: critical, warning, unknown, ok\n"
+			subject.status.should == 2
+		end
+
+		it "warning" do
+			subject.stub(:perform_checks) do
+				subject.warning "warning"
+				subject.unknown "unknown"
+				subject.message "ok"
+			end
+			subject.main
+			subject.stdout.string.should ==
+				"Unnamed WARNING: warning, unknown, ok\n"
+			subject.status.should == 1
+		end
+
+		it "unknown" do
+			subject.stub(:perform_checks) do
+				subject.unknown "unknown"
+				subject.message "ok"
+			end
+			subject.main
+			subject.stdout.string.should ==
+				"Unnamed UNKNOWN: unknown, ok\n"
+			subject.status.should == 3
+		end
+
+		it "ok" do
+			subject.stub(:perform_checks) do
+				subject.message "ok"
+			end
+			subject.main
+			subject.stdout.string.should ==
+				"Unnamed OK: ok\n"
+			subject.status.should == 0
+		end
+
+	end
+
 	context "performance data" do
 
 		it "none" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hq-tools
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
   prerelease: 
 platform: ruby
 authors: