diff --git a/test.py b/test.py index 3b5be2b4fa..7c2488bb86 100755 --- a/test.py +++ b/test.py @@ -6,50 +6,16 @@ from transport_pipe import PipeTransport from transport_serial import SerialTransport from bitkey_proto import bitkey_pb2 as proto -def pprint(msg): - return "<%s>:\n%s" % (msg.__class__.__name__, msg) +from client import BitkeyClient -def call(msg, tries=3): - print '----------------------' - print "Sending", pprint(msg) - d.write(msg) - resp = d.read() - - if isinstance(resp, proto.OtpRequest): - if resp.message: - print "Message:", resp.message - otp = raw_input("OTP required: ") - d.write(proto.OtpAck(otp=otp)) - resp = d.read() +bitkey = BitkeyClient('../bitkey-python/device.socket', debug=True) +bitkey.open() +bitkey.call(proto.Ping(message='ahoj!')) +bitkey.call(proto.SetMaxFeeKb(maxfee_kb=200000)) +bitkey.close() - if isinstance(resp, proto.PinRequest): - if resp.message: - print "Message:", resp.message - pin = raw_input("PIN required: ") - d.write(proto.PinAck(pin=pin)) - resp = d.read() - - if isinstance(resp, proto.Failure) and resp.code in (3, 6): - if tries <= 1 and resp.code == 3: - raise Exception("OTP is invalid, too many retries") - if tries <= 1 and resp.code == 6: - raise Exception("PIN is invalid, too many retries") - - # Invalid OTP or PIN, try again - if resp.code == 3: - print "OTP is invalid, let's try again..." - elif resp.code == 6: - print "PIN is invalid, let's try again..." - - return call(msg, tries-1) - - if isinstance(resp, proto.Failure): - raise Exception(resp.code, resp.message) - - print "Received", pprint(resp) - return resp - -d = PipeTransport('../../bitkey-python/device.socket', is_device=False) +''' +d = PipeTransport('../bitkey-python/device.socket', is_device=False) #d = SerialTransport('../../bitkey-python/COM9') #start = time.time() @@ -66,5 +32,6 @@ call(proto.GetUUID()) #call(proto.ResetDevice()) call(proto.GetMasterPublicKey(algo=proto.ELECTRUM)) #call(proto.ResetDevice()) - +''' + #print 10000 / (time.time() - start)