t/lib/Mocks.pm

   1 package t::lib::Mocks;
   2
   3 use Modern::Perl;
   4 use C4::Context;
   5
   6 use Koha::Schema;
   7 use Test::MockModule;
   8
   9 my %configs;
  10 sub mock_config {
  11     my $context = new Test::MockModule('C4::Context');
  12     my ( $conf, $value ) = @_;
  13     $configs{$conf} = $value;
  14     $context->mock('config', sub {
  15         my ( $self, $conf ) = @_;
  16         if ( exists $configs{$conf} ) {
  17             return $configs{$conf}
  18         } else {
  19             my $method = $context->original('config');
  20             return $method->($self, $conf);
  21         }
  22     });
  23 }
  24
  25 my %preferences;
  26 sub mock_preference {
  27     my $context = new Test::MockModule('C4::Context');
  28     my ( $pref, $value ) = @_;
  29     $preferences{$pref} = $value;
  30     $context->mock('preference', sub {
  31         my ( $self, $pref ) = @_;
  32         if ( exists $preferences{$pref} ) {
  33             return $preferences{$pref}
  34         } else {
  35             my $method = $context->original('preference');
  36             return $method->($self, $pref);
  37         }
  38     });
  39 }
  40
  41 sub mock_dbh {
  42     our $context = new Test::MockModule('Koha::Database');
  43     $context->mock( '_new_schema', sub {
  44         my $dbh = Koha::Schema->connect( 'DBI:Mock:', '', '' );
  45         return $dbh;
  46     } );
  47     return $context;
  48 }
  49
  50 1;