#!/usr/bin/perl -I../lib

use strict;
use warnings;
use Test::Simple tests => 8;

use Mail::DKIM::Signer;
use Mail::DKIM::DkSignature;

# The main purpose of this set of tests is to ensure that I am aware
# whenever a change in my code causes the generated signature to change
# for DomainKeys signatures. Generally this should never occur for a
# change in DomainKeys code. (In contrast to DKIM, where the generated
# signature gets included in the signature, so any changes to the format
# of the signature will cause the hash to change.)
#

{
    my %sig_args;
    my $policyfn = sub {
        my $signer = shift;
        if ( $sig_args{Headers} && $sig_args{Headers} eq "*" ) {
            $sig_args{Headers} = $signer->headers;
        }
        $signer->add_signature( Mail::DKIM::DkSignature->new(%sig_args) );
        return;
    };
    my $tdir         = -f "t/test.key" ? "t" : ".";
    my $keyfile      = "$tdir/test.key";
    my $sample_email = <<END_OF_SAMPLE;
From: jason <jlong\@messiah.edu>
Subject: hi there
Comment: what is a comment

this is a sample message
END_OF_SAMPLE
    $sample_email =~ s/\n/\015\012/gs;

    sub sign_sample_using_args {
        %sig_args = (
            Algorithm => "rsa-sha1",
            Selector  => "test8",
            Domain    => "messiah.edu",
            @_,
        );
        my $dkim = Mail::DKIM::Signer->new(
            Policy  => $policyfn,
            KeyFile => $keyfile,
        );
        $dkim->PRINT($sample_email);
        $dkim->CLOSE;

        my $signature = $dkim->signature;
        return $signature;
    }
}

my $signature;
$signature = sign_sample_using_args( Method => "simple", );
ok( $signature, "signature() works" );
print "# " . $signature->as_string . "\n";
ok( $signature->data =~ /^TL93PzPvedAijHChCAt/, "got expected signature" );

$signature = sign_sample_using_args(
    Method  => "simple",
    Headers => "*",
);
ok( $signature, "signature() works" );
print "# " . $signature->as_string . "\n";
ok( $signature->data =~ /^n\+qfVkhQPch80atOC7/, "got expected signature" );

$signature = sign_sample_using_args( Method => "nofws", );
ok( $signature, "signature() works" );
print "# " . $signature->as_string . "\n";
ok( $signature->data =~ /^JWzIzvCZBIYnoMebzKU/, "got expected signature" );

$signature = sign_sample_using_args(
    Method  => "nofws",
    Headers => "*",
);
ok( $signature, "signature() works" );
print "# " . $signature->as_string . "\n";
ok( $signature->data =~ /^fkDC8iF/, "got expected signature" );