Added TCP protocol test.

1 files changed, 94 insertions, 0 deletions

diff --git a/t/test-protocol-tcp.rb b/t/test-protocol-tcp.rb
new file mode 100755
index 0000000..655b6a1
--- /dev/null
+++ b/t/test-protocol-tcp.rb
@@ -0,0 +1,94 @@
+#!/usr/bin/ruby -I../lib/ -Ilib/
+
+
+require 'test/unit'
+require 'custodian/protocol-tests/tcp.rb'
+
+
+
+
+#
+# Unit test for the TCP-protocol probe.
+#
+class TestTCPProtocolProbe < Test::Unit::TestCase
+
+  #
+  # Create the test suite environment: NOP.
+  #
+  def setup
+  end
+
+  #
+  # Destroy the test suite environment: NOP.
+  #
+  def teardown
+  end
+
+
+  #
+  #  Test we can create a new TCPTest object.
+  #
+  def test_init
+    test_data_good = {
+      "target_host" => "smtp.bytemark.co.uk",
+      "test_type"   => "tcp",
+      "banner"      => 220,
+      "verbose"     => 1,
+      "test_port"   => 25,
+      "test_alert"  => "SMTP service down",
+    }
+
+    #
+    # Missing a port number
+    #
+    test_data_bad_one = {
+      "target_host" => "smtp.bytemark.co.uk",
+      "test_type"   => "tcp",
+      "banner"      => 220,
+      "verbose"     => 1,
+      "test_alert"  => "SMTP service down",
+    }
+
+    #
+    #  Missing a target host to probe
+    #
+    test_data_bad_two = {
+      "test_type"   => "tcp",
+      "banner"      => 220,
+      "verbose"     => 1,
+      "test_port"   => 25,
+      "test_alert"  => "SMTP service down",
+    }
+
+
+    #
+    #  Create a new FTPTest object.  This should succeed
+    #
+    good = TCPTest.new( test_data_good )
+    assert( good )
+
+    #
+    #  There will be no error setup
+    #
+    assert( good.error().nil? )
+
+    #
+    #  Now create a probe with a missing port.
+    #
+    assert_raise ArgumentError do
+      bad = TCPTest.new( test_data_bad_one )
+    end
+
+
+    #
+    #  Now create a probe with a missing host.
+    #
+    assert_raise ArgumentError do
+      bad = TCPTest.new( test_data_bad_two )
+    end
+
+  end
+
+
+
+end