summaryrefslogtreecommitdiff
path: root/lib/custodian
diff options
context:
space:
mode:
authorSteve Kemp <steve@steve.org.uk>2013-02-25 17:08:33 +0000
committerSteve Kemp <steve@steve.org.uk>2013-02-25 17:08:33 +0000
commit14013d8db1f5f7ed35f6132a078b7d6726da2d7c (patch)
tree879eda8d5cc7a1667a93d3f154bf55221dcff73b /lib/custodian
parent33f21283a3e8dd3e7c0f6254b7a88beee940192f (diff)
Added MySQL protocol-test.
Diffstat (limited to 'lib/custodian')
-rw-r--r--lib/custodian/protocoltest/mysql.rb101
-rw-r--r--lib/custodian/protocoltests.rb3
2 files changed, 103 insertions, 1 deletions
diff --git a/lib/custodian/protocoltest/mysql.rb b/lib/custodian/protocoltest/mysql.rb
new file mode 100644
index 0000000..32867ab
--- /dev/null
+++ b/lib/custodian/protocoltest/mysql.rb
@@ -0,0 +1,101 @@
+require 'custodian/protocoltest/tcp'
+
+#
+# The MySQL-protocol test.
+#
+# This object is instantiated if the parser sees a line such as:
+#
+###
+### foo.vm.bytemark.co.uk must run mysql otherwise 'database fail'.
+###
+#
+# The specification of the port is optional.
+#
+module Custodian
+
+ module ProtocolTest
+
+ class MySQLTest < TCPTest
+
+ #
+ # Constructor
+ #
+ # Ensure we received a port to run the test against.
+ #
+ def initialize( line )
+
+ #
+ # Save the line
+ #
+ @line = line
+
+ #
+ # Save the host
+ #
+ @host = line.split( /\s+/)[0]
+
+
+ #
+ # Is this test inverted?
+ #
+ if ( line =~ /must\s+not\s+run\s+/ )
+ @inverted = true
+ else
+ @inverted = false
+ end
+
+
+ #
+ # Save the port
+ #
+ if ( line =~ /on\s+([0-9]+)/ )
+ @port = $1.dup
+ else
+ @port = 3306
+ end
+ end
+
+
+
+
+ #
+ # Allow this test to be serialized.
+ #
+ def to_s
+ return( @line )
+ end
+
+
+
+
+ #
+ # Run the test.
+ #
+ def run_test
+
+ # reset the error, in case we were previously executed.
+ @error = nil
+
+ run_test_internal( @host, @port )
+ end
+
+
+
+
+ #
+ # If the test fails then report the error.
+ #
+ def error
+ @error
+ end
+
+
+
+
+ register_test_type "mysql"
+
+
+
+ end
+ end
+end
diff --git a/lib/custodian/protocoltests.rb b/lib/custodian/protocoltests.rb
index da599c9..cd5eea5 100644
--- a/lib/custodian/protocoltests.rb
+++ b/lib/custodian/protocoltests.rb
@@ -16,6 +16,7 @@ require 'custodian/protocoltest/ftp'
require 'custodian/protocoltest/http'
require 'custodian/protocoltest/jabber'
require 'custodian/protocoltest/ldap'
+require 'custodian/protocoltest/mysql'
require 'custodian/protocoltest/ping'
require 'custodian/protocoltest/pop3'
require 'custodian/protocoltest/redis'
@@ -23,7 +24,7 @@ require 'custodian/protocoltest/rsync'
require 'custodian/protocoltest/openproxy'
require 'custodian/protocoltest/ssh'
require 'custodian/protocoltest/smtp'
-require 'custodian/protocoltest/smtprelay.rb'
+require 'custodian/protocoltest/smtprelay'
require 'custodian/protocoltest/telnet'