rt/t/mail/digest-attributes.t

   1
   2 use warnings;
   3 use strict;
   4 use RT;
   5 use RT::Test tests => 31;
   6 my @users = qw/ emailnormal@example.com emaildaily@example.com emailweekly@example.com emailsusp@example.com /;
   7
   8 my( $ret, $msg );
   9 my $user_n = RT::User->new( RT->SystemUser );
  10 ( $ret, $msg ) = $user_n->LoadOrCreateByEmail( $users[0] );
  11 ok( $ret, "user with default email prefs created: $msg" );
  12 $user_n->SetPrivileged( 1 );
  13
  14 my $user_d = RT::User->new( RT->SystemUser );
  15 ( $ret, $msg ) = $user_d->LoadOrCreateByEmail( $users[1] );
  16 ok( $ret, "user with daily digest email prefs created: $msg" );
  17 # Set a username & password for testing the interface.
  18 $user_d->SetPrivileged( 1 );
  19 $user_d->SetPreferences($RT::System => { %{ $user_d->Preferences( $RT::System ) || {}}, EmailFrequency => 'Daily digest'});
  20
  21
  22
  23 my $user_w = RT::User->new( RT->SystemUser );
  24 ( $ret, $msg ) = $user_w->LoadOrCreateByEmail( $users[2] );
  25 ok( $ret, "user with weekly digest email prefs created: $msg" );
  26 $user_w->SetPrivileged( 1 );
  27 $user_w->SetPreferences($RT::System => { %{ $user_w->Preferences( $RT::System ) || {}}, EmailFrequency => 'Weekly digest'});
  28
  29 my $user_s = RT::User->new( RT->SystemUser );
  30 ( $ret, $msg ) = $user_s->LoadOrCreateByEmail( $users[3] );
  31 ok( $ret, "user with suspended email prefs created: $msg" );
  32 $user_s->SetPreferences($RT::System => { %{ $user_s->Preferences( $RT::System ) || {}}, EmailFrequency => 'Suspended'});
  33 $user_s->SetPrivileged( 1 );
  34
  35
  36 is(RT::Config->Get('EmailFrequency' => $user_s), 'Suspended');
  37
  38 # Make a testing queue for ourselves.
  39 my $testq = RT::Queue->new( RT->SystemUser );
  40 if( $testq->ValidateName( 'EmailDigest-testqueue' ) ) {
  41     ( $ret, $msg ) = $testq->Create( Name => 'EmailDigest-testqueue' );
  42     ok( $ret, "Our test queue is created: $msg" );
  43 } else {
  44     $testq->Load( 'EmailDigest-testqueue' );
  45     ok( $testq->id, "Our test queue is loaded" );
  46 }
  47
  48 # Allow anyone to open a ticket on the test queue.
  49 my $everyone = RT::Group->new( RT->SystemUser );
  50 ( $ret, $msg ) = $everyone->LoadSystemInternalGroup( 'Everyone' );
  51 ok( $ret, "Loaded 'everyone' group: $msg" );
  52
  53 ( $ret, $msg ) = $everyone->PrincipalObj->GrantRight( Right => 'CreateTicket',
  54                                                       Object => $testq );
  55 ok( $ret || $msg =~ /already has/, "Granted everyone CreateTicket on testq: $msg" );
  56
  57 # Make user_d an admincc for the queue.
  58 ( $ret, $msg ) = $user_d->PrincipalObj->GrantRight( Right => 'AdminQueue',
  59                                                     Object => $testq );
  60 ok( $ret || $msg =~ /already has/, "Granted dduser AdminQueue on testq: $msg" );
  61 ( $ret, $msg ) = $testq->AddWatcher( Type => 'AdminCc',
  62                              PrincipalId => $user_d->PrincipalObj->id );
  63 ok( $ret || $msg =~ /already/, "dduser added as a queue watcher: $msg" );
  64
  65 # Give the others queue rights.
  66 ( $ret, $msg ) = $user_n->PrincipalObj->GrantRight( Right => 'AdminQueue',
  67                                                     Object => $testq );
  68 ok( $ret || $msg =~ /already has/, "Granted emailnormal right on testq: $msg" );
  69 ( $ret, $msg ) = $user_w->PrincipalObj->GrantRight( Right => 'AdminQueue',
  70                                                     Object => $testq );
  71 ok( $ret || $msg =~ /already has/, "Granted emailweekly right on testq: $msg" );
  72 ( $ret, $msg ) = $user_s->PrincipalObj->GrantRight( Right => 'AdminQueue',
  73                                                     Object => $testq );
  74 ok( $ret || $msg =~ /already has/, "Granted emailsusp right on testq: $msg" );
  75
  76 # Create a ticket with To: Cc: Bcc: fields using our four users.
  77 my $id;
  78 my $ticket = RT::Ticket->new( RT->SystemUser );
  79 ( $id, $ret, $msg ) = $ticket->Create( Queue => $testq->Name,
  80                                        Requestor => [ $user_w->Name ],
  81                                        Subject => 'Test ticket for RT::Extension::EmailDigest',
  82                                        );
  83 ok( $ret, "Ticket $id created: $msg" );
  84
  85 # Make the other users ticket watchers.
  86 ( $ret, $msg ) = $ticket->AddWatcher( Type => 'Cc',
  87                       PrincipalId => $user_n->PrincipalObj->id );
  88 ok( $ret, "Added user_n as a ticket watcher: $msg" );
  89 ( $ret, $msg ) = $ticket->AddWatcher( Type => 'Cc',
  90                       PrincipalId => $user_s->PrincipalObj->id );
  91 ok( $ret, "Added user_s as a ticket watcher: $msg" );
  92
  93 my $obj;
  94 ($id, $msg, $obj ) = $ticket->Correspond(
  95         Content => "This is a ticket response for CC action" );
  96 ok( $ret, "Transaction created: $msg" );
  97
  98 # Get the deferred notifications that should result.  Should be two for
  99 # email daily, and one apiece for emailweekly and emailsusp.
 100 my @notifications;
 101
 102 my $txns = RT::Transactions->new( RT->SystemUser );
 103 $txns->LimitToTicket( $ticket->id );
 104 my( $c_daily, $c_weekly, $c_susp ) = ( 0, 0, 0 );
 105 while( my $txn = $txns->Next ) {
 106     my @daily_rcpt = $txn->DeferredRecipients( 'daily' );
 107     my @weekly_rcpt = $txn->DeferredRecipients('weekly' );
 108     my @susp_rcpt = $txn->DeferredRecipients(  'susp' );
 109
 110     $c_daily++ if @daily_rcpt;
 111     $c_weekly++ if @weekly_rcpt;
 112     $c_susp++ if @susp_rcpt;
 113
 114     # If the transaction has content...
 115     if( $txn->ContentObj ) {
 116         # ...none of the deferred folk should be in the header.
 117         my $headerstr = $txn->ContentObj->Headers;
 118         foreach my $rcpt( @daily_rcpt, @weekly_rcpt, @susp_rcpt ) {
 119             ok( $headerstr !~ /$rcpt/, "Deferred recipient $rcpt not found in header" );
 120         }
 121     }
 122 }
 123
 124 # Finally, check to see that we got the correct number of each sort of
 125 # deferred recipient.
 126 is( $c_daily, 2, "correct number of daily-sent messages" );
 127 is( $c_weekly, 2, "correct number of weekly-sent messages" );
 128 is( $c_susp, 1, "correct number of suspended messages" );
 129
 130
 131
 132
 133
 134 # Now let's actually run the daily and weekly digest tool to make sure we generate those
 135
 136 # the first time get the content
 137 email_digest_like( '--mode daily --print', qr/in the last day/ );
 138 # The second time run it for real so we make sure that we get RT to mark the txn as sent
 139 email_digest_like( '--mode daily --verbose', qr/maildaily\@/ );
 140 # now we should have nothing to do, so no content.
 141 email_digest_like( '--mode daily --print', '' );
 142
 143 # the first time get the content
 144 email_digest_like( '--mode weekly --print', qr/in the last seven days/ );
 145 # The second time run it for real so we make sure that we get RT to mark the txn as sent
 146 email_digest_like( '--mode weekly --verbose', qr/mailweekly\@/ );
 147 # now we should have nothing to do, so no content.
 148 email_digest_like( '--mode weekly --print', '' );
 149
 150 sub email_digest_like {
 151     my $arg = shift;
 152     my $pattern = shift;
 153
 154     local $Test::Builder::Level = $Test::Builder::Level + 1;
 155
 156     my $perl = $^X . ' ' . join ' ', map { "-I$_" } @INC;
 157     open my $digester, "-|", "$perl $RT::SbinPath/rt-email-digest $arg";
 158     my @results = <$digester>;
 159     my $content = join '', @results;
 160     if ( ref $pattern && ref $pattern eq 'Regexp' ) {
 161         like($content, $pattern);
 162     }
 163     else {
 164         is( $content, $pattern );
 165     }
 166     close $digester;
 167 }