tests: move ipv6 detection into NPTest module

2 files changed, 18 insertions, 18 deletions

diff --git a/plugins/t/check_dig.t b/plugins/t/check_dig.t
index 1ab4b426..d447f084 100644
--- a/plugins/t/check_dig.t
+++ b/plugins/t/check_dig.t
@@ -6,11 +6,16 @@
 
 use strict;
 use Test::More;
-use NPTest;
 
-plan skip_all => "check_dig not compiled" unless (-x "check_dig");
 
-plan tests => 16;
+use vars qw($tests $has_ipv6);
+BEGIN {
+    plan skip_all => "check_dig not compiled" unless (-x "check_dig");
+    use NPTest;
+    $has_ipv6 = NPTest::has_ipv6();
+    $tests = $has_ipv6 ? 16 : 14;
+    plan tests => $tests;
+}
 
 my $successOutput = '/DNS OK - [\.0-9]+ seconds? response time/';
 
@@ -73,10 +78,6 @@ SKIP: {
 	cmp_ok( $res->return_code, '==', 0, "Found $hostname_valid on $dns_server");
 	like  ( $res->output, $successOutput, "Output OK for IPv4" );
 
-	$res = NPTest->testCmd("./check_dig -H $dns_server -l $hostname_valid  -t 5 -6");
-	cmp_ok( $res->return_code, '==', 0, "Found $hostname_valid on $dns_server");
-	like  ( $res->output, $successOutput, "Output OK for IPv6" );
-
 	$res = NPTest->testCmd("./check_dig -H $dns_server -l $hostname_valid -a $hostname_valid_ip -t 5");
 	cmp_ok( $res->return_code, '==', 0, "Got expected address");
 
@@ -89,4 +90,9 @@ SKIP: {
 	cmp_ok( $res->return_code, '==', 0, "Got expected fqdn");
 	like  ( $res->output, $successOutput, "Output OK");
 
+    if($has_ipv6) {
+	    $res = NPTest->testCmd("./check_dig -H $dns_server -l $hostname_valid  -t 5 -6");
+	    cmp_ok( $res->return_code, '==', 0, "Found $hostname_valid on $dns_server");
+	    like  ( $res->output, $successOutput, "Output OK for IPv6" );
+    }
 }
diff --git a/plugins/t/check_tcp.t b/plugins/t/check_tcp.t
index 0e6a9641..abb16aec 100644
--- a/plugins/t/check_tcp.t
+++ b/plugins/t/check_tcp.t
@@ -6,19 +6,13 @@
 
 use strict;
 use Test;
-use NPTest;
 
-use vars qw($tests);
-my $has_ipv6;
+use vars qw($tests $has_ipv6);
 BEGIN {
-  $tests = 11;
-  # do we have ipv6
-  `ping6 -c 1 2a02:2e0:3fe:100::7 2>&1`;
-  if($? == 0) {
-    $has_ipv6 = 1;
-    $tests += 3;
-  }
-  plan tests => $tests;
+    use NPTest;
+    $has_ipv6 = NPTest::has_ipv6();
+    $tests = $has_ipv6 ? 14 : 11;
+    plan tests => $tests;
 }