Ensure we have a URL

2 files changed, 21 insertions, 1 deletions

diff --git a/worker/tests/http.rb b/worker/tests/http.rb
index f1b5843..4f2cf0b 100755
--- a/worker/tests/http.rb
+++ b/worker/tests/http.rb
@@ -42,6 +42,16 @@ class HTTPTest
 
     @error = ""
 
+
+    #
+    # Ensure we had a URL
+    #
+    if ( @test_data[:target_host].nil? )
+      @error = "Missing URL for the test"
+      return false
+    end
+
+
     #
     #  Do the fetch, if this success then we'll have the
     # @status + @text setup
diff --git a/worker/tests/https.rb b/worker/tests/https.rb
index f1b5843..2575a35 100755
--- a/worker/tests/https.rb
+++ b/worker/tests/https.rb
@@ -6,7 +6,7 @@ require 'uri'
 
 
 
-class HTTPTest
+class HTTPSTest
 
   #
   # Data passed from the JSON hash.
@@ -42,6 +42,16 @@ class HTTPTest
 
     @error = ""
 
+
+    #
+    # Ensure we had a URL
+    #
+    if ( @test_data[:target_host].nil? )
+      @error = "Missing URL for the test"
+      return false
+    end
+
+
     #
     #  Do the fetch, if this success then we'll have the
     # @status + @text setup