1 files changed, 61 insertions, 11 deletions

diff --git a/plugins/t/check_swap.t b/plugins/t/check_swap.t
index eaa81083..93e481c3 100644
--- a/plugins/t/check_swap.t
+++ b/plugins/t/check_swap.t
@@ -5,39 +5,89 @@
 #
 
 use strict;
-use Test::More tests => 14;
+use warnings;
+use Test::More tests => 35;
 use NPTest;
+use JSON;
 
-my $successOutput = '/^SWAP OK - [0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
-my $failureOutput = '/^SWAP CRITICAL - [0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
-my $warnOutput    = '/^SWAP WARNING - [0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
+my $successOutput = '/^OK.* - [0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
+my $failureOutput = '/^CRITICAL: .*- [0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
+my $warnOutput    = '/^WARNING: .*- [0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
 
+my $outputFormat = '--output-format one-line';
 my $result;
 
-$result = NPTest->testCmd( "./check_swap" );                                    # Always OK
+$result = NPTest->testCmd( "./check_swap $outputFormat" );                                      # Always OK
 cmp_ok( $result->return_code, "==", 0, "Always OK" );
 like( $result->output, $successOutput, "Right output" );
 
-$result = NPTest->testCmd( "./check_swap -w 1048576 -c 1048576" );              # 1 MB free
+$result = NPTest->testCmd( "./check_swap -w 1048576 -c 1048576 $outputFormat" );                # 1 MB free
 cmp_ok( $result->return_code, "==", 0, "At least 1MB free" );
 like( $result->output, $successOutput, "Right output" );
 
-$result = NPTest->testCmd( "./check_swap -w 1% -c 1%" );                        # 1% free
+$result = NPTest->testCmd( "./check_swap -w 1% -c 1% $outputFormat" );                  # 1% free
 cmp_ok( $result->return_code, "==", 0, 'At least 1% free' );
 like( $result->output, $successOutput, "Right output" );
 
-$result = NPTest->testCmd( "./check_swap -w 100% -c 100%" );                    # 100% (always critical)
+$result = NPTest->testCmd( "./check_swap -w 100% -c 100% $outputFormat" );                      # 100% (always critical)
 cmp_ok( $result->return_code, "==", 2, 'Get critical because not 100% free' );
 like( $result->output, $failureOutput, "Right output" );
 
-$result = NPTest->testCmd( "./check_swap -w 100% -c 1%" );                      # 100% (always warn)
+$result = NPTest->testCmd( "./check_swap -w 100% -c 1% $outputFormat" );                        # 100% (always warn)
 cmp_ok( $result->return_code, "==", 1, 'Get warning because not 100% free' );
 like( $result->output, $warnOutput, "Right output" );
 
-$result = NPTest->testCmd( "./check_swap -w 100%" );                            # 100% (single threshold, always warn)
+$result = NPTest->testCmd( "./check_swap -w 100% $outputFormat" );                              # 100% (single threshold, always warn)
 cmp_ok( $result->return_code, "==", 1, 'Get warning because not 100% free' );
 like( $result->output, $warnOutput, "Right output" );
 
-$result = NPTest->testCmd( "./check_swap -c 100%" );                            # 100% (single threshold, always critical)
+$result = NPTest->testCmd( "./check_swap -c 100% $outputFormat" );                              # 100% (single threshold, always critical)
 cmp_ok( $result->return_code, "==", 2, 'Get critical because not 100% free' );
 like( $result->output, $failureOutput, "Right output" );
+
+
+$outputFormat = '--output-format mp-test-json';
+my $output;
+my $message = '/^[0-9]+\% free \([0-9]+MiB out of [0-9]+MiB\)/';
+
+$result = NPTest->testCmd( "./check_swap $outputFormat" );                                      # Always OK
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "OK", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");
+
+$result = NPTest->testCmd( "./check_swap -w 1048576 -c 1048576 $outputFormat" );                # 1 MB free
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "OK", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");
+
+$result = NPTest->testCmd( "./check_swap -w 1% -c 1% $outputFormat" );                  # 1% free
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "OK", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");
+
+$result = NPTest->testCmd( "./check_swap -w 100% -c 100% $outputFormat" );                      # 100% (always critical)
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "CRITICAL", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");
+
+$result = NPTest->testCmd( "./check_swap -w 100% -c 1% $outputFormat" );                        # 100% (always warn)
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "WARNING", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");
+
+$result = NPTest->testCmd( "./check_swap -w 100% $outputFormat" );                              # 100% (single threshold, always warn)
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "WARNING", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");
+
+$result = NPTest->testCmd( "./check_swap -c 100% $outputFormat" );                              # 100% (single threshold, always critical)
+cmp_ok( $result->return_code, "==", 0, "Always OK" );
+$output = decode_json($result->output);
+is($output->{'state'}, "CRITICAL", "State was correct");
+like($output->{'checks'}->[0]->{'output'}, $message, "Output was correct");