hq-sms-global-balance-check 0.0.0

data/features/basic.feature

@@ -0,0 +1,65 @@
+Feature: Basic functionality
+
+  Background:
+
+    Given a file "default.config":
+      """
+      <hq-sms-global-balance-check-config>
+        <server url="${server-url}"/>
+        <account name="account" username="USER" password="PASS"/>
+      </hq-sms-global-balance-check-config>
+      """
+
+  Scenario: Balance ok
+
+    Given that the balance is 750
+
+    When I run the script with args:
+      """
+      --config default.config
+      --account account
+      --warning 500
+      --critical 100
+      """
+
+    Then the status should be 0
+    And the output should be:
+      """
+      SMS Global account OK: 750 credits
+      """
+
+  Scenario: Balance warning
+
+    Given that the balance is 250
+
+    When I run the script with args:
+      """
+      --config default.config
+      --account account
+      --warning 500
+      --critical 100
+      """
+
+    Then the status should be 1
+    And the output should be:
+      """
+      SMS Global account WARNING: 250 credits (warning is 500)
+      """
+
+  Scenario: Balance critical
+
+    Given that the balance is 50
+
+    When I run the script with args:
+      """
+      --config default.config
+      --account account
+      --warning 500
+      --critical 100
+      """
+
+    Then the status should be 2
+    And the output should be:
+      """
+      SMS Global account CRITICAL: 50 credits (critical is 100)
+      """

data/features/support/steps.rb

@@ -0,0 +1,48 @@
+require "shellwords"
+
+require "hq/sms-global/balance-check/script"
+
+Given /^a file "(.*?)":$/ do
+	|file_name, file_contents|
+
+	File.open file_name, "w" do
+		|file_io|
+
+		file_contents =
+			file_contents.gsub "${server-url}", $web_url
+
+		file_io.write file_contents
+
+	end
+
+end
+
+Given /^that the balance is (\d+)$/ do
+	|balance_str|
+
+	$balance = balance_str.to_f
+
+end
+
+When /^I run the script with args:$/ do
+	|args_str|
+
+	@script = HQ::SmsGlobal::BalanceCheck::Script.new
+	@script.args = Shellwords.split args_str
+
+	@script.stdout = StringIO.new
+	@script.stderr = StringIO.new
+
+	@script.main
+
+end
+
+Then /^the status should be (\d+)$/ do
+	|status_str|
+	@script.status.should == status_str.to_i
+end
+
+Then /^the output should be:$/ do
+	|expected_output|
+	@script.stdout.string.strip.should == expected_output
+end

data/features/support/temp-dir.rb

@@ -0,0 +1,16 @@
+require "tmpdir"
+
+Before do
+
+	@old_dir = Dir.pwd
+	@temp_dir = Dir.mktmpdir
+	Dir.chdir @temp_dir
+
+end
+
+After do
+
+	FileUtils.remove_entry_secure @temp_dir
+	Dir.chdir @old_dir
+
+end

data/features/support/web-server.rb

@@ -0,0 +1,40 @@
+require "webrick"
+
+$web_config = {
+	:Port => 0,
+	:AccessLog => [],
+	:Logger => WEBrick::Log::new("/dev/null", 7),
+	:DoNotReverseLookup => true,
+}
+
+$web_server =
+	WEBrick::HTTPServer.new \
+		$web_config
+
+$web_port =
+	$web_server.listeners[0].local_address.ip_port
+
+$web_url =
+	"http://localhost:#{$web_port}"
+
+Thread.new do
+	$web_server.start
+end
+
+at_exit do
+	$web_server.shutdown
+end
+
+$web_server.mount_proc "/balance-api.php" do
+	|request, response|
+
+	request.query["user"].should == "USER"
+	request.query["password"].should == "PASS"
+
+	response.body = "BALANCE: #{$balance}; USER: USER"
+
+end
+
+Before do
+	$balance = 0.0
+end

data/lib/hq/sms-global/balance-check/script.rb

@@ -0,0 +1,125 @@
+require "hq/tools/check-script"
+require "hq/tools/escape"
+require "hq/tools/getopt"
+
+require "net/http"
+
+require "xml"
+
+module HQ
+module SmsGlobal
+module BalanceCheck
+
+class Script < Tools::CheckScript
+
+	include Tools::Escape
+
+	def process_args
+
+		@opts, @args =
+			Tools::Getopt.process @args, [
+
+				{ :name => :config,
+					:required => true },
+
+				{ :name => :account,
+					:required => true },
+
+				{ :name => :warning,
+					:convert => :to_f,
+					:required => true },
+
+				{ :name => :critical,
+					:convert => :to_f,
+					:required => true },
+
+			]
+
+		raise "Extra args" unless @args.empty?
+
+		@name = "SMS Global #{@opts[:account]}"
+
+	end
+
+	def prepare
+
+		load_config
+
+	end
+
+	def load_config
+
+		config_doc =
+			XML::Document.file @opts[:config]
+
+		@config_elem =
+			config_doc.root
+
+		@server_elem =
+			@config_elem.find_first "server"
+
+		@account_elem =
+			@config_elem.find_first "account [
+				@name = #{esc_xp @opts[:account]}
+			]"
+
+	end
+
+	def perform_checks
+
+		url =
+			URI.parse "#{@server_elem["url"]}/balance-api.php"
+
+		url.query =
+			URI.encode_www_form({
+				"user" => @account_elem["username"],
+				"password" => @account_elem["password"],
+			})
+
+		Net::HTTP.start url.host, url.port do
+			|http|
+
+
+			request = Net::HTTP::Get.new "#{url.path}?#{url.query}"
+
+			response = http.request request
+
+			raise "Error 1" \
+				unless response.code == "200"
+
+			raise "Error 2" \
+				unless response.body =~ /^BALANCE: ([^;]+); USER: (.+)$/
+
+			actual_balance = $1.to_f
+
+			if actual_balance < @opts[:critical]
+
+				critical "%s credits (critical is %s)" % [
+					actual_balance.to_i,
+					@opts[:critical].to_i,
+				]
+
+			elsif actual_balance < @opts[:warning]
+
+				warning "%s credits (warning is %s)" % [
+					actual_balance.to_i,
+					@opts[:warning].to_i,
+				]
+
+			else
+
+				message "%s credits" % [
+					actual_balance.to_i,
+				]
+
+			end
+
+		end
+
+	end
+
+end
+
+end
+end
+end

metadata

@@ -0,0 +1,200 @@
+--- !ruby/object:Gem::Specification
+name: hq-sms-global-balance-check
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+  prerelease: 
+platform: ruby
+authors:
+- James Pharaoh
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: hq-tools
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: libxml-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+- !ruby/object:Gem::Dependency
+  name: cucumber
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.1
+- !ruby/object:Gem::Dependency
+  name: hq-dev
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.0.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.0.2
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.7.7
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.7.7
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 10.0.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 10.0.4
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+- !ruby/object:Gem::Dependency
+  name: rspec_junit_formatter
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.6
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.6
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.7.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.7.1
+description: HQ nagios/icinga plugin to check balance of SMS global account
+email:
+- james@phsys.co.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/hq/sms-global/balance-check/script.rb
+- features/basic.feature
+- features/support/temp-dir.rb
+- features/support/steps.rb
+- features/support/env.rb
+- features/support/web-server.rb
+homepage: https://github.com/jamespharaoh/hq-sms-global-balance-check
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.3.6
+requirements: []
+rubyforge_project: hq-sms-global-balance-check
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: HQ SMS Global balance check
+test_files:
+- features/basic.feature
+- features/support/temp-dir.rb
+- features/support/steps.rb
+- features/support/env.rb
+- features/support/web-server.rb