def GetBalance(self, request: qrl_pb2.GetBalanceReq,
context) -> qrl_pb2.GetBalanceResp:
logger.debug("[PublicAPI] GetBalance")
address_state = self.qrlnode.get_optimized_address_state(
request.address)
response = qrl_pb2.GetBalanceResp(balance=address_state.balance)
return response
def test_get_balance(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
walletd._public_stub.GetBalance = Mock(
return_value=qrl_pb2.GetBalanceResp(balance=1000))
balance = walletd.get_balance(self.qaddress)
self.assertEqual(balance, 1000)
def GetTotalBalance(self, request: qrl_pb2.GetTotalBalanceReq, context) -> qrl_pb2.GetTotalBalanceResp:
logger.debug("[PublicAPI] GetTotalBalance")
response = qrl_pb2.GetBalanceResp(balance=0)
for address in request.addresses:
address_state = self.qrlnode.get_address_state(address)
response.balance += address_state.balance
return response
def test_getBalance(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
service = WalletAPIService(walletd)
walletd._public_stub.GetBalance = Mock(
return_value=qrl_pb2.GetBalanceResp(balance=1000))
resp = service.GetBalance(qrlwallet_pb2.BalanceReq(address=self.qaddress), context=None)
self.assertEqual(resp.code, 0)
self.assertEqual(resp.balance, "1000")
