--- /dev/null
+#!/usr/bin/env perl
+
+# This file is part of Koha.
+#
+# Koha is free software; you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation; either version 3 of the License, or (at your option) any later
+# version.
+#
+# Koha is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE. See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with Koha; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+use Modern::Perl;
+
+use Test::More tests => 5;
+use Test::Mojo;
+use Test::Warn;
+
+use t::lib::TestBuilder;
+use t::lib::Mocks;
+
+use Data::Printer colored => 1;
+
+use C4::Auth;
+use Koha::Cities;
+use Koha::Database;
+
+my $schema = Koha::Database->new->schema;
+my $builder = t::lib::TestBuilder->new;
+# FIXME: sessionStorage defaults to mysql, but it seems to break transaction handling
+# this affects the other REST api tests
+t::lib::Mocks::mock_preference( 'SessionStorage', 'tmp' );
+
+my $remote_address = '127.0.0.1';
+my $t = Test::Mojo->new('Koha::REST::V1');
+
+subtest 'list() tests' => sub {
+
+ plan tests => 19;
+
+ $schema->storage->txn_begin;
+
+ Koha::Cities->search->delete;
+ my ( $borrowernumber, $session_id ) =
+ create_user_and_session( { authorized => 0 } );
+
+ ## Authorized user tests
+ # No cities, so empty array should be returned
+ my $tx = $t->ua->build_tx( GET => '/api/v1/cities' );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->json_is( [] );
+
+ my $city_country = 'France';
+ my $city = $builder->build(
+ { source => 'City', value => { city_country => $city_country } } );
+
+ # One city created, should get returned
+ $tx = $t->ua->build_tx( GET => '/api/v1/cities' );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->json_is( [$city] );
+
+ my $another_city = $builder->build(
+ { source => 'City', value => { city_country => $city_country } } );
+ my $city_with_another_country = $builder->build( { source => 'City' } );
+
+ # Two cities created, they should both be returned
+ $tx = $t->ua->build_tx( GET => '/api/v1/cities' );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)
+ ->json_is( [ $city, $another_city, $city_with_another_country ] );
+
+ # Filtering works, two cities sharing city_country
+ $tx =
+ $t->ua->build_tx( GET => "/api/v1/cities?city_country=" . $city_country );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->json_is( [ $city, $another_city ] );
+
+ $tx = $t->ua->build_tx(
+ GET => "/api/v1/cities?city_name=" . $city->{city_name} );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->json_is( [$city] );
+
+ $tx = $t->ua->build_tx( GET => '/api/v1/cities?city_blah=blah' );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+
+ warning_like {
+ $t->request_ok($tx)->status_is(500)
+ ->json_like( '/error' => qr/Unknown column/ );
+ }
+ qr/Unknown column/, 'Wrong parameters raise warnings';
+
+ $schema->storage->txn_rollback;
+};
+
+subtest 'get() tests' => sub {
+
+ plan tests => 6;
+
+ $schema->storage->txn_begin;
+
+ my $city = $builder->build( { source => 'City' } );
+ my ( $borrowernumber, $session_id ) =
+ create_user_and_session( { authorized => 0 } );
+
+ my $tx = $t->ua->build_tx( GET => "/api/v1/cities/" . $city->{cityid} );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->json_is($city);
+
+ my $non_existent_id = $city->{cityid} + 1;
+ $tx = $t->ua->build_tx( GET => "/api/v1/cities/" . $non_existent_id );
+ $tx->req->cookies( { name => 'CGISESSID', value => $session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(404)
+ ->json_is( '/error' => 'City not found' );
+
+ $schema->storage->txn_rollback;
+};
+
+subtest 'add() tests' => sub {
+
+ plan tests => 10;
+
+ $schema->storage->txn_begin;
+
+ my ( $unauthorized_borrowernumber, $unauthorized_session_id ) =
+ create_user_and_session( { authorized => 0 } );
+ my ( $authorized_borrowernumber, $authorized_session_id ) =
+ create_user_and_session( { authorized => 1 } );
+ my $city = {
+ city_name => "City Name",
+ city_state => "City State",
+ city_zipcode => "City Zipcode",
+ city_country => "City Country"
+ };
+
+ # Unauthorized attempt to write
+ my $tx = $t->ua->build_tx( POST => "/api/v1/cities/" => json => $city );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $unauthorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(403);
+
+ # Authorized attempt to write
+ $tx = $t->ua->build_tx( POST => "/api/v1/cities/" => json => $city );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)
+ ->json_is( '/city_name' => $city->{city_name} )
+ ->json_is( '/city_state' => $city->{city_state} )
+ ->json_is( '/city_zipcode' => $city->{city_zipcode} )
+ ->json_is( '/city_country' => $city->{city_country} );
+
+ my $city_with_invalid_field = {
+ city_blah => "City Blah",
+ city_state => "City State",
+ city_zipcode => "City Zipcode",
+ city_country => "City Country"
+ };
+
+ # Authorized attempt to write invalid data
+ $tx = $t->ua->build_tx(
+ POST => "/api/v1/cities/" => json => $city_with_invalid_field );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(500);
+
+ $schema->storage->txn_rollback;
+};
+
+subtest 'update() tests' => sub {
+
+ plan tests => 10;
+
+ $schema->storage->txn_begin;
+
+ my ( $unauthorized_borrowernumber, $unauthorized_session_id ) =
+ create_user_and_session( { authorized => 0 } );
+ my ( $authorized_borrowernumber, $authorized_session_id ) =
+ create_user_and_session( { authorized => 1 } );
+
+ my $city_id = $builder->build( { source => 'City' } )->{cityid};
+
+ # Unauthorized attempt to update
+ my $tx = $t->ua->build_tx( PUT => "/api/v1/cities/$city_id" => json =>
+ { city_name => 'New unauthorized name change' } );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $unauthorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(403);
+
+ $tx = $t->ua->build_tx(
+ PUT => "/api/v1/cities/$city_id" => json => { city_name => 'New name' }
+ );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->json_is( '/city_name' => 'New name' );
+
+ $tx = $t->ua->build_tx(
+ PUT => "/api/v1/cities/$city_id" => json => { city_blah => 'New blah' }
+ );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(500)
+ ->json_is( '/error' => "No method city_blah for Koha::City" );
+
+ my $non_existent_id = $city_id + 1;
+ $tx = $t->ua->build_tx( PUT => "/api/v1/cities/$non_existent_id" => json =>
+ { city_name => 'New name' } );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(404);
+
+ $schema->storage->txn_rollback;
+};
+
+subtest 'delete() tests' => sub {
+
+ plan tests => 7;
+
+ $schema->storage->txn_begin;
+
+ my ( $unauthorized_borrowernumber, $unauthorized_session_id ) =
+ create_user_and_session( { authorized => 0 } );
+ my ( $authorized_borrowernumber, $authorized_session_id ) =
+ create_user_and_session( { authorized => 1 } );
+
+ my $city_id = $builder->build( { source => 'City' } )->{cityid};
+
+ # Unauthorized attempt to update
+ my $tx = $t->ua->build_tx( DELETE => "/api/v1/cities/$city_id" );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $unauthorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(403);
+
+ $tx = $t->ua->build_tx( DELETE => "/api/v1/cities/$city_id" );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(200)->content_is('');
+
+ $tx = $t->ua->build_tx( DELETE => "/api/v1/cities/$city_id" );
+ $tx->req->cookies(
+ { name => 'CGISESSID', value => $authorized_session_id } );
+ $tx->req->env( { REMOTE_ADDR => $remote_address } );
+ $t->request_ok($tx)->status_is(404);
+
+ $schema->storage->txn_rollback;
+};
+
+sub create_user_and_session {
+
+ my $args = shift;
+ my $flags = ( $args->{authorized} ) ? $args->{authorized} : 0;
+ my $dbh = C4::Context->dbh;
+
+ my $user = $builder->build(
+ {
+ source => 'Borrower',
+ value => {
+ flags => $flags
+ }
+ }
+ );
+
+ # Create a session for the authorized user
+ my $session = C4::Auth::get_session('');
+ $session->param( 'number', $user->{borrowernumber} );
+ $session->param( 'id', $user->{userid} );
+ $session->param( 'ip', '127.0.0.1' );
+ $session->param( 'lasttime', time() );
+ $session->flush;
+
+ if ( $args->{authorized} ) {
+ $dbh->do( "
+ INSERT INTO user_permissions (borrowernumber,module_bit,code)
+ VALUES (?,3,'parameters_remaining_permissions')", undef,
+ $user->{borrowernumber} );
+ }
+
+ return ( $user->{borrowernumber}, $session->id );
+}
+
+1;