Allow for cleanup of inbox messages in tests.

This commit is contained in:
Colin Kuskie 2010-06-07 10:46:28 -07:00
parent 30e869b66d
commit 99063a3f76
2 changed files with 5 additions and 3 deletions

View file

@ -40,6 +40,7 @@ my $new_message = {
};
my $message = $inbox->addMessage($new_message,{ no_email => 1, });
WebGUI::Test->addToCleanup($message);
isa_ok($message, 'WebGUI::Inbox::Message');
ok(defined($message), 'addMessage returned a response');
@ -51,8 +52,8 @@ ok($messageId, 'messageId retrieved');
####################################
# get a message based on messageId #
####################################
$message = $inbox->getMessage($messageId);
ok($message->getId == $messageId, 'getMessage returns message object');
my $messageCopy = $inbox->getMessage($messageId);
ok($messageCopy->getId == $messageId, 'getMessage returns message object');
#########################################################
# get a list (arrayref) of messages for a specific user #
@ -133,7 +134,6 @@ note $messages->[0]->isRead;
is($inbox->getUnreadMessageCount($admin->userId), 3, '... really tracks unread messages');
END {
$session->db->write('delete from inbox where messageId = ?', [$message->getId]);
foreach my $message (@{ $inbox->getMessagesForUser($admin, 1000) } ) {
$message->setDeleted(3);
$message->delete(3);

View file

@ -799,6 +799,7 @@ were passed in. Currently able to destroy:
WebGUI::Shop::AddressBook
WebGUI::DatabaseLink
WebGUI::LDAPLink
WebGUI::Inbox::Message
Example call:
@ -887,6 +888,7 @@ Example call:
'WebGUI::Shop::Transaction' => 'delete',
'WebGUI::Shop::ShipDriver' => 'delete',
'WebGUI::Shop::PayDriver' => 'delete',
'WebGUI::Inbox::Message' => 'purge',
'WebGUI::Shop::Cart' => sub {
my $cart = shift;
my $addressBook = $cart->getAddressBook();