blob: df0b4c572f53b9fc6a2a6dbbc6d8bb0a1736fec4 [file] [log] [blame]
(*
* 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.
*)
unit TestClient;
{$I ../src/Thrift.Defines.inc}
{.$DEFINE StressTest} // activate to stress-test the server with frequent connects/disconnects
{.$DEFINE PerfTest} // activate the performance test
{$DEFINE Exceptions} // activate the exceptions test (or disable while debugging)
{$if CompilerVersion >= 28}
{$DEFINE SupportsAsync}
{$ifend}
{$WARN SYMBOL_PLATFORM OFF} // Win32Check
interface
uses
Windows, SysUtils, Classes, Math, ComObj, ActiveX,
{$IFDEF SupportsAsync} System.Threading, {$ENDIF}
DateUtils,
Generics.Collections,
TestConstants,
ConsoleHelper,
PerfTests,
Thrift,
Thrift.Protocol.Compact,
Thrift.Protocol.JSON,
Thrift.Protocol,
Thrift.Transport.Pipes,
Thrift.Transport.WinHTTP,
Thrift.Transport.MsxmlHTTP,
Thrift.Transport,
Thrift.Stream,
Thrift.Test,
Thrift.WinHTTP,
Thrift.Utils,
Thrift.Configuration,
Thrift.Collections;
type
TThreadConsole = class
private
FThread : TThread;
public
procedure Write( const S : string);
procedure WriteLine( const S : string);
constructor Create( AThread: TThread);
end;
TTestSetup = record
protType : TKnownProtocol;
endpoint : TEndpointTransport;
layered : TLayeredTransports;
useSSL : Boolean; // include where appropriate (TLayeredTransport?)
host : string;
port : Integer;
sPipeName : string;
hAnonRead, hAnonWrite : THandle;
end;
TClientThread = class( TThread )
private type
TTestGroup = (
test_Unknown,
test_BaseTypes,
test_Structs,
test_Containers,
test_Exceptions
// new values here
);
TTestGroups = set of TTestGroup;
TTestSize = (
Empty, // Edge case: the zero-length empty binary
Normal, // Fairly small array of usual size (256 bytes)
ByteArrayTest, // THRIFT-4454 Large writes/reads may cause range check errors in debug mode
PipeWriteLimit, // THRIFT-4372 Pipe write operations across a network are limited to 65,535 bytes per write.
FifteenMB // quite a bit of data, but still below the default max frame size
);
private
FSetup : TTestSetup;
FTransport : ITransport;
FProtocol : IProtocol;
FNumIteration : Integer;
FConsole : TThreadConsole;
// test reporting, will be refactored out into separate class later
FTestGroup : string;
FCurrentTest : TTestGroup;
FSuccesses : Integer;
FErrors : TStringList;
FFailed : TTestGroups;
FExecuted : TTestGroups;
procedure StartTestGroup( const aGroup : string; const aTest : TTestGroup);
procedure Expect( aTestResult : Boolean; const aTestInfo : string);
procedure ReportResults;
function CalculateExitCode : Byte;
procedure ClientTest;
{$IFDEF SupportsAsync}
procedure ClientAsyncTest;
{$ENDIF}
procedure InitializeProtocolTransportStack;
procedure ShutdownProtocolTransportStack;
function InitializeHttpTransport( const aTimeoutSetting : Integer; const aConfig : IThriftConfiguration = nil) : IHTTPClient;
procedure JSONProtocolReadWriteTest;
function PrepareBinaryData( aRandomDist : Boolean; aSize : TTestSize) : TBytes;
{$IFDEF StressTest}
procedure StressTest(const client : TThriftTest.Iface);
{$ENDIF}
{$IFDEF Win64}
procedure UseInterlockedExchangeAdd64;
{$ENDIF}
protected
procedure Execute; override;
public
constructor Create( const aSetup : TTestSetup; const aNumIteration: Integer);
destructor Destroy; override;
end;
TTestClient = class
private
class var
FNumIteration : Integer;
FNumThread : Integer;
class procedure PrintCmdLineHelp;
class procedure InvalidArgs;
public
class function Execute( const arguments: array of string) : Byte;
end;
implementation
const
EXITCODE_SUCCESS = $00; // no errors bits set
//
EXITCODE_FAILBIT_BASETYPES = $01;
EXITCODE_FAILBIT_STRUCTS = $02;
EXITCODE_FAILBIT_CONTAINERS = $04;
EXITCODE_FAILBIT_EXCEPTIONS = $08;
MAP_FAILURES_TO_EXITCODE_BITS : array[TClientThread.TTestGroup] of Byte = (
EXITCODE_SUCCESS, // no bits here
EXITCODE_FAILBIT_BASETYPES,
EXITCODE_FAILBIT_STRUCTS,
EXITCODE_FAILBIT_CONTAINERS,
EXITCODE_FAILBIT_EXCEPTIONS
);
function BoolToString( b : Boolean) : string;
// overrides global BoolToString()
begin
if b
then result := 'true'
else result := 'false';
end;
// not available in all versions, so make sure we have this one imported
function IsDebuggerPresent: BOOL; stdcall; external KERNEL32 name 'IsDebuggerPresent';
{ TTestClient }
class procedure TTestClient.PrintCmdLineHelp;
const HELPTEXT = ' [options]'#10
+ #10
+ 'Allowed options:'#10
+ ' -h | --help Produces this help message'#10
+ ' --host=arg (localhost) Host to connect'#10
+ ' --port=arg (9090) Port number to connect'#10
+ ' --pipe=arg Windows Named Pipe (e.g. MyThriftPipe)'#10
+ ' --anon-pipes hRead hWrite Windows Anonymous Pipes pair (handles)'#10
+ ' --transport=arg (sockets) Transport: buffered, framed, http, winhttp'#10
+ ' --protocol=arg (binary) Protocol: binary, compact, json'#10
+ ' --ssl Encrypted Transport using SSL'#10
+ ' -n=num | --testloops=num (1) Number of Tests'#10
+ ' -t=num | --threads=num (1) Number of Test threads'#10
+ ' --performance Run the built-in performance test (no other arguments)'#10
;
begin
Writeln( ChangeFileExt(ExtractFileName(ParamStr(0)),'') + HELPTEXT);
end;
class procedure TTestClient.InvalidArgs;
begin
Console.WriteLine( 'Invalid args.');
Console.WriteLine( ChangeFileExt(ExtractFileName(ParamStr(0)),'') + ' -h for more information');
Abort;
end;
class function TTestClient.Execute(const arguments: array of string) : Byte;
function IsSwitch( const aArgument, aSwitch : string; out sValue : string) : Boolean;
begin
sValue := '';
result := (Copy( aArgument, 1, Length(aSwitch)) = aSwitch);
if result then begin
if (Copy( aArgument, 1, Length(aSwitch)+1) = (aSwitch+'='))
then sValue := Copy( aArgument, Length(aSwitch)+2, MAXINT);
end;
end;
var
iArg : Integer;
threadExitCode : Byte;
sArg, sValue : string;
threads : array of TThread;
dtStart : TDateTime;
test : Integer;
thread : TThread;
setup : TTestSetup;
begin
// init record
with setup do begin
protType := prot_Binary;
endpoint := trns_Sockets;
layered := [];
useSSL := FALSE;
host := 'localhost';
port := 9090;
sPipeName := '';
hAnonRead := INVALID_HANDLE_VALUE;
hAnonWrite := INVALID_HANDLE_VALUE;
end;
try
iArg := 0;
while iArg < Length(arguments) do begin
sArg := arguments[iArg];
Inc(iArg);
if IsSwitch( sArg, '-h', sValue)
or IsSwitch( sArg, '--help', sValue)
then begin
// -h [ --help ] produce help message
PrintCmdLineHelp;
result := $FF; // all tests failed
Exit;
end
else if IsSwitch( sArg, '--host', sValue) then begin
// --host arg (=localhost) Host to connect
setup.host := sValue;
end
else if IsSwitch( sArg, '--port', sValue) then begin
// --port arg (=9090) Port number to connect
setup.port := StrToIntDef(sValue,0);
if setup.port <= 0 then InvalidArgs;
end
else if IsSwitch( sArg, '--domain-socket', sValue) then begin
// --domain-socket arg Domain Socket (e.g. /tmp/ThriftTest.thrift), instead of host and port
raise Exception.Create('domain-socket not supported');
end
// --pipe arg Windows Named Pipe (e.g. MyThriftPipe)
else if IsSwitch( sArg, '--pipe', sValue) then begin
// --pipe arg Windows Named Pipe (e.g. MyThriftPipe)
setup.endpoint := trns_NamedPipes;
setup.sPipeName := sValue;
Console.WriteLine('Using named pipe ('+setup.sPipeName+')');
end
else if IsSwitch( sArg, '--anon-pipes', sValue) then begin
// --anon-pipes hRead hWrite Windows Anonymous Pipes pair (handles)
setup.endpoint := trns_AnonPipes;
setup.hAnonRead := THandle( StrToIntDef( arguments[iArg], Integer(INVALID_HANDLE_VALUE)));
Inc(iArg);
setup.hAnonWrite := THandle( StrToIntDef( arguments[iArg], Integer(INVALID_HANDLE_VALUE)));
Inc(iArg);
Console.WriteLine('Using anonymous pipes ('+IntToStr(Integer(setup.hAnonRead))+' and '+IntToStr(Integer(setup.hAnonWrite))+')');
end
else if IsSwitch( sArg, '--transport', sValue) then begin
// --transport arg (=sockets) Transport: buffered, framed, http, winhttp, evhttp
if sValue = 'buffered' then Include( setup.layered, trns_Buffered)
else if sValue = 'framed' then Include( setup.layered, trns_Framed)
else if sValue = 'http' then setup.endpoint := trns_MsXmlHttp
else if sValue = 'winhttp' then setup.endpoint := trns_WinHttp
else if sValue = 'evhttp' then setup.endpoint := trns_EvHttp // recognized, but not supported
else InvalidArgs;
end
else if IsSwitch( sArg, '--protocol', sValue) then begin
// --protocol arg (=binary) Protocol: binary, compact, json
if sValue = 'binary' then setup.protType := prot_Binary
else if sValue = 'compact' then setup.protType := prot_Compact
else if sValue = 'json' then setup.protType := prot_JSON
else InvalidArgs;
end
else if IsSwitch( sArg, '--ssl', sValue) then begin
// --ssl Encrypted Transport using SSL
setup.useSSL := TRUE;
end
else if IsSwitch( sArg, '-n', sValue) or IsSwitch( sArg, '--testloops', sValue) then begin
// -n [ --testloops ] arg (=1) Number of Tests
FNumIteration := StrToIntDef( sValue, 0);
if FNumIteration <= 0
then InvalidArgs;
end
else if IsSwitch( sArg, '-t', sValue) or IsSwitch( sArg, '--threads', sValue) then begin
// -t [ --threads ] arg (=1) Number of Test threads
FNumThread := StrToIntDef( sValue, 0);
if FNumThread <= 0
then InvalidArgs;
end
else if IsSwitch( sArg, '--performance', sValue) then begin
result := TPerformanceTests.Execute;
Exit;
end
else begin
InvalidArgs;
end;
end;
// In the anonymous pipes mode the client is launched by the test server
// -> behave nicely and allow for attaching a debugger to this process
if (setup.endpoint = trns_AnonPipes) and not IsDebuggerPresent
then MessageBox( 0, 'Attach Debugger and/or click OK to continue.',
'Thrift TestClient (Delphi)',
MB_OK or MB_ICONEXCLAMATION);
SetLength( threads, FNumThread);
dtStart := Now;
// layered transports are not really meant to be stacked upon each other
if (trns_Framed in setup.layered) then begin
Console.WriteLine('Using framed transport');
end
else if (trns_Buffered in setup.layered) then begin
Console.WriteLine('Using buffered transport');
end;
Console.WriteLine(THRIFT_PROTOCOLS[setup.protType]+' protocol');
for test := 0 to FNumThread - 1 do begin
thread := TClientThread.Create( setup, FNumIteration);
threads[test] := thread;
thread.Start;
end;
result := 0;
for test := 0 to FNumThread - 1 do begin
threadExitCode := threads[test].WaitFor;
result := result or threadExitCode;
threads[test].Free;
threads[test] := nil;
end;
Console.Write('Total time: ' + IntToStr( MilliSecondsBetween(Now, dtStart)));
except
on E: EAbort do raise;
on E: Exception do begin
Console.WriteLine( E.Message + #10 + E.StackTrace);
raise;
end;
end;
Console.WriteLine('');
Console.WriteLine('done!');
end;
{ TClientThread }
procedure TClientThread.ClientTest;
var
client : TThriftTest.Iface;
s : string;
i8 : ShortInt;
i32 : Integer;
i64 : Int64;
binOut,binIn : TBytes;
dub : Double;
o : IXtruct;
o2 : IXtruct2;
i : IXtruct;
i2 : IXtruct2;
mapout : IThriftDictionary<Integer,Integer>;
mapin : IThriftDictionary<Integer,Integer>;
strmapout : IThriftDictionary<string,string>;
strmapin : IThriftDictionary<string,string>;
j : Integer;
first : Boolean;
key : Integer;
strkey : string;
listout : IThriftList<Integer>;
listin : IThriftList<Integer>;
setout : IHashSet<Integer>;
setin : IHashSet<Integer>;
ret : TNumberz;
uid : Int64;
mm : IThriftDictionary<Integer, IThriftDictionary<Integer, Integer>>;
pos : IThriftDictionary<Integer, Integer>;
neg : IThriftDictionary<Integer, Integer>;
m2 : IThriftDictionary<Integer, Integer>;
k2 : Integer;
insane : IInsanity;
truck : IXtruct;
whoa : IThriftDictionary<Int64, IThriftDictionary<TNumberz, IInsanity>>;
key64 : Int64;
val : IThriftDictionary<TNumberz, IInsanity>;
k2_2 : TNumberz;
k3 : TNumberz;
v2 : IInsanity;
userMap : IThriftDictionary<TNumberz, Int64>;
xtructs : IThriftList<IXtruct>;
x : IXtruct;
arg0 : ShortInt;
arg1 : Integer;
arg2 : Int64;
arg3 : IThriftDictionary<SmallInt, string>;
arg4 : TNumberz;
arg5 : Int64;
{$IFDEF PerfTest}
StartTick : Cardinal;
k : Integer;
{$ENDIF}
hello, goodbye : IXtruct;
crazy : IInsanity;
looney : IInsanity;
first_map : IThriftDictionary<TNumberz, IInsanity>;
second_map : IThriftDictionary<TNumberz, IInsanity>;
pair : TPair<TNumberz, TUserId>;
testsize : TTestSize;
begin
client := TThriftTest.TClient.Create( FProtocol);
FTransport.Open;
{$IFDEF StressTest}
StressTest( client);
{$ENDIF StressTest}
{$IFDEF Exceptions}
// in-depth exception test
// (1) do we get an exception at all?
// (2) do we get the right exception?
// (3) does the exception contain the expected data?
StartTestGroup( 'testException', test_Exceptions);
// case 1: exception type declared in IDL at the function call
try
client.testException('Xception');
Expect( FALSE, 'testException(''Xception''): must trow an exception');
except
on e:TXception do begin
Expect( e.ErrorCode = 1001, 'error code');
Expect( e.Message_ = 'Xception', 'error message');
Console.WriteLine( ' = ' + IntToStr(e.ErrorCode) + ', ' + e.Message_ );
end;
on e:TTransportException do Expect( FALSE, 'Unexpected : "'+e.ToString+'"');
on e:Exception do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
// re-open connection if needed
if not FTransport.IsOpen
then FTransport.Open;
// case 2: exception type NOT declared in IDL at the function call
// this will close the connection
try
client.testException('TException');
Expect( FALSE, 'testException(''TException''): must trow an exception');
except
on e:TTransportException do begin
Console.WriteLine( e.ClassName+' = '+e.Message); // this is what we get
end;
on e:TApplicationException do begin
Console.WriteLine( e.ClassName+' = '+e.Message); // this is what we get
end;
on e:TException do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
on e:Exception do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
if FTransport.IsOpen then FTransport.Close;
FTransport.Open; // re-open connection, server has already closed
// case 3: no exception
try
client.testException('something');
Expect( TRUE, 'testException(''something''): must not trow an exception');
except
on e:TTransportException do Expect( FALSE, 'Unexpected : "'+e.ToString+'"');
on e:Exception do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
{$ENDIF Exceptions}
// re-open connection if needed
if not FTransport.IsOpen
then FTransport.Open;
// simple things
StartTestGroup( 'simple Thrift calls', test_BaseTypes);
client.testVoid();
Expect( TRUE, 'testVoid()'); // success := no exception
s := BoolToString( client.testBool(TRUE));
Expect( s = BoolToString(TRUE), 'testBool(TRUE) = '+s);
s := BoolToString( client.testBool(FALSE));
Expect( s = BoolToString(FALSE), 'testBool(FALSE) = '+s);
s := client.testString('Test');
Expect( s = 'Test', 'testString(''Test'') = "'+s+'"');
s := client.testString(''); // empty string
Expect( s = '', 'testString('''') = "'+s+'"');
s := client.testString(HUGE_TEST_STRING);
Expect( length(s) = length(HUGE_TEST_STRING),
'testString( length(HUGE_TEST_STRING) = '+IntToStr(Length(HUGE_TEST_STRING))+') '
+'=> length(result) = '+IntToStr(Length(s)));
i8 := client.testByte(1);
Expect( i8 = 1, 'testByte(1) = ' + IntToStr( i8 ));
i32 := client.testI32(-1);
Expect( i32 = -1, 'testI32(-1) = ' + IntToStr(i32));
Console.WriteLine('testI64(-34359738368)');
i64 := client.testI64(-34359738368);
Expect( i64 = -34359738368, 'testI64(-34359738368) = ' + IntToStr( i64));
// random binary small
for testsize := Low(TTestSize) to High(TTestSize) do begin
binOut := PrepareBinaryData( TRUE, testsize);
Console.WriteLine('testBinary('+IntToStr(Length(binOut))+' bytes)');
try
binIn := client.testBinary(binOut);
Expect( Length(binOut) = Length(binIn), 'testBinary('+IntToStr(Length(binOut))+' bytes): '+IntToStr(Length(binIn))+' bytes received');
i32 := Min( Length(binOut), Length(binIn));
Expect( CompareMem( binOut, binIn, i32), 'testBinary('+IntToStr(Length(binOut))+' bytes): validating received data');
except
on e:TApplicationException do Console.WriteLine('testBinary(): '+e.Message);
on e:Exception do Expect( FALSE, 'testBinary(): Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
end;
Console.WriteLine('testDouble(5.325098235)');
dub := client.testDouble(5.325098235);
Expect( abs(dub-5.325098235) < 1e-14, 'testDouble(5.325098235) = ' + FloatToStr( dub));
// structs
StartTestGroup( 'testStruct', test_Structs);
Console.WriteLine('testStruct({''Zero'', 1, -3, -5})');
o := TXtructImpl.Create;
o.String_thing := 'Zero';
o.Byte_thing := 1;
o.I32_thing := -3;
o.I64_thing := -5;
i := client.testStruct(o);
Expect( i.String_thing = 'Zero', 'i.String_thing = "'+i.String_thing+'"');
Expect( i.Byte_thing = 1, 'i.Byte_thing = '+IntToStr(i.Byte_thing));
Expect( i.I32_thing = -3, 'i.I32_thing = '+IntToStr(i.I32_thing));
Expect( i.I64_thing = -5, 'i.I64_thing = '+IntToStr(i.I64_thing));
Expect( i.__isset_String_thing, 'i.__isset_String_thing = '+BoolToString(i.__isset_String_thing));
Expect( i.__isset_Byte_thing, 'i.__isset_Byte_thing = '+BoolToString(i.__isset_Byte_thing));
Expect( i.__isset_I32_thing, 'i.__isset_I32_thing = '+BoolToString(i.__isset_I32_thing));
Expect( i.__isset_I64_thing, 'i.__isset_I64_thing = '+BoolToString(i.__isset_I64_thing));
// nested structs
StartTestGroup( 'testNest', test_Structs);
Console.WriteLine('testNest({1, {''Zero'', 1, -3, -5}, 5})');
o2 := TXtruct2Impl.Create;
o2.Byte_thing := 1;
o2.Struct_thing := o;
o2.I32_thing := 5;
i2 := client.testNest(o2);
i := i2.Struct_thing;
Expect( i.String_thing = 'Zero', 'i.String_thing = "'+i.String_thing+'"');
Expect( i.Byte_thing = 1, 'i.Byte_thing = '+IntToStr(i.Byte_thing));
Expect( i.I32_thing = -3, 'i.I32_thing = '+IntToStr(i.I32_thing));
Expect( i.I64_thing = -5, 'i.I64_thing = '+IntToStr(i.I64_thing));
Expect( i2.Byte_thing = 1, 'i2.Byte_thing = '+IntToStr(i2.Byte_thing));
Expect( i2.I32_thing = 5, 'i2.I32_thing = '+IntToStr(i2.I32_thing));
Expect( i.__isset_String_thing, 'i.__isset_String_thing = '+BoolToString(i.__isset_String_thing));
Expect( i.__isset_Byte_thing, 'i.__isset_Byte_thing = '+BoolToString(i.__isset_Byte_thing));
Expect( i.__isset_I32_thing, 'i.__isset_I32_thing = '+BoolToString(i.__isset_I32_thing));
Expect( i.__isset_I64_thing, 'i.__isset_I64_thing = '+BoolToString(i.__isset_I64_thing));
Expect( i2.__isset_Byte_thing, 'i2.__isset_Byte_thing');
Expect( i2.__isset_I32_thing, 'i2.__isset_I32_thing');
// map<type1,type2>: A map of strictly unique keys to values.
// Translates to an STL map, Java HashMap, PHP associative array, Python/Ruby dictionary, etc.
StartTestGroup( 'testMap', test_Containers);
mapout := TThriftDictionaryImpl<Integer,Integer>.Create;
for j := 0 to 4 do
begin
mapout.AddOrSetValue( j, j - 10);
end;
Console.Write('testMap({');
first := True;
for key in mapout.Keys do
begin
if first
then first := False
else Console.Write( ', ' );
Console.Write( IntToStr( key) + ' => ' + IntToStr( mapout[key]));
end;
Console.WriteLine('})');
mapin := client.testMap( mapout );
Expect( mapin.Count = mapout.Count, 'testMap: mapin.Count = mapout.Count');
for j := 0 to 4 do
begin
Expect( mapout.ContainsKey(j), 'testMap: mapout.ContainsKey('+IntToStr(j)+') = '+BoolToString(mapout.ContainsKey(j)));
end;
for key in mapin.Keys do
begin
Expect( mapin[key] = mapout[key], 'testMap: '+IntToStr(key) + ' => ' + IntToStr( mapin[key]));
Expect( mapin[key] = key - 10, 'testMap: mapin['+IntToStr(key)+'] = '+IntToStr( mapin[key]));
end;
// map<type1,type2>: A map of strictly unique keys to values.
// Translates to an STL map, Java HashMap, PHP associative array, Python/Ruby dictionary, etc.
StartTestGroup( 'testStringMap', test_Containers);
strmapout := TThriftDictionaryImpl<string,string>.Create;
for j := 0 to 4 do
begin
strmapout.AddOrSetValue( IntToStr(j), IntToStr(j - 10));
end;
Console.Write('testStringMap({');
first := True;
for strkey in strmapout.Keys do
begin
if first
then first := False
else Console.Write( ', ' );
Console.Write( strkey + ' => ' + strmapout[strkey]);
end;
Console.WriteLine('})');
strmapin := client.testStringMap( strmapout );
Expect( strmapin.Count = strmapout.Count, 'testStringMap: strmapin.Count = strmapout.Count');
for j := 0 to 4 do
begin
Expect( strmapout.ContainsKey(IntToStr(j)),
'testStringMap: strmapout.ContainsKey('+IntToStr(j)+') = '
+ BoolToString(strmapout.ContainsKey(IntToStr(j))));
end;
for strkey in strmapin.Keys do
begin
Expect( strmapin[strkey] = strmapout[strkey], 'testStringMap: '+strkey + ' => ' + strmapin[strkey]);
Expect( strmapin[strkey] = IntToStr( StrToInt(strkey) - 10), 'testStringMap: strmapin['+strkey+'] = '+strmapin[strkey]);
end;
// set<type>: An unordered set of unique elements.
// Translates to an STL set, Java HashSet, set in Python, etc.
// Note: PHP does not support sets, so it is treated similar to a List
StartTestGroup( 'testSet', test_Containers);
setout := THashSetImpl<Integer>.Create;
for j := -2 to 2 do
begin
setout.Add( j );
end;
Console.Write('testSet({');
first := True;
for j in setout do
begin
if first
then first := False
else Console.Write(', ');
Console.Write(IntToStr( j));
end;
Console.WriteLine('})');
setin := client.testSet(setout);
Expect( setin.Count = setout.Count, 'testSet: setin.Count = setout.Count');
Expect( setin.Count = 5, 'testSet: setin.Count = '+IntToStr(setin.Count));
for j := -2 to 2 do // unordered, we can't rely on the order => test for known elements only
begin
Expect( setin.Contains(j), 'testSet: setin.Contains('+IntToStr(j)+') => '+BoolToString(setin.Contains(j)));
end;
// list<type>: An ordered list of elements.
// Translates to an STL vector, Java ArrayList, native arrays in scripting languages, etc.
StartTestGroup( 'testList', test_Containers);
listout := TThriftListImpl<Integer>.Create;
listout.Add( +1);
listout.Add( -2);
listout.Add( +3);
listout.Add( -4);
listout.Add( 0);
Console.Write('testList({');
first := True;
for j in listout do
begin
if first
then first := False
else Console.Write(', ');
Console.Write(IntToStr( j));
end;
Console.WriteLine('})');
listin := client.testList(listout);
Expect( listin.Count = listout.Count, 'testList: listin.Count = listout.Count');
Expect( listin.Count = 5, 'testList: listin.Count = '+IntToStr(listin.Count));
Expect( listin[0] = +1, 'listin[0] = '+IntToStr( listin[0]));
Expect( listin[1] = -2, 'listin[1] = '+IntToStr( listin[1]));
Expect( listin[2] = +3, 'listin[2] = '+IntToStr( listin[2]));
Expect( listin[3] = -4, 'listin[3] = '+IntToStr( listin[3]));
Expect( listin[4] = 0, 'listin[4] = '+IntToStr( listin[4]));
// enums
ret := client.testEnum(TNumberz.ONE);
Expect( ret = TNumberz.ONE, 'testEnum(ONE) = '+IntToStr(Ord(ret)));
ret := client.testEnum(TNumberz.TWO);
Expect( ret = TNumberz.TWO, 'testEnum(TWO) = '+IntToStr(Ord(ret)));
ret := client.testEnum(TNumberz.THREE);
Expect( ret = TNumberz.THREE, 'testEnum(THREE) = '+IntToStr(Ord(ret)));
ret := client.testEnum(TNumberz.FIVE);
Expect( ret = TNumberz.FIVE, 'testEnum(FIVE) = '+IntToStr(Ord(ret)));
ret := client.testEnum(TNumberz.EIGHT);
Expect( ret = TNumberz.EIGHT, 'testEnum(EIGHT) = '+IntToStr(Ord(ret)));
// typedef
uid := client.testTypedef(309858235082523);
Expect( uid = 309858235082523, 'testTypedef(309858235082523) = '+IntToStr(uid));
// maps of maps
StartTestGroup( 'testMapMap(1)', test_Containers);
mm := client.testMapMap(1);
Console.Write(' = {');
for key in mm.Keys do
begin
Console.Write( IntToStr( key) + ' => {');
m2 := mm[key];
for k2 in m2.Keys do
begin
Console.Write( IntToStr( k2) + ' => ' + IntToStr( m2[k2]) + ', ');
end;
Console.Write('}, ');
end;
Console.WriteLine('}');
// verify result data
Expect( mm.Count = 2, 'mm.Count = '+IntToStr(mm.Count));
pos := mm[4];
neg := mm[-4];
for j := 1 to 4 do
begin
Expect( pos[j] = j, 'pos[j] = '+IntToStr(pos[j]));
Expect( neg[-j] = -j, 'neg[-j] = '+IntToStr(neg[-j]));
end;
// insanity
StartTestGroup( 'testInsanity', test_Structs);
insane := TInsanityImpl.Create;
insane.UserMap := TThriftDictionaryImpl<TNumberz, Int64>.Create;
insane.UserMap.AddOrSetValue( TNumberz.FIVE, 5000);
truck := TXtructImpl.Create;
truck.String_thing := 'Truck';
truck.Byte_thing := -8; // byte is signed
truck.I32_thing := 32;
truck.I64_thing := 64;
insane.Xtructs := TThriftListImpl<IXtruct>.Create;
insane.Xtructs.Add( truck );
whoa := client.testInsanity( insane );
Console.Write(' = {');
for key64 in whoa.Keys do
begin
val := whoa[key64];
Console.Write( IntToStr( key64) + ' => {');
for k2_2 in val.Keys do
begin
v2 := val[k2_2];
Console.Write( IntToStr( Integer( k2_2)) + ' => {');
userMap := v2.UserMap;
Console.Write('{');
if userMap <> nil then
begin
for k3 in userMap.Keys do
begin
Console.Write( IntToStr( Integer( k3)) + ' => ' + IntToStr( userMap[k3]) + ', ');
end;
end else
begin
Console.Write('null');
end;
Console.Write('}, ');
xtructs := v2.Xtructs;
Console.Write('{');
if xtructs <> nil then
begin
for x in xtructs do
begin
Console.Write('{"' + x.String_thing + '", ' +
IntToStr( x.Byte_thing) + ', ' +
IntToStr( x.I32_thing) + ', ' +
IntToStr( x.I32_thing) + '}, ');
end;
end else
begin
Console.Write('null');
end;
Console.Write('}');
Console.Write('}, ');
end;
Console.Write('}, ');
end;
Console.WriteLine('}');
(**
* So you think you've got this all worked, out eh?
*
* Creates a the returned map with these values and prints it out:
* { 1 => { 2 => argument,
* 3 => argument,
* },
* 2 => { 6 => <empty Insanity struct>, },
* }
* @return map<UserId, map<Numberz,Insanity>> - a map with the above values
*)
// verify result data
Expect( whoa.Count = 2, 'whoa.Count = '+IntToStr(whoa.Count));
//
first_map := whoa[1];
second_map := whoa[2];
Expect( first_map.Count = 2, 'first_map.Count = '+IntToStr(first_map.Count));
Expect( second_map.Count = 1, 'second_map.Count = '+IntToStr(second_map.Count));
//
looney := second_map[TNumberz.SIX];
Expect( Assigned(looney), 'Assigned(looney) = '+BoolToString(Assigned(looney)));
Expect( not looney.__isset_UserMap, 'looney.__isset_UserMap = '+BoolToString(looney.__isset_UserMap));
Expect( not looney.__isset_Xtructs, 'looney.__isset_Xtructs = '+BoolToString(looney.__isset_Xtructs));
//
for ret in [TNumberz.TWO, TNumberz.THREE] do begin
crazy := first_map[ret];
Console.WriteLine('first_map['+intToStr(Ord(ret))+']');
Expect( crazy.__isset_UserMap, 'crazy.__isset_UserMap = '+BoolToString(crazy.__isset_UserMap));
Expect( crazy.__isset_Xtructs, 'crazy.__isset_Xtructs = '+BoolToString(crazy.__isset_Xtructs));
Expect( crazy.UserMap.Count = insane.UserMap.Count, 'crazy.UserMap.Count = '+IntToStr(crazy.UserMap.Count));
for pair in insane.UserMap do begin
Expect( crazy.UserMap[pair.Key] = pair.Value, 'crazy.UserMap['+IntToStr(Ord(pair.key))+'] = '+IntToStr(crazy.UserMap[pair.Key]));
end;
Expect( crazy.Xtructs.Count = insane.Xtructs.Count, 'crazy.Xtructs.Count = '+IntToStr(crazy.Xtructs.Count));
for arg0 := 0 to insane.Xtructs.Count-1 do begin
hello := insane.Xtructs[arg0];
goodbye := crazy.Xtructs[arg0];
Expect( goodbye.String_thing = hello.String_thing, 'goodbye.String_thing = '+goodbye.String_thing);
Expect( goodbye.Byte_thing = hello.Byte_thing, 'goodbye.Byte_thing = '+IntToStr(goodbye.Byte_thing));
Expect( goodbye.I32_thing = hello.I32_thing, 'goodbye.I32_thing = '+IntToStr(goodbye.I32_thing));
Expect( goodbye.I64_thing = hello.I64_thing, 'goodbye.I64_thing = '+IntToStr(goodbye.I64_thing));
end;
end;
// multi args
StartTestGroup( 'testMulti', test_BaseTypes);
arg0 := 1;
arg1 := 2;
arg2 := High(Int64);
arg3 := TThriftDictionaryImpl<SmallInt, string>.Create;
arg3.AddOrSetValue( 1, 'one');
arg4 := TNumberz.FIVE;
arg5 := 5000000;
Console.WriteLine('Test Multi(' + IntToStr( arg0) + ',' +
IntToStr( arg1) + ',' + IntToStr( arg2) + ',' +
arg3.ToString + ',' + IntToStr( Integer( arg4)) + ',' +
IntToStr( arg5) + ')');
i := client.testMulti( arg0, arg1, arg2, arg3, arg4, arg5);
Expect( i.String_thing = 'Hello2', 'testMulti: i.String_thing = "'+i.String_thing+'"');
Expect( i.Byte_thing = arg0, 'testMulti: i.Byte_thing = '+IntToStr(i.Byte_thing));
Expect( i.I32_thing = arg1, 'testMulti: i.I32_thing = '+IntToStr(i.I32_thing));
Expect( i.I64_thing = arg2, 'testMulti: i.I64_thing = '+IntToStr(i.I64_thing));
Expect( i.__isset_String_thing, 'testMulti: i.__isset_String_thing = '+BoolToString(i.__isset_String_thing));
Expect( i.__isset_Byte_thing, 'testMulti: i.__isset_Byte_thing = '+BoolToString(i.__isset_Byte_thing));
Expect( i.__isset_I32_thing, 'testMulti: i.__isset_I32_thing = '+BoolToString(i.__isset_I32_thing));
Expect( i.__isset_I64_thing, 'testMulti: i.__isset_I64_thing = '+BoolToString(i.__isset_I64_thing));
// multi exception
StartTestGroup( 'testMultiException(1)', test_Exceptions);
try
i := client.testMultiException( 'need more pizza', 'run out of beer');
Expect( i.String_thing = 'run out of beer', 'i.String_thing = "' +i.String_thing+ '"');
Expect( i.__isset_String_thing, 'i.__isset_String_thing = '+BoolToString(i.__isset_String_thing));
{ this is not necessarily true, these fields are default-serialized
Expect( not i.__isset_Byte_thing, 'i.__isset_Byte_thing = '+BoolToString(i.__isset_Byte_thing));
Expect( not i.__isset_I32_thing, 'i.__isset_I32_thing = '+BoolToString(i.__isset_I32_thing));
Expect( not i.__isset_I64_thing, 'i.__isset_I64_thing = '+BoolToString(i.__isset_I64_thing));
}
except
on e:Exception do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
StartTestGroup( 'testMultiException(Xception)', test_Exceptions);
try
i := client.testMultiException( 'Xception', 'second test');
Expect( FALSE, 'testMultiException(''Xception''): must trow an exception');
except
on x:TXception do begin
Expect( x.__isset_ErrorCode, 'x.__isset_ErrorCode = '+BoolToString(x.__isset_ErrorCode));
Expect( x.__isset_Message_, 'x.__isset_Message_ = '+BoolToString(x.__isset_Message_));
Expect( x.ErrorCode = 1001, 'x.ErrorCode = '+IntToStr(x.ErrorCode));
Expect( x.Message_ = 'This is an Xception', 'x.Message = "'+x.Message_+'"');
end;
on e:Exception do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
StartTestGroup( 'testMultiException(Xception2)', test_Exceptions);
try
i := client.testMultiException( 'Xception2', 'third test');
Expect( FALSE, 'testMultiException(''Xception2''): must trow an exception');
except
on x:TXception2 do begin
Expect( x.__isset_ErrorCode, 'x.__isset_ErrorCode = '+BoolToString(x.__isset_ErrorCode));
Expect( x.__isset_Struct_thing, 'x.__isset_Struct_thing = '+BoolToString(x.__isset_Struct_thing));
Expect( x.ErrorCode = 2002, 'x.ErrorCode = '+IntToStr(x.ErrorCode));
Expect( x.Struct_thing.String_thing = 'This is an Xception2', 'x.Struct_thing.String_thing = "'+x.Struct_thing.String_thing+'"');
Expect( x.Struct_thing.__isset_String_thing, 'x.Struct_thing.__isset_String_thing = '+BoolToString(x.Struct_thing.__isset_String_thing));
{ this is not necessarily true, these fields are default-serialized
Expect( not x.Struct_thing.__isset_Byte_thing, 'x.Struct_thing.__isset_Byte_thing = '+BoolToString(x.Struct_thing.__isset_Byte_thing));
Expect( not x.Struct_thing.__isset_I32_thing, 'x.Struct_thing.__isset_I32_thing = '+BoolToString(x.Struct_thing.__isset_I32_thing));
Expect( not x.Struct_thing.__isset_I64_thing, 'x.Struct_thing.__isset_I64_thing = '+BoolToString(x.Struct_thing.__isset_I64_thing));
}
end;
on e:Exception do Expect( FALSE, 'Unexpected exception "'+e.ClassName+'": '+e.Message);
end;
// oneway functions
StartTestGroup( 'Test Oneway(1)', test_Unknown);
client.testOneway(1);
Expect( TRUE, 'Test Oneway(1)'); // success := no exception
// call time
{$IFDEF PerfTest}
StartTestGroup( 'Test Calltime()');
StartTick := GetTickCount;
for k := 0 to 1000 - 1 do
begin
client.testVoid();
end;
Console.WriteLine(' = ' + FloatToStr( (GetTickCount - StartTick) / 1000 ) + ' ms a testVoid() call' );
{$ENDIF PerfTest}
// no more tests here
StartTestGroup( '', test_Unknown);
end;
{$IFDEF SupportsAsync}
procedure TClientThread.ClientAsyncTest;
var
client : TThriftTest.IAsync;
s : string;
i8 : ShortInt;
begin
StartTestGroup( 'Async Tests', test_Unknown);
client := TThriftTest.TClient.Create( FProtocol);
FTransport.Open;
// oneway void functions
client.testOnewayAsync(1).Wait;
Expect( TRUE, 'Test Oneway(1)'); // success := no exception
// normal functions
s := client.testStringAsync(HUGE_TEST_STRING).Value;
Expect( length(s) = length(HUGE_TEST_STRING),
'testString( length(HUGE_TEST_STRING) = '+IntToStr(Length(HUGE_TEST_STRING))+') '
+'=> length(result) = '+IntToStr(Length(s)));
i8 := client.testByte(1).Value;
Expect( i8 = 1, 'testByte(1) = ' + IntToStr( i8 ));
end;
{$ENDIF}
{$IFDEF StressTest}
procedure TClientThread.StressTest(const client : TThriftTest.Iface);
begin
while TRUE do begin
try
if not FTransport.IsOpen then FTransport.Open; // re-open connection, server has already closed
try
client.testString('Test');
Write('.');
finally
if FTransport.IsOpen then FTransport.Close;
end;
except
on e:Exception do Writeln(#10+e.message);
end;
end;
end;
{$ENDIF}
function TClientThread.PrepareBinaryData( aRandomDist : Boolean; aSize : TTestSize) : TBytes;
var i : Integer;
begin
case aSize of
Empty : SetLength( result, 0);
Normal : SetLength( result, $100);
ByteArrayTest : SetLength( result, SizeOf(TByteArray) + 128);
PipeWriteLimit : SetLength( result, 65535 + 128);
FifteenMB : SetLength( result, 15 * 1024 * 1024);
else
raise EArgumentException.Create('aSize');
end;
ASSERT( Low(result) = 0);
if Length(result) = 0 then Exit;
// linear distribution, unless random is requested
if not aRandomDist then begin
for i := Low(result) to High(result) do begin
result[i] := i mod $100;
end;
Exit;
end;
// random distribution of all 256 values
FillChar( result[0], Length(result) * SizeOf(result[0]), $0);
for i := Low(result) to High(result) do begin
result[i] := Byte( Random($100));
end;
end;
{$IFDEF Win64}
procedure TClientThread.UseInterlockedExchangeAdd64;
var a,b : Int64;
begin
a := 1;
b := 2;
Thrift.Utils.InterlockedExchangeAdd64( a,b);
Expect( a = 3, 'InterlockedExchangeAdd64');
end;
{$ENDIF}
procedure TClientThread.JSONProtocolReadWriteTest;
// Tests only then read/write procedures of the JSON protocol
// All tests succeed, if we can read what we wrote before
// Note that passing this test does not imply, that our JSON is really compatible to what
// other clients or servers expect as the real JSON. This is beyond the scope of this test.
var prot : IProtocol;
stm : TStringStream;
list : TThriftList;
config : IThriftConfiguration;
binary, binRead, emptyBinary : TBytes;
i,iErr : Integer;
const
TEST_SHORT = ShortInt( $FE);
TEST_SMALL = SmallInt( $FEDC);
TEST_LONG = LongInt( $FEDCBA98);
TEST_I64 = Int64( $FEDCBA9876543210);
TEST_DOUBLE = -1.234e-56;
DELTA_DOUBLE = TEST_DOUBLE * 1e-14;
TEST_STRING = 'abc-'#$00E4#$00f6#$00fc; // german umlauts (en-us: "funny chars")
// Test THRIFT-2336 and THRIFT-3404 with U+1D11E (G Clef symbol) and 'Русское Название';
G_CLEF_AND_CYRILLIC_TEXT = #$1d11e' '#$0420#$0443#$0441#$0441#$043a#$043e#$0435' '#$041d#$0430#$0437#$0432#$0430#$043d#$0438#$0435;
G_CLEF_AND_CYRILLIC_JSON = '"\ud834\udd1e \u0420\u0443\u0441\u0441\u043a\u043e\u0435 \u041d\u0430\u0437\u0432\u0430\u043d\u0438\u0435"';
// test both possible solidus encodings
SOLIDUS_JSON_DATA = '"one/two\/three"';
SOLIDUS_EXCPECTED = 'one/two/three';
begin
stm := TStringStream.Create;
try
StartTestGroup( 'JsonProtocolTest', test_Unknown);
config := TThriftConfigurationImpl.Create;
// prepare binary data
binary := PrepareBinaryData( FALSE, Normal);
SetLength( emptyBinary, 0); // empty binary data block
// output setup
prot := TJSONProtocolImpl.Create(
TStreamTransportImpl.Create(
nil, TThriftStreamAdapterDelphi.Create( stm, FALSE), config));
// write
Init( list, TType.String_, 9);
prot.WriteListBegin( list);
prot.WriteBool( TRUE);
prot.WriteBool( FALSE);
prot.WriteByte( TEST_SHORT);
prot.WriteI16( TEST_SMALL);
prot.WriteI32( TEST_LONG);
prot.WriteI64( TEST_I64);
prot.WriteDouble( TEST_DOUBLE);
prot.WriteString( TEST_STRING);
prot.WriteBinary( binary);
prot.WriteString( ''); // empty string
prot.WriteBinary( emptyBinary); // empty binary data block
prot.WriteListEnd;
// input setup
Expect( stm.Position = stm.Size, 'Stream position/length after write');
stm.Position := 0;
prot := TJSONProtocolImpl.Create(
TStreamTransportImpl.Create(
TThriftStreamAdapterDelphi.Create( stm, FALSE), nil, config));
// read and compare
list := prot.ReadListBegin;
Expect( list.ElementType = TType.String_, 'list element type');
Expect( list.Count = 9, 'list element count');
Expect( prot.ReadBool, 'WriteBool/ReadBool: TRUE');
Expect( not prot.ReadBool, 'WriteBool/ReadBool: FALSE');
Expect( prot.ReadByte = TEST_SHORT, 'WriteByte/ReadByte');
Expect( prot.ReadI16 = TEST_SMALL, 'WriteI16/ReadI16');
Expect( prot.ReadI32 = TEST_LONG, 'WriteI32/ReadI32');
Expect( prot.ReadI64 = TEST_I64, 'WriteI64/ReadI64');
Expect( abs(prot.ReadDouble-TEST_DOUBLE) < abs(DELTA_DOUBLE), 'WriteDouble/ReadDouble');
Expect( prot.ReadString = TEST_STRING, 'WriteString/ReadString');
binRead := prot.ReadBinary;
Expect( Length(prot.ReadString) = 0, 'WriteString/ReadString (empty string)');
Expect( Length(prot.ReadBinary) = 0, 'empty WriteBinary/ReadBinary (empty data block)');
prot.ReadListEnd;
// test binary data
Expect( Length(binary) = Length(binRead), 'Binary data length check');
iErr := -1;
for i := Low(binary) to High(binary) do begin
if binary[i] <> binRead[i] then begin
iErr := i;
Break;
end;
end;
if iErr < 0
then Expect( TRUE, 'Binary data check ('+IntToStr(Length(binary))+' Bytes)')
else Expect( FALSE, 'Binary data check at offset '+IntToStr(iErr));
Expect( stm.Position = stm.Size, 'Stream position after read');
// Solidus can be encoded in two ways. Make sure we can read both
stm.Position := 0;
stm.Size := 0;
stm.WriteString(SOLIDUS_JSON_DATA);
stm.Position := 0;
prot := TJSONProtocolImpl.Create(
TStreamTransportImpl.Create(
TThriftStreamAdapterDelphi.Create( stm, FALSE), nil, config));
Expect( prot.ReadString = SOLIDUS_EXCPECTED, 'Solidus encoding');
// Widechars should work too. Do they?
// After writing, we ensure that we are able to read it back
// We can't assume hex-encoding, since (nearly) any Unicode char is valid JSON
stm.Position := 0;
stm.Size := 0;
prot := TJSONProtocolImpl.Create(
TStreamTransportImpl.Create(
nil, TThriftStreamAdapterDelphi.Create( stm, FALSE), config));
prot.WriteString( G_CLEF_AND_CYRILLIC_TEXT);
stm.Position := 0;
prot := TJSONProtocolImpl.Create(
TStreamTransportImpl.Create(
TThriftStreamAdapterDelphi.Create( stm, FALSE), nil, config));
Expect( prot.ReadString = G_CLEF_AND_CYRILLIC_TEXT, 'Writing JSON with chars > 8 bit');
// Widechars should work with hex-encoding too. Do they?
stm.Position := 0;
stm.Size := 0;
stm.WriteString( G_CLEF_AND_CYRILLIC_JSON);
stm.Position := 0;
prot := TJSONProtocolImpl.Create(
TStreamTransportImpl.Create(
TThriftStreamAdapterDelphi.Create( stm, FALSE), nil, config));
Expect( prot.ReadString = G_CLEF_AND_CYRILLIC_TEXT, 'Reading JSON with chars > 8 bit');
finally
stm.Free;
prot := nil; //-> Release
StartTestGroup( '', test_Unknown); // no more tests here
end;
end;
procedure TClientThread.StartTestGroup( const aGroup : string; const aTest : TTestGroup);
begin
FTestGroup := aGroup;
FCurrentTest := aTest;
Include( FExecuted, aTest);
if FTestGroup <> '' then begin
Console.WriteLine('');
Console.WriteLine( aGroup+' tests');
Console.WriteLine( StringOfChar('-',60));
end;
end;
procedure TClientThread.Expect( aTestResult : Boolean; const aTestInfo : string);
begin
if aTestResult then begin
Inc(FSuccesses);
Console.WriteLine( aTestInfo+': passed');
end
else begin
FErrors.Add( FTestGroup+': '+aTestInfo);
Include( FFailed, FCurrentTest);
Console.WriteLine( aTestInfo+': *** FAILED ***');
// We have a failed test!
// -> issue DebugBreak ONLY if a debugger is attached,
// -> unhandled DebugBreaks would cause Windows to terminate the app otherwise
if IsDebuggerPresent
then {$IFDEF CPUX64} DebugBreak {$ELSE} asm int 3 end {$ENDIF};
end;
end;
procedure TClientThread.ReportResults;
var nTotal : Integer;
sLine : string;
begin
// prevent us from stupid DIV/0 errors
nTotal := FSuccesses + FErrors.Count;
if nTotal = 0 then begin
Console.WriteLine('No results logged');
Exit;
end;
Console.WriteLine('');
Console.WriteLine( StringOfChar('=',60));
Console.WriteLine( IntToStr(nTotal)+' tests performed');
Console.WriteLine( IntToStr(FSuccesses)+' tests succeeded ('+IntToStr(round(100*FSuccesses/nTotal))+'%)');
Console.WriteLine( IntToStr(FErrors.Count)+' tests failed ('+IntToStr(round(100*FErrors.Count/nTotal))+'%)');
Console.WriteLine( StringOfChar('=',60));
if FErrors.Count > 0 then begin
Console.WriteLine('FAILED TESTS:');
for sLine in FErrors do Console.WriteLine('- '+sLine);
Console.WriteLine( StringOfChar('=',60));
InterlockedIncrement( ExitCode); // return <> 0 on errors
end;
Console.WriteLine('');
end;
function TClientThread.CalculateExitCode : Byte;
var test : TTestGroup;
begin
result := EXITCODE_SUCCESS;
for test := Low(TTestGroup) to High(TTestGroup) do begin
if (test in FFailed) or not (test in FExecuted)
then result := result or MAP_FAILURES_TO_EXITCODE_BITS[test];
end;
end;
constructor TClientThread.Create( const aSetup : TTestSetup; const aNumIteration: Integer);
begin
FSetup := aSetup;
FNumIteration := ANumIteration;
FConsole := TThreadConsole.Create( Self );
FCurrentTest := test_Unknown;
// error list: keep correct order, allow for duplicates
FErrors := TStringList.Create;
FErrors.Sorted := FALSE;
FErrors.Duplicates := dupAccept;
inherited Create( TRUE);
end;
destructor TClientThread.Destroy;
begin
FreeAndNil( FConsole);
FreeAndNil( FErrors);
inherited;
end;
procedure TClientThread.Execute;
var
i : Integer;
begin
// perform all tests
try
{$IFDEF Win64}
UseInterlockedExchangeAdd64;
{$ENDIF}
JSONProtocolReadWriteTest;
// must be run in the context of the thread
InitializeProtocolTransportStack;
try
for i := 0 to FNumIteration - 1 do begin
ClientTest;
{$IFDEF SupportsAsync}
ClientAsyncTest;
{$ENDIF}
end;
// report the outcome
ReportResults;
SetReturnValue( CalculateExitCode);
finally
ShutdownProtocolTransportStack;
end;
except
on e:Exception do Expect( FALSE, 'unexpected exception: "'+e.message+'"');
end;
end;
function TClientThread.InitializeHttpTransport( const aTimeoutSetting : Integer; const aConfig : IThriftConfiguration) : IHTTPClient;
var sUrl : string;
comps : URL_COMPONENTS;
dwChars : DWORD;
begin
ASSERT( FSetup.endpoint in [trns_MsxmlHttp, trns_WinHttp]);
if FSetup.useSSL
then sUrl := 'https://'
else sUrl := 'http://';
sUrl := sUrl + FSetup.host;
// add the port number if necessary and at the right place
FillChar( comps, SizeOf(comps), 0);
comps.dwStructSize := SizeOf(comps);
comps.dwSchemeLength := MAXINT;
comps.dwHostNameLength := MAXINT;
comps.dwUserNameLength := MAXINT;
comps.dwPasswordLength := MAXINT;
comps.dwUrlPathLength := MAXINT;
comps.dwExtraInfoLength := MAXINT;
Win32Check( WinHttpCrackUrl( PChar(sUrl), Length(sUrl), 0, comps));
case FSetup.port of
80 : if FSetup.useSSL then comps.nPort := FSetup.port;
443 : if not FSetup.useSSL then comps.nPort := FSetup.port;
else
if FSetup.port > 0 then comps.nPort := FSetup.port;
end;
dwChars := Length(sUrl) + 64;
SetLength( sUrl, dwChars);
Win32Check( WinHttpCreateUrl( comps, 0, @sUrl[1], dwChars));
SetLength( sUrl, dwChars);
Console.WriteLine('Target URL: '+sUrl);
case FSetup.endpoint of
trns_MsxmlHttp : result := TMsxmlHTTPClientImpl.Create( sUrl, aConfig);
trns_WinHttp : result := TWinHTTPClientImpl.Create( sUrl, aConfig);
else
raise Exception.Create(ENDPOINT_TRANSPORTS[FSetup.endpoint]+' unhandled case');
end;
result.DnsResolveTimeout := aTimeoutSetting;
result.ConnectionTimeout := aTimeoutSetting;
result.SendTimeout := aTimeoutSetting;
result.ReadTimeout := aTimeoutSetting;
end;
procedure TClientThread.InitializeProtocolTransportStack;
var streamtrans : IStreamTransport;
canSSL : Boolean;
const
DEBUG_TIMEOUT = 30 * 1000;
RELEASE_TIMEOUT = DEFAULT_THRIFT_TIMEOUT;
PIPE_TIMEOUT = RELEASE_TIMEOUT;
HTTP_TIMEOUTS = 10 * 1000;
begin
// needed for HTTP clients as they utilize the MSXML COM components
OleCheck( CoInitialize( nil));
canSSL := FALSE;
case FSetup.endpoint of
trns_Sockets: begin
Console.WriteLine('Using sockets ('+FSetup.host+' port '+IntToStr(FSetup.port)+')');
streamtrans := TSocketImpl.Create( FSetup.host, FSetup.port);
FTransport := streamtrans;
end;
trns_MsxmlHttp,
trns_WinHttp: begin
Console.WriteLine('Using HTTPClient');
FTransport := InitializeHttpTransport( HTTP_TIMEOUTS);
canSSL := TRUE;
end;
trns_EvHttp: begin
raise Exception.Create(ENDPOINT_TRANSPORTS[FSetup.endpoint]+' transport not implemented');
end;
trns_NamedPipes: begin
streamtrans := TNamedPipeTransportClientEndImpl.Create( FSetup.sPipeName, 0, nil, PIPE_TIMEOUT, PIPE_TIMEOUT);
FTransport := streamtrans;
end;
trns_AnonPipes: begin
streamtrans := TAnonymousPipeTransportImpl.Create( FSetup.hAnonRead, FSetup.hAnonWrite, FALSE, PIPE_TIMEOUT);
FTransport := streamtrans;
end;
else
raise Exception.Create('Unhandled endpoint transport');
end;
ASSERT( FTransport <> nil);
// layered transports are not really meant to be stacked upon each other
if (trns_Framed in FSetup.layered) then begin
FTransport := TFramedTransportImpl.Create( FTransport);
end
else if (trns_Buffered in FSetup.layered) and (streamtrans <> nil) then begin
FTransport := TBufferedTransportImpl.Create( streamtrans, 32); // small buffer to test read()
end;
if FSetup.useSSL and not canSSL then begin
raise Exception.Create('SSL/TLS not implemented');
end;
// create protocol instance, default to BinaryProtocol
FProtocol := PROTOCOL_CLASSES[FSetup.protType].Create(FTransport);
ASSERT( (FTransport <> nil) and (FProtocol <> nil));
end;
procedure TClientThread.ShutdownProtocolTransportStack;
begin
try
FProtocol := nil;
if FTransport <> nil then begin
FTransport.Close;
FTransport := nil;
end;
finally
CoUninitialize;
end;
end;
{ TThreadConsole }
constructor TThreadConsole.Create(AThread: TThread);
begin
inherited Create;
FThread := AThread;
end;
procedure TThreadConsole.Write(const S: string);
var
proc : TThreadProcedure;
begin
proc := procedure
begin
Console.Write( S );
end;
TThread.Synchronize( FThread, proc);
end;
procedure TThreadConsole.WriteLine(const S: string);
var
proc : TThreadProcedure;
begin
proc := procedure
begin
Console.WriteLine( S );
end;
TThread.Synchronize( FThread, proc);
end;
initialization
begin
TTestClient.FNumIteration := 1;
TTestClient.FNumThread := 1;
end;
end.