From 5ed5667803eb44780cdbc5e4183c1f55daaf2555 Mon Sep 17 00:00:00 2001
From: Steve Kemp <steve@steve.org.uk>
Date: Fri, 23 Nov 2012 16:03:27 +0000
Subject:   Test that inverted() works for each protocol test we know about.

---
 lib/custodian/protocoltest/dns.rb  | 8 ++++++++
 lib/custodian/protocoltest/http.rb | 9 +++++++++
 2 files changed, 17 insertions(+)

(limited to 'lib/custodian')

diff --git a/lib/custodian/protocoltest/dns.rb b/lib/custodian/protocoltest/dns.rb
index 6c5e8db..d1c6ee2 100644
--- a/lib/custodian/protocoltest/dns.rb
+++ b/lib/custodian/protocoltest/dns.rb
@@ -33,6 +33,14 @@ module Custodian
         #
         @line = line
 
+        #
+        # Is this test inverted?
+        #
+        if ( line =~ /must\s+not\s+run\s+/ )
+          @inverted = true
+        else
+          @inverted = false
+        end
       end
 
 
diff --git a/lib/custodian/protocoltest/http.rb b/lib/custodian/protocoltest/http.rb
index 1b7f442..133fdaf 100644
--- a/lib/custodian/protocoltest/http.rb
+++ b/lib/custodian/protocoltest/http.rb
@@ -47,6 +47,15 @@ module Custodian
           raise ArgumentError, "The target wasn't an URL"
         end
 
+        #
+        # Is this test inverted?
+        #
+        if ( line =~ /must\s+not\s+run\s+/ )
+          @inverted = true
+        else
+          @inverted = false
+        end
+
       end
 
 
-- 
cgit v1.2.1