oqapi/webgui
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move Auth cleanup into a addToCleanup sub.

Browse source
This commit is contained in:
Colin Kuskie 2010-10-05 11:30:17 -07:00
parent ab2be62157
commit d93238d53c
1 changed files with 19 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

41
t/Auth.t
View file

@ -49,6 +49,25 @@ $auth = WebGUI::Auth->new( $session, $AUTH_METHOD );
my $username = $session->id->generate; my $username = $session->id->generate;
push @cleanupUsernames, $username; push @cleanupUsernames, $username;
$output = $auth->createAccountSave( $username, { }, "PASSWORD" ); $output = $auth->createAccountSave( $username, { }, "PASSWORD" );
WebGUI::Test->addToCleanup(sub {
for my $username ( @cleanupUsernames ) {
# We don't create actual, real users, so we have to cleanup by hand
my $userId = $session->db->quickScalar(
"SELECT userId FROM users WHERE username=?",
[ $username ]
);
my @tableList
= qw{authentication users userProfileData groupings inbox userLoginLog};
for my $table ( @tableList ) {
$session->db->write(
"DELETE FROM $table WHERE userId=?",
[ $userId ]
);
}
}
});
is( is(
$session->http->getRedirectLocation, 'REDIRECT_URL', $session->http->getRedirectLocation, 'REDIRECT_URL',
@ -83,25 +102,3 @@ is $output, undef, 'login returns undef when showMessageOnLogin is false';
# Session Cleanup # Session Cleanup
$session->{_request} = $oldRequest; $session->{_request} = $oldRequest;
#----------------------------------------------------------------------------
# Cleanup
END {
for my $username ( @cleanupUsernames ) {
# We don't create actual, real users, so we have to cleanup by hand
my $userId = $session->db->quickScalar(
"SELECT userId FROM users WHERE username=?",
[ $username ]
);
my @tableList
= qw{authentication users userProfileData groupings inbox userLoginLog};
for my $table ( @tableList ) {
$session->db->write(
"DELETE FROM $table WHERE userId=?",
[ $userId ]
);
}
}
}