=head3 keyed_on_code

Return all restriction types as a hashref keyed on the code

=cut

sub keyed_on_code {

    my ( $self ) = @_;

    my @all = $self->_resultset()->search();

    my $out = {};

    for my $r( @all ) {

        my %col = $r->get_columns;

        $out->{$r->code} = \%col;

    }

    return $out;

}

#!/usr/bin/perl

use Modern::Perl;

use C4::Context;

use Koha::Database;

use t::lib::TestBuilder;

use Test::More tests => 2;

my $schema = Koha::Database->new->schema;

$schema->storage->txn_begin;

my $dbh     = C4::Context->dbh;

my $builder = t::lib::TestBuilder->new;

use_ok('Koha::Patron::Restriction::Types');

$dbh->do(q|DELETE FROM borrower_debarments|);

$dbh->do(q|DELETE FROM restriction_types|);

$builder->build(

    {

        source => 'RestrictionType',

        value  => {

            code         => 'ONE',

            display_text => 'One',

            is_system    => 1,

            is_default   => 0

        }

    }

);

$builder->build(

    {

        source => 'RestrictionType',

        value  => {

            code         => 'TWO',

            display_text => 'Two',

            is_system    => 1,

            is_default   => 1

        }

    }

);

# keyed_on_code

my $keyed     = Koha::Patron::Restriction::Types->keyed_on_code;

my $expecting = {

    ONE => {

        code         => 'ONE',

        display_text => 'One',

        is_system    => 1,

        is_default   => 0

    },

    TWO => {

        code         => 'TWO',

        display_text => 'Two',

        is_system    => 1,

        is_default   => 1

    }

};

is_deeply( $keyed, $expecting, 'keyed_on_code returns correctly' );

$schema->storage->txn_rollback;

-