t/db_dependent/selenium/authentication_2fa.t

   1 #!/usr/bin/perl
   2
   3 # This file is part of Koha.
   4 #
   5 # Koha is free software; you can redistribute it and/or modify it
   6 # under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # Koha is distributed in the hope that it will be useful, but
  11 # WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with Koha; if not, see <http://www.gnu.org/licenses>.
  17
  18 use Modern::Perl;
  19 use Test::More tests => 2;
  20
  21 use C4::Context;
  22 use Koha::AuthUtils;
  23 use Koha::Auth::TwoFactorAuth;
  24 use t::lib::Mocks;
  25 use t::lib::Selenium;
  26 use t::lib::TestBuilder;
  27
  28 my @data_to_cleanup;
  29 my $pref_value = C4::Context->preference('TwoFactorAuthentication');
  30
  31 SKIP: {
  32     eval { require Selenium::Remote::Driver; };
  33     skip "Selenium::Remote::Driver is needed for selenium tests.", 2 if $@;
  34
  35     my $builder  = t::lib::TestBuilder->new;
  36
  37     my $patron = $builder->build_object({ class => 'Koha::Patrons', value => { flags => 1 }});
  38     $patron->flags(1)->store; # superlibrarian permission
  39     my $password = Koha::AuthUtils::generate_password($patron->category);
  40     t::lib::Mocks::mock_preference( 'RequireStrongPassword', 0 );
  41     $patron->set_password({ password => $password });
  42
  43     push @data_to_cleanup, $patron, $patron->category, $patron->library;
  44
  45     my $s        = t::lib::Selenium->new({ login => $patron->userid, password => $password });
  46     my $driver   = $s->driver;
  47
  48     subtest 'Setup' => sub {
  49         plan tests => 10;
  50
  51         my $mainpage = $s->base_url . q|mainpage.pl|;
  52         $driver->get($mainpage);
  53         like( $driver->get_title, qr(Log in to Koha), 'Hitting the main page should redirect to the login form');
  54
  55         fill_login_form($s);
  56         like( $driver->get_title, qr(Koha staff interface), 'Patron with flags superlibrarian should be able to login' );
  57
  58         C4::Context->set_preference('TwoFactorAuthentication', 0);
  59         $driver->get($s->base_url . q|members/two_factor_auth.pl|);
  60         like( $driver->get_title, qr(Error 404), 'Must be redirected to 404 is the pref is off' );
  61
  62         C4::Context->set_preference('TwoFactorAuthentication', 1);
  63         $driver->get($s->base_url . q|members/two_factor_auth.pl|);
  64         like( $driver->get_title, qr(Two-factor authentication), 'Must be on the page with the pref on' );
  65
  66         is( $driver->find_element('//div[@class="two-factor-status"]')->get_text(), 'Status: Disabled', '2FA is disabled' );
  67
  68         $driver->find_element('//form[@id="two-factor-auth"]//input[@type="submit"]')->click;
  69         ok($driver->find_element('//img[@id="qr_code"]'), 'There is a QR code');
  70
  71         $s->fill_form({pin_code => 'wrong_code'});
  72         $s->submit_form;
  73         ok($driver->find_element('//div[@class="dialog error"][contains(text(), "Invalid pin code")]'));
  74         is( $patron->get_from_storage->secret, undef, 'secret is not set in DB yet' );
  75
  76         my $secret32 = $driver->find_element('//form[@id="two-factor-auth"]//input[@name="secret32"]')->get_value();
  77         my $auth = Koha::Auth::TwoFactorAuth->new({patron => $patron, secret32 => $secret32});
  78         my $code = $auth->code();
  79         $s->fill_form({pin_code => $code});
  80         $s->submit_form;
  81         is( $driver->find_element('//div[@class="two-factor-status"]')->get_text(), 'Status: Enabled', '2FA is enabled' );
  82         $patron = $patron->get_from_storage;
  83         is( $patron->secret, $secret32, 'secret is set in DB' );
  84
  85     };
  86
  87     subtest 'Login' => sub {
  88         plan tests => 19;
  89
  90         my $mainpage = $s->base_url . q|mainpage.pl|;
  91
  92         { # ok first try
  93             $driver->get($mainpage . q|?logout.x=1|);
  94             $driver->get($s->base_url . q|circ/circulation.pl?borrowernumber=|.$patron->borrowernumber);
  95             like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );
  96             $driver->capture_screenshot('selenium_failure_2.png');
  97             fill_login_form($s);
  98             like( $driver->get_title, qr(Two-factor authentication), 'Must be on the second auth screen' );
  99             is( login_error($s), undef );
 100
 101             my $auth = Koha::Auth::TwoFactorAuth->new({patron => $patron});
 102             my $code = $auth->code();
 103             $auth->clear;
 104             $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys($code);
 105             $driver->find_element('//input[@type="submit"]')->click;
 106             like( $driver->get_title, qr(Checking out to ), 'Must be redirected to the original page' );
 107         }
 108
 109         { # second try and logout
 110             $driver->get($mainpage . q|?logout.x=1|);
 111             $driver->get($s->base_url . q|circ/circulation.pl?borrowernumber=|.$patron->borrowernumber);
 112             like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );
 113             fill_login_form($s);
 114             like( $driver->get_title, qr(Two-factor authentication), 'Must be on the second auth screen' );
 115             is( login_error($s), undef );
 116             $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys('wrong_code');
 117             $driver->find_element('//input[@type="submit"]')->click;
 118             ok($driver->find_element('//div[@class="dialog error"][contains(text(), "Invalid two-factor code")]'));
 119             is( login_error($s), undef );
 120
 121             $driver->get($mainpage);
 122             like( $driver->get_title, qr(Two-factor authentication), 'Must still be on the second auth screen' );
 123             is( login_error($s), undef );
 124             $driver->find_element('//a[@id="logout"]')->click();
 125             like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );
 126             is( login_error($s), undef );
 127         }
 128
 129         { # second try and success
 130
 131             $driver->get($mainpage . q|?logout.x=1|);
 132             $driver->get($s->base_url . q|circ/circulation.pl?borrowernumber=|.$patron->borrowernumber);
 133             like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );
 134             like( login_error($s), qr(Session timed out) );
 135             fill_login_form($s);
 136             like( $driver->get_title, qr(Two-factor authentication), 'Must be on the second auth screen' );
 137             is( login_error($s), undef );
 138             $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys('wrong_code');
 139             $driver->find_element('//input[@type="submit"]')->click;
 140             ok($driver->find_element('//div[@class="dialog error"][contains(text(), "Invalid two-factor code")]'));
 141
 142             my $auth = Koha::Auth::TwoFactorAuth->new({patron => $patron});
 143             my $code = $auth->code();
 144             $auth->clear;
 145             $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys($code);
 146             $driver->find_element('//input[@type="submit"]')->click;
 147             like( $driver->get_title, qr(Checking out to ), 'Must be redirected to the original page' );
 148         }
 149     };
 150
 151     $driver->quit();
 152 };
 153
 154 END {
 155     $_->delete for @data_to_cleanup;
 156     C4::Context->set_preference('TwoFactorAuthentication', $pref_value);
 157 };
 158
 159
 160 sub login_error {
 161     my ( $s ) = @_;
 162     my $driver   = $s->driver;
 163
 164     $s->remove_error_handler;
 165     my $login_error = eval {
 166         my $elt = $driver->find_element('//div[@id="login_error"]');
 167         return $elt->get_text if $elt && $elt->id;
 168     };
 169     $s->add_error_handler;
 170     return $login_error;
 171 }
 172
 173 # Don't use the usual t::lib::Selenium->auth as we don't want the ->get($mainpage) to test the redirect
 174 sub fill_login_form {
 175     my ( $s ) = @_;
 176     $s->fill_form({ userid => $s->login, password => $s->password });
 177     $s->driver->find_element('//input[@id="submit-button"]')->click;
 178 }