1 files changed, 37 insertions, 0 deletions
diff --git a/t/utils/email.t b/t/utils/email.t
new file mode 100644
index 000000000..9e556a865
--- /dev/null
+++ b/t/utils/email.t
@@ -0,0 +1,37 @@
+#!/usr/bin/env perl
+
+use strict;
+use warnings;
+use Test::More;
+use Test::MockModule;
+
+use Utils::Email;
+
+my $resolver = Test::MockModule->new('Net::DNS::Resolver');
+$resolver->mock('send', sub {
+    my ($self, $domain, $type) = @_;
+    my @rrs;
+    is $type, 'TXT';
+    if ($domain eq '_dmarc.yahoo.com') {
+        @rrs = (
+            Net::DNS::RR->new(name => '_dmarc.yahoo.com', type => 'TXT', txtdata => 'p=reject'),
+            Net::DNS::RR->new(name => '_dmarc.yahoo.com', type => 'A'),
+        );
+    } elsif ($domain eq 'cname.example.com') {
+        @rrs = Net::DNS::RR->new(name => 'cname.example.com', type => 'TXT', txtdata => 'p=none');
+    } else {
+        @rrs = Net::DNS::RR->new(name => '_dmarc.example.net', type => 'CNAME', cname => 'cname.example.com');
+    }
+    my $pkt = Net::DNS::Packet->new;
+    push @{$pkt->{answer}}, @rrs;
+    return $pkt;
+});
+
+is Utils::Email::test_dmarc('BAD'), undef;
+is Utils::Email::test_dmarc('test@yahoo.com'), 1;
+is Utils::Email::test_dmarc('test@example.net'), undef;
+
+is Utils::Email::same_domain(['test@example.net', ''], [ ['to@example.net', ''], ['to@example.com', ''] ]), 1;
+is Utils::Email::same_domain(['test@example.org', ''], [ ['to@example.net', ''] ]), '';
+
+done_testing();