thrift/lib/perl/test/processor.t
2017-03-30 17:09:04 -04:00

105 lines
2.8 KiB
Perl

#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
#
use Test::More tests => 2;
use strict;
use warnings;
use Thrift::BinaryProtocol;
use Thrift::MemoryBuffer;
use Thrift::MessageType;
use ThriftTest::ThriftTest;
use ThriftTest::Types;
use Data::Dumper;
my $buffer = Thrift::MemoryBuffer->new(1024);
my $protocol = Thrift::BinaryProtocol->new($buffer);
my $client = ThriftTest::ThriftTestClient->new($protocol);
$buffer->open();
$client->send_testString("foo");
$client->{seqid}++;
$client->send_testString("bar");
my $client_command_binary = $buffer->getBuffer;
$buffer->resetBuffer;
# Process by server
my $server_output_binary;
{
my $protocol_factory = Thrift::BinaryProtocolFactory->new();
my $input_buffer = Thrift::MemoryBuffer->new();
$input_buffer->write($client_command_binary);
my $input_protocol = $protocol_factory->getProtocol($input_buffer);
my $output_buffer = Thrift::MemoryBuffer->new();
my $output_protocol = $protocol_factory->getProtocol($output_buffer);
my $processor = ThriftTest::ThriftTestProcessor->new( My::ThriftTest->new() );
my $result = $processor->process($input_protocol, $output_protocol);
print "process resulted in $result\n";
$result = $processor->process($input_protocol, $output_protocol);
print "process resulted in $result\n";
$server_output_binary = $output_buffer->getBuffer();
}
$buffer->write($server_output_binary);
foreach my $val (("got foo","got bar")){
my ($function_name, $message_type, $sequence_id);
$protocol->readMessageBegin(\$function_name, \$message_type, \$sequence_id);
print " $function_name, $message_type, $sequence_id\n";
if ($message_type == Thrift::TMessageType::EXCEPTION) {
die;
}
my $result = ThriftTest::ThriftTest_testString_result->new();
$result->read($protocol);
$protocol->readMessageEnd();
is($result->success(),$val);
}
package My::ThriftTest;
use strict;
use warnings;
use Data::Dumper;
sub new {
my $class = shift;
return bless {}, $class;
}
sub testString {
my ($self, $string) = @_;
print __PACKAGE__ . "->testString()\n";
return "got ".$string;
}