mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-07 02:45:22 +00:00
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
from util import add_common_args, init_protocol
|
|
from local_thrift import thrift # noqa
|
|
from thrift.Thrift import TMessageType, TType
|
|
|
|
|
|
# TODO: generate from ThriftTest.thrift
|
|
def test_string(proto, value):
|
|
method_name = 'testString'
|
|
ttype = TType.STRING
|
|
proto.writeMessageBegin(method_name, TMessageType.CALL, 3)
|
|
proto.writeStructBegin(method_name + '_args')
|
|
proto.writeFieldBegin('thing', ttype, 1)
|
|
proto.writeString(value)
|
|
proto.writeFieldEnd()
|
|
proto.writeFieldStop()
|
|
proto.writeStructEnd()
|
|
proto.writeMessageEnd()
|
|
proto.trans.flush()
|
|
|
|
_, mtype, _ = proto.readMessageBegin()
|
|
assert mtype == TMessageType.REPLY
|
|
proto.readStructBegin()
|
|
_, ftype, fid = proto.readFieldBegin()
|
|
assert fid == 0
|
|
assert ftype == ttype
|
|
result = proto.readString()
|
|
proto.readFieldEnd()
|
|
_, ftype, _ = proto.readFieldBegin()
|
|
assert ftype == TType.STOP
|
|
proto.readStructEnd()
|
|
proto.readMessageEnd()
|
|
assert value == result
|
|
|
|
|
|
def main(argv):
|
|
p = argparse.ArgumentParser()
|
|
add_common_args(p)
|
|
p.add_argument('--limit', type=int)
|
|
args = p.parse_args()
|
|
proto = init_protocol(args)
|
|
test_string(proto, 'a' * (args.limit - 1))
|
|
test_string(proto, 'a' * (args.limit - 1))
|
|
print('[OK]: limit - 1')
|
|
test_string(proto, 'a' * args.limit)
|
|
test_string(proto, 'a' * args.limit)
|
|
print('[OK]: just limit')
|
|
try:
|
|
test_string(proto, 'a' * (args.limit + 1))
|
|
except:
|
|
print('[OK]: limit + 1')
|
|
else:
|
|
print('[ERROR]: limit + 1')
|
|
assert False
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv[1:])
|