testotron 0.0.3

data/lib/test.rb

@@ -0,0 +1,4 @@
+module Testotron
+	class Test
+	end
+end

data/lib/testotron.rb

@@ -0,0 +1,95 @@
+#!/usr/bin/ruby
+
+require 'mail'
+require 'net/smtp'
+require 'net/http'
+require 'uri'
+
+require 'tests/http'
+require 'tests/smtp'
+
+module Testotron
+	class TestFailed < Exception
+	end
+	
+	TEST_CLASSES = [ Tests::HTTP, Tests::SMTP ]
+
+	class TestBuilder
+		def initialize(runner)
+			@runner = runner
+			@report_methods = [ :local_mail ]
+		end
+
+		# TODO: save for later execution
+		TEST_CLASSES.each { |klass|
+			define_method klass.const_get(:KEY) do |*args|
+				test = klass.new(*args)	
+				begin
+					test.run(@runner)
+				rescue TestFailed => failure
+					@runner.report_error(@report_methods, test, failure)
+				end
+			end
+		}
+
+		def report_with(*methods)
+			@report_methods = methods.map &:to_sym
+		end
+	end
+
+	# TODO: set mail target
+	class TestRunner
+		def report(test, msg)
+			puts test.class.to_s.rjust(20) + ": #{msg}"
+		end
+
+		REPORT_METHODS = [ :local_mail, :stderr, :xosdutil ]
+
+		def report_error(methods, test, failure)
+			if methods.include?(:local_mail)
+				mail = Mail.new do
+					from 'conftest@localhost'
+					to `whoami`.chomp
+					subject "Configuration test failed!"
+					body <<EOF
+		Test:
+			#{test.human_name}
+
+		Message: 
+			#{failure.message}
+EOF
+				end
+
+				mail.delivery_method :sendmail
+				mail.deliver
+			end
+
+			if methods.include? :xosdutil
+				system "xosdutilctl", "echo", "Configuration test (#{test.human_name}) failed!"
+			end
+
+			if methods.include? :stderr
+				STDERR.puts "Test failed: #{test.human_name} (message: #{failure.message})"
+			end
+		end
+	end
+
+	def self.test(*args)
+		runner = TestRunner.new
+
+		if block_given?
+			yield(TestBuilder.new(runner))
+		else
+			raise ArgumentError if args.empty?
+			test = args.shift.to_sym
+			keys = TEST_CLASSES.map { |x| x.const_get(:KEY).to_sym }
+			raise KeyError, "Unknown test: #{test}" unless TEST_CLASSES.map { |x| x.const_get(:KEY).to_sym }.include?(test)
+			TEST_CLASSES.each { |klass|
+				if klass.const_get(:KEY).to_sym == test
+					klass.new(*args).run(runner)
+					break
+				end
+			}
+		end
+	end
+end

data/lib/tests/http.rb

@@ -0,0 +1,39 @@
+require 'test'
+
+module Testotron::Tests
+	class HTTP < Testotron::Test
+		KEY = "http"
+
+		def initialize(host, port = 80, requests = nil)
+			if requests.nil?
+				requests = "http://#{host}/"
+			end
+			@host, @port, @requests = host, port, [*requests]
+		end
+
+		def human_name
+			"HTTP test of #{@host}, port #{@port}, requests #{@requests.join ','}"
+		end
+
+		def run(runner)
+			runner.report self, "Testing HTTP server on #{@host} port #{@port}..."
+			http = Net::HTTP.new(@host, @port)
+			@requests.each do |page|
+				runner.report self, "Trying #{page}..."
+				request = Net::HTTP::Get.new URI.parse(page).request_uri
+
+				begin
+					response = http.request(request)
+				rescue Errno::ETIMEDOUT
+					raise TestFailed, "HTTP connection timed out"
+				rescue Errno::ECONNREFUSED
+					raise TestFailed, "HTTP connection refused"
+				end
+
+				if response.code.to_i != 200
+					raise TestFailed, "Response not 200 on #{@post}:#{@port} GET #{page}"
+				end
+			end
+		end
+	end
+end

data/lib/tests/smtp.rb

@@ -0,0 +1,30 @@
+require 'test'
+
+module Testotron::Tests
+	class SMTP < Testotron::Test
+		KEY = "smtp"
+
+		def initialize(host, port = 25)
+			@host, @port = host, port
+		end
+
+		def human_name
+			"SMTP test of #{@host}, port #{@port}"
+		end
+
+		def run(runner)
+			runner.report self, "Testing SMTP server of #{@host} port #{@port}..."
+			smtp = Net::SMTP.new(@host, @port)
+			begin
+				smtp.start
+				smtp.finish
+			rescue Errno::ECONNREFUSED => e
+				raise TestFailed, "Server refused SMTP connection"
+			rescue EOFError => e
+				raise TestFailed, "EOF reached while connecting to SMTP server"
+			rescue Exception
+				raise TestFailed
+			end
+		end
+	end
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: testotron
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- Michal Pokorny
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mail
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: ! "\t\tTestotron can automatically test basic functions of common servers,
+  so you\n\t\tcan get an alert when some piece of your configuration stops working.\n"
+email: pok@rny.cz
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/testotron.rb
+- lib/test.rb
+- lib/tests/smtp.rb
+- lib/tests/http.rb
+homepage: http://github.com/MichalPokorny/testotron
+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: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Simple server testing
+test_files: []