wallet-core/tests/components/wallet.py

51 lines
1.7 KiB
Python
Raw Normal View History

import json
import os
from subprocess import run
class Wallet:
def __init__(self, config):
self.db = os.path.join(config.tmpdir, "wallet-db.json")
self.arg_db = "--wallet-db=%s" % self.db
self.log_path = os.path.join(config.tmpdir, "wallet.log")
def cmd(self, command, request=None):
if request is None:
request = dict()
request = json.dumps(request)
r = run(["taler-wallet-cli", self.arg_db, "api", command, request],
timeout=10, text=True, capture_output=True)
self.write_to_log(r.stderr)
if r.returncode != 0:
print(r)
assert r.returncode == 0
json_r = json.loads(r.stdout)
if json_r["isError"]:
print(r)
assert not json_r["isError"]
if "result" not in json_r:
# TODO should there not always be a "result"?
return None
return json_r["result"]
def testing_withdraw(self, amount, exchange_url, bank_url):
r = run(["taler-wallet-cli", self.arg_db, "--no-throttle", "testing", "withdraw",
"-a", amount,
"-e", exchange_url,
"-b", bank_url
], timeout=10, check=True, text=True, capture_output=True)
self.write_to_log(r.stderr)
def gen_withdraw_uri(self, amount, bank_url):
r = run(["taler-wallet-cli", self.arg_db, "testing", "gen-withdraw-uri",
"-a", amount,
"-b", bank_url
], timeout=10, check=True, text=True, capture_output=True)
self.write_to_log(r.stderr)
return r.stdout.rstrip()
def write_to_log(self, data):
with open(self.log_path, "a") as f:
f.write(data)