From bd3b0de882352e0ac47a6a2bd1f6d3d325688b02 Mon Sep 17 00:00:00 2001
From: Matt Simerson <matt@tnpi.net>
Date: Sat, 20 Apr 2013 17:53:03 -0400
Subject: [PATCH] fix dmarc plugin tests

---
 t/config/plugins     | 1 +
 t/plugin_tests/dmarc | 5 ++---
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/t/config/plugins b/t/config/plugins
index 0c3ea77..7e7ce5b 100644
--- a/t/config/plugins
+++ b/t/config/plugins
@@ -59,6 +59,7 @@ rcpt_ok
 headers days 5 reject_type temp require From,Date
 domainkeys
 dkim
+dmarc
 
 # content filters
 virus/klez_filter
diff --git a/t/plugin_tests/dmarc b/t/plugin_tests/dmarc
index 14cc73e..4c8ef1c 100644
--- a/t/plugin_tests/dmarc
+++ b/t/plugin_tests/dmarc
@@ -14,7 +14,7 @@ sub register_tests {
 
     $self->register_test('test_get_organizational_domain', 2);
     $self->register_test("test_fetch_dmarc_record", 3);
-    $self->register_test("test_discover_policy", 3);
+    $self->register_test("test_discover_policy", 1);
 }
 
 sub setup_test_headers {
@@ -62,7 +62,6 @@ sub test_discover_policy {
     my $self = shift;
 
     $self->setup_test_headers();
-    my $transaction = $self->qp->transaction;
 
-    ok( $self->discover_policy( $transaction ), 'discover_policy' );
+    ok( $self->discover_policy( 'tnpi.net' ), 'discover_policy' );
 };