summaryrefslogtreecommitdiff
path: root/t
diff options
context:
space:
mode:
authorSteve Kemp <steve@steve.org.uk>2012-11-17 15:16:17 +0000
committerSteve Kemp <steve@steve.org.uk>2012-11-17 15:16:17 +0000
commite7c69a0dda13d719de97d30070f4f14345cf5ee8 (patch)
tree5b9c5366b341c04f442b547870c0845863496c07 /t
parent8cbf2b1c43b37d534529a0b69cddd69e3e65f42a (diff)
Added TCP protocol test.
Diffstat (limited to 't')
-rwxr-xr-xt/test-protocol-tcp.rb94
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