From 3d3ee6446a71cfff6b29ffe95cc08d86e8e85388 Mon Sep 17 00:00:00 2001
From: Steve Kemp <steve@steve.org.uk>
Date: Mon, 25 Feb 2013 17:08:33 +0000
Subject:   Added MySQL protocol-test.

---
 lib/custodian/protocoltest/mysql.rb | 101 ++++++++++++++++++++++++++++++++++++
 lib/custodian/protocoltests.rb      |   3 +-
 2 files changed, 103 insertions(+), 1 deletion(-)
 create mode 100644 lib/custodian/protocoltest/mysql.rb

(limited to 'lib/custodian')

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'
 
 
-- 
cgit v1.2.1