Ensure we have a target & a port.

1 files changed, 19 insertions, 0 deletions

diff --git a/worker/tests/ftp.rb b/worker/tests/ftp.rb
index f5eccbb..63b6714 100755
--- a/worker/tests/ftp.rb
+++ b/worker/tests/ftp.rb
@@ -28,6 +28,25 @@ class FTPTest
   #
   def initialize( data )
     @test_data = data
+    @error     = nil
+
+
+    #
+    # Ensure we have a host to probe
+    #
+    if ( @test_data["target_host"].nil? )
+      @error = "Missing target for the test."
+      raise ArgumentError, @error
+    end
+
+    #
+    # Ensure we have a port to test.
+    #
+    if ( @test_data["test_port"].nil? )
+      @error = "Missing port for the test."
+      raise ArgumentError, @error
+    end
+
   end