# Copyright (C) Extensible Service Proxy Authors

# All rights reserved.

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions

# are met:

# 1. Redistributions of source code must retain the above copyright

# notice, this list of conditions and the following disclaimer.

# 2. Redistributions in binary form must reproduce the above copyright

# notice, this list of conditions and the following disclaimer in the

# documentation and/or other materials provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND

# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE

# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL

# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS

# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF

# SUCH DAMAGE.

#

################################################################################

#

use strict;

use warnings;

################################################################################

use src::nginx::t::ApiManager; # Must be first (sets up import path to the Nginx test module)

use src::nginx::t::HttpServer;

use Test::Nginx; # Imports Nginx's test module

use Test::More; # And the test framework

use utf8;

################################################################################

# Port assignments

my $NginxPort = ApiManager::pick_port();

my $ServiceControlPort = ApiManager::pick_port();

my $GrpcServerPort = ApiManager::pick_port();

my $t = Test::Nginx->new()->has(qw/http proxy/)->plan(14);

$t->write_file('service.pb.txt',

ApiManager::get_transcoding_test_service_config(

'endpoints-transcoding-test.cloudendpointsapis.com',

"http://127.0.0.1:${ServiceControlPort}"));

$t->write_file_expand('nginx.conf', <<EOF);

%%TEST_GLOBALS%%

daemon off;

events {

worker_connections 32;

}

http {

%%TEST_GLOBALS_HTTP%%

server_tokens off;

server {

listen 127.0.0.1:${NginxPort};

server_name localhost;

location / {

endpoints {

api service.pb.txt;

on;

}

grpc_pass 127.0.0.1:${GrpcServerPort};

}

}

}

EOF

$t->run_daemon(\&service_control, $t, $ServiceControlPort, 'servicecontrol.log');

ApiManager::run_transcoding_test_server($t, 'backend.log', "127.0.0.1:${GrpcServerPort}");

is($t->waitforsocket("127.0.0.1:${ServiceControlPort}"), 1, "Service control socket ready.");

is($t->waitforsocket("127.0.0.1:${GrpcServerPort}"), 1, "GRPC test backend socket ready.");

$t->run();

is($t->waitforsocket("127.0.0.1:${NginxPort}"), 1, "Nginx socket ready.");

################################################################################

# A request with a valid UTF8 characters

my $request = <<"EOF";

{

"id" : 100,

"theme" : "\xC2\xA9\xF0\x9D\x8C\x86\xE2\x98\x83\xC2\xB6",

}

EOF

my $request_size = length($request);

my $response = ApiManager::http($NginxPort,<<EOF);

POST /shelves?key=api-key HTTP/1.0

Host: 127.0.0.1:${NginxPort}

Content-Type: application/json

Content-Length: $request_size

$request

EOF

ok(ApiManager::verify_http_json_response($response,

{'id'=>'100', 'theme' => utf8_to_wide("\xC2\xA9\xF0\x9D\x8C\x86\xE2\x98\x83\xC2\xB6")}),

"The shelf with valid UTF-8 chars was created successfully.");

# A request with invalid UTF8 characters

$request = <<"EOF";

{

"id" : 100,

"theme" : "\xC2 \xE2\x98 \xF0\x9D\x8C",

}

EOF

$request_size = length($request);

$response = ApiManager::http($NginxPort,<<EOF);

POST /shelves?key=api-key HTTP/1.0

Host: 127.0.0.1:${NginxPort}

Content-Type: application/json

Content-Length: $request_size

$request

EOF

like($response, qr/HTTP\/1\.1 400 Bad Request/, 'Got a 400.');

like($response, qr/Content-Type: application\/json/i, 'Content-type is application/json');

like($response, qr/Encountered non UTF-8 code points/i, "Got the invalid UTF-8 message");

# A request with valid UTF8 characters in query params

$response = ApiManager::http_get($NginxPort,

"/query/shelves?shelf.theme=%C2%A9%F0%9D%8C%86%E2%98%83%C2%B6&key=api-key");

ok(ApiManager::verify_http_json_response(

$response, {'shelves' => [{'id' => '100',

'theme' => utf8_to_wide("\xC2\xA9\xF0\x9D\x8C\x86\xE2\x98\x83\xC2\xB6")}]}),

'Got the "Children" shelf');

# A request with invalid UTF8 characters in query params

$response = ApiManager::http_get($NginxPort,

"/query/shelves?shelf.theme=%C2%E2%98%F0%9D%8C&key=api-key");

like($response, qr/HTTP\/1\.1 400 Bad Request/, 'Got a 400.');

like($response, qr/Content-Type: application\/json/i, 'Content-type is application/json');

like($response, qr/Encountered non UTF-8 code points/i, "Got the invalid UTF-8 message");

$t->stop_daemons();

# Verify the translated requests

my $backend_output = $t->read_file('backend.log');

my @translated_requests = split /\r\n\r\n/, $backend_output;

is (scalar @translated_requests, 2, 'Only the two valid UTF-8 successful requests went through');

ok(ApiManager::compare_json($translated_requests[0],

{'shelf' => {'id'=>'100', 'theme' => utf8_to_wide("\xC2\xA9\xF0\x9D\x8C\x86\xE2\x98\x83\xC2\xB6")}}),

"The request with UTF-8 chars in JSON was translated correctly.");

ok(ApiManager::compare_json($translated_requests[1],

{'shelf' => {'theme' => utf8_to_wide("\xC2\xA9\xF0\x9D\x8C\x86\xE2\x98\x83\xC2\xB6")}}),

"The request with UTF-8 chars in query params was translated correctly.");

################################################################################

sub service_control {

my ($t, $port, $file) = @_;

my $server = HttpServer->new($port, $t->testdir() . '/' . $file)

or die "Can't create test server socket: $!\n";

$server->on_sub('POST', '/v1/services/endpoints-transcoding-test.cloudendpointsapis.com:check', sub {

my ($headers, $body, $client) = @_;

print $client <<EOF;

HTTP/1.1 200 OK

Content-Type: application/json

Connection: close

EOF

});

$server->on_sub('POST', '/v1/services/endpoints-transcoding-test.cloudendpointsapis.com:report', sub {

my ($headers, $body, $client) = @_;

print $client <<EOF;

HTTP/1.1 200 OK

Content-Type: application/json

Connection: close

EOF

});

$server->run();

}

################################################################################

sub utf8_to_wide {

my $str = shift;

utf8::decode($str);

return $str;

}