(* | |
* 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; | |
interface | |
uses | |
Windows, SysUtils, Classes, | |
DateUtils, | |
Generics.Collections, | |
TestConstants, | |
Thrift, | |
Thrift.Protocol.JSON, | |
Thrift.Protocol, | |
Thrift.Transport, | |
Thrift.Stream, | |
Thrift.Test, | |
Thrift.Collections, | |
Thrift.Console; | |
type | |
TThreadConsole = class | |
private | |
FThread : TThread; | |
public | |
procedure Write( const S : string); | |
procedure WriteLine( const S : string); | |
constructor Create( AThread: TThread); | |
end; | |
TClientThread = class( TThread ) | |
private | |
FTransport : ITransport; | |
FProtocol : IProtocol; | |
FNumIteration : Integer; | |
FConsole : TThreadConsole; | |
// test reporting, will be refactored out into separate class later | |
FTestGroup : string; | |
FSuccesses : Integer; | |
FErrors : TStringList; | |
procedure StartTestGroup( const aGroup : string); | |
procedure Expect( aTestResult : Boolean; const aTestInfo : string); | |
procedure ReportResults; | |
procedure ClientTest; | |
procedure JSONProtocolReadWriteTest; | |
protected | |
procedure Execute; override; | |
public | |
constructor Create(ATransport: ITransport; AProtocol : IProtocol; ANumIteration: Integer); | |
destructor Destroy; override; | |
end; | |
TTestClient = class | |
private | |
class var | |
FNumIteration : Integer; | |
FNumThread : Integer; | |
public | |
class procedure Execute( const args: array of string); | |
end; | |
implementation | |
function BoolToString( b : Boolean) : string; | |
// overrides global BoolToString() | |
begin | |
if b | |
then result := 'true' | |
else result := 'false'; | |
end; | |
{ TTestClient } | |
class procedure TTestClient.Execute(const args: array of string); | |
var | |
i : Integer; | |
host : string; | |
port : Integer; | |
url : string; | |
bBuffered : Boolean; | |
bFramed : Boolean; | |
s : string; | |
n : Integer; | |
threads : array of TThread; | |
dtStart : TDateTime; | |
test : Integer; | |
thread : TThread; | |
trans : ITransport; | |
prot : IProtocol; | |
streamtrans : IStreamTransport; | |
http : IHTTPClient; | |
protType, p : TKnownProtocol; | |
begin | |
bBuffered := False;; | |
bFramed := False; | |
protType := prot_Binary; | |
try | |
host := 'localhost'; | |
port := 9090; | |
url := ''; | |
i := 0; | |
try | |
while ( i < Length(args) ) do | |
begin | |
try | |
if ( args[i] = '-h') then | |
begin | |
Inc( i ); | |
s := args[i]; | |
n := Pos( ':', s); | |
if ( n > 0 ) then | |
begin | |
host := Copy( s, 1, n - 1); | |
port := StrToInt( Copy( s, n + 1, MaxInt)); | |
end else | |
begin | |
host := s; | |
end; | |
end else | |
if (args[i] = '-u') then | |
begin | |
Inc( i ); | |
url := args[i]; | |
end else | |
if (args[i] = '-n') then | |
begin | |
Inc( i ); | |
FNumIteration := StrToInt( args[i] ); | |
end else | |
if (args[i] = '-b') then | |
begin | |
bBuffered := True; | |
Console.WriteLine('Using buffered transport'); | |
end else | |
if (args[i] = '-f' ) or ( args[i] = '-framed') then | |
begin | |
bFramed := True; | |
Console.WriteLine('Using framed transport'); | |
end else | |
if (args[i] = '-t') then | |
begin | |
Inc( i ); | |
FNumThread := StrToInt( args[i] ); | |
end else | |
if (args[i] = '-prot') then // -prot JSON|binary | |
begin | |
Inc( i ); | |
s := args[i]; | |
for p:= Low(TKnownProtocol) to High(TKnownProtocol) do begin | |
if SameText( s, KNOWN_PROTOCOLS[p]) then begin | |
protType := p; | |
Console.WriteLine('Using '+KNOWN_PROTOCOLS[protType]+' protocol'); | |
Break; | |
end; | |
end; | |
end; | |
finally | |
Inc( i ); | |
end; | |
end; | |
except | |
on E: Exception do | |
begin | |
Console.WriteLine( E.Message ); | |
end; | |
end; | |
SetLength( threads, FNumThread); | |
dtStart := Now; | |
for test := 0 to FNumThread - 1 do | |
begin | |
if url = '' then | |
begin | |
streamtrans := TSocketImpl.Create( host, port ); | |
trans := streamtrans; | |
if bBuffered then | |
begin | |
trans := TBufferedTransportImpl.Create( streamtrans ); | |
end; | |
if bFramed then | |
begin | |
trans := TFramedTransportImpl.Create( trans ); | |
end; | |
end else | |
begin | |
http := THTTPClientImpl.Create( url ); | |
trans := http; | |
end; | |
// create protocol instance, default to BinaryProtocol | |
case protType of | |
prot_Binary: prot := TBinaryProtocolImpl.Create( trans); | |
prot_JSON : prot := TJSONProtocolImpl.Create( trans); | |
else | |
ASSERT( FALSE); // unhandled case! | |
prot := TBinaryProtocolImpl.Create( trans); // use default | |
end; | |
thread := TClientThread.Create( trans, prot, FNumIteration); | |
threads[test] := thread; | |
{$WARN SYMBOL_DEPRECATED OFF} | |
thread.Resume; | |
{$WARN SYMBOL_DEPRECATED ON} | |
end; | |
for test := 0 to FNumThread - 1 do | |
begin | |
threads[test].WaitFor; | |
end; | |
for test := 0 to FNumThread - 1 do | |
begin | |
threads[test].Free; | |
end; | |
Console.Write('Total time: ' + IntToStr( MilliSecondsBetween(Now, dtStart))); | |
except | |
on E: Exception do | |
begin | |
Console.WriteLine( E.Message + ' ST: ' + E.StackTrace ); | |
end; | |
end; | |
Console.WriteLine(''); | |
Console.WriteLine('done!'); | |
end; | |
{ TClientThread } | |
procedure TClientThread.ClientTest; | |
var | |
client : TThriftTest.Iface; | |
s : string; | |
i8 : ShortInt; | |
i32 : Integer; | |
i64 : Int64; | |
dub : Double; | |
o : IXtruct; | |
o2 : IXtruct2; | |
i : IXtruct; | |
i2 : IXtruct2; | |
mapout : IThriftDictionary<Integer,Integer>; | |
mapin : IThriftDictionary<Integer,Integer>; | |
j : Integer; | |
first : Boolean; | |
key : Integer; | |
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; | |
StartTick : Cardinal; | |
k : Integer; | |
proc : TThreadProcedure; | |
hello, goodbye : IXtruct; | |
crazy : IInsanity; | |
looney : IInsanity; | |
first_map : IThriftDictionary<TNumberz, IInsanity>; | |
second_map : IThriftDictionary<TNumberz, IInsanity>; | |
begin | |
client := TThriftTest.TClient.Create( FProtocol); | |
try | |
if not FTransport.IsOpen then | |
begin | |
FTransport.Open; | |
end; | |
except | |
on E: Exception do | |
begin | |
Console.WriteLine( E.Message ); | |
Exit; | |
end; | |
end; | |
// 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'); | |
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_ = 'This is an 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 type "'+e.ClassName+'"'); | |
end; | |
// simple things | |
StartTestGroup( 'simple Thrift calls'); | |
client.testVoid(); | |
Expect( TRUE, 'testVoid()'); // success := no exception | |
s := client.testString('Test'); | |
Expect( s = 'Test', 'testString(''Test'') = "'+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)); | |
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'); | |
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'); | |
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'); | |
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; | |
// 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'); | |
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'); | |
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)'); | |
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'); | |
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; | |
truck.I32_thing := 8; | |
truck.I64_thing := 8; | |
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('}'); | |
// 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.SIX, 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 = 2, 'crazy.UserMap.Count = '+IntToStr(crazy.UserMap.Count)); | |
Expect( crazy.UserMap[TNumberz.FIVE] = 5, 'crazy.UserMap[TNumberz.FIVE] = '+IntToStr(crazy.UserMap[TNumberz.FIVE])); | |
Expect( crazy.UserMap[TNumberz.EIGHT] = 8, 'crazy.UserMap[TNumberz.EIGHT] = '+IntToStr(crazy.UserMap[TNumberz.EIGHT])); | |
Expect( crazy.Xtructs.Count = 2, 'crazy.Xtructs.Count = '+IntToStr(crazy.Xtructs.Count)); | |
goodbye := crazy.Xtructs[0]; // lists are ordered, so we are allowed to assume this order | |
hello := crazy.Xtructs[1]; | |
Expect( goodbye.String_thing = 'Goodbye4', 'goodbye.String_thing = "'+goodbye.String_thing+'"'); | |
Expect( goodbye.Byte_thing = 4, 'goodbye.Byte_thing = '+IntToStr(goodbye.Byte_thing)); | |
Expect( goodbye.I32_thing = 4, 'goodbye.I32_thing = '+IntToStr(goodbye.I32_thing)); | |
Expect( goodbye.I64_thing = 4, 'goodbye.I64_thing = '+IntToStr(goodbye.I64_thing)); | |
Expect( goodbye.__isset_String_thing, 'goodbye.__isset_String_thing = '+BoolToString(goodbye.__isset_String_thing)); | |
Expect( goodbye.__isset_Byte_thing, 'goodbye.__isset_Byte_thing = '+BoolToString(goodbye.__isset_Byte_thing)); | |
Expect( goodbye.__isset_I32_thing, 'goodbye.__isset_I32_thing = '+BoolToString(goodbye.__isset_I32_thing)); | |
Expect( goodbye.__isset_I64_thing, 'goodbye.__isset_I64_thing = '+BoolToString(goodbye.__isset_I64_thing)); | |
Expect( hello.String_thing = 'hello', 'hello.String_thing = "'+hello.String_thing+'"'); | |
Expect( hello.Byte_thing = 2, 'hello.Byte_thing = '+IntToStr(hello.Byte_thing)); | |
Expect( hello.I32_thing = 2, 'hello.I32_thing = '+IntToStr(hello.I32_thing)); | |
Expect( hello.I64_thing = 2, 'hello.I64_thing = '+IntToStr(hello.I64_thing)); | |
Expect( hello.__isset_String_thing, 'hello.__isset_String_thing = '+BoolToString(hello.__isset_String_thing)); | |
Expect( hello.__isset_Byte_thing, 'hello.__isset_Byte_thing = '+BoolToString(hello.__isset_Byte_thing)); | |
Expect( hello.__isset_I32_thing, 'hello.__isset_I32_thing = '+BoolToString(hello.__isset_I32_thing)); | |
Expect( hello.__isset_I64_thing, 'hello.__isset_I64_thing = '+BoolToString(hello.__isset_I64_thing)); | |
end; | |
// multi args | |
StartTestGroup( 'testMulti'); | |
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)'); | |
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)); | |
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+'"'); | |
end; | |
StartTestGroup( 'testMultiException(Xception)'); | |
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+'"'); | |
end; | |
StartTestGroup( 'testMultiException(Xception2)'); | |
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)); | |
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+'"'); | |
end; | |
// oneway functions | |
StartTestGroup( 'Test Oneway(1)'); | |
client.testOneway(1); | |
Expect( TRUE, 'Test Oneway(1)'); // success := no exception | |
// call time | |
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' ); | |
// no more tests here | |
StartTestGroup( ''); | |
end; | |
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 : IList; | |
binary, binRead : 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") | |
begin | |
stm := TStringStream.Create; | |
try | |
StartTestGroup( 'JsonProtocolTest'); // no more tests here | |
// prepare binary data | |
SetLength( binary, $100); | |
for i := Low(binary) to High(binary) do binary[i] := i; | |
// output setup | |
prot := TJSONProtocolImpl.Create( | |
TStreamTransportImpl.Create( | |
nil, TThriftStreamAdapterDelphi.Create( stm, FALSE))); | |
// write | |
prot.WriteListBegin( TListImpl.Create( TType.String_, 9)); | |
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.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)); | |
// 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; | |
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'); | |
finally | |
stm.Free; | |
prot := nil; //-> Release | |
StartTestGroup( ''); // no more tests here | |
end; | |
end; | |
procedure TClientThread.StartTestGroup( const aGroup : string); | |
begin | |
FTestGroup := aGroup; | |
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); | |
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 asm int 3 end; | |
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)); | |
end; | |
Console.WriteLine(''); | |
end; | |
constructor TClientThread.Create(ATransport: ITransport; AProtocol : IProtocol; ANumIteration: Integer); | |
begin | |
inherited Create( True ); | |
FNumIteration := ANumIteration; | |
FTransport := ATransport; | |
FProtocol := AProtocol; | |
FConsole := TThreadConsole.Create( Self ); | |
// error list: keep correct order, allow for duplicates | |
FErrors := TStringList.Create; | |
FErrors.Sorted := FALSE; | |
FErrors.Duplicates := dupAccept; | |
end; | |
destructor TClientThread.Destroy; | |
begin | |
FreeAndNil( FConsole); | |
FreeAndNil( FErrors); | |
inherited; | |
end; | |
procedure TClientThread.Execute; | |
var | |
i : Integer; | |
proc : TThreadProcedure; | |
begin | |
// perform all tests | |
try | |
for i := 0 to FNumIteration - 1 do | |
begin | |
ClientTest; | |
JSONProtocolReadWriteTest; | |
end; | |
except | |
on e:Exception do Expect( FALSE, 'unexpected exception: "'+e.message+'"'); | |
end; | |
// report the outcome | |
ReportResults; | |
// shutdown | |
proc := procedure | |
begin | |
if FTransport <> nil then | |
begin | |
FTransport.Close; | |
FTransport := nil; | |
end; | |
end; | |
Synchronize( proc ); | |
end; | |
{ TThreadConsole } | |
constructor TThreadConsole.Create(AThread: TThread); | |
begin | |
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. |