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")