robosats/tests/mocks/lnd.py

from unittest.mock import MagicMock

# Mock up of LND gRPC responses
# Unfinished, during integration tests we spin up a regtest LND instance


class MockLightningStub:
    def __init__(self, channel):
        pass

    def GetInfo(self, request):
        response = MagicMock()
        response.testnet = True
        return response

    def EstimateFee(self, request):
        response = MagicMock()
        response.fee_sat = 1500
        response.sat_per_vbyte = 13
        return response

    def DecodePayReq(self, request):
        response = MagicMock()
        if (
            request.pay_req
            == "lntb17310n1pj552mdpp50p2utgzfwss8w6tvdssxyefqw4hxcmmrddjkggpgveskjmpfyp6kumr9wdejq7t0w5sxx6r9v96zqmmjyp3kzmnrv4kzqatwd9kxzar9wfskcmre9ccqz52xqzwzsp5hkzegrhn6kegr33z8qfxtcudaklugygdrakgyy7va0wt2qs7drfq9qyyssqc6rztchzl4m7mlulrhlcajszcl9fan8908k9n5x7gmz8g8d6ht5pj4l8r0dushq6j5s8x7yv9a5klz0kfxwy8v6ze6adyrrp4wu0q0sq3t604x"
        ):
            response.destination = (
                "033b58d7681fe5dd2fb21fd741996cda5449616f77317dd1156b80128d6a71b807"
            )
            response.payment_hash = (
                "7855c58afed86098f01ee7b0e857a5d2e51b1014adf176d82e38b38eecc5b7c6"
            )
            response.num_satoshis = 1731
            response.timestamp = 1699359597
            response.expiry = 450
            response.description = "Payment reference: 7458199b-87ba-4da7-8438-8469f7899da5. This payment WILL FREEZE IN YOUR WALLET, check on RoboSats if the lock was successful. It will be unlocked (fail) unless you cheat or cancel unilaterally."
            response.cltv_expiry = 650
            response.payment_addr = (
                "\275\205\224\016\363\325\262\201\353\334\265\002\036h\322"
            )
            response.num_msat = 1731000

            return response

    def CancelInvoice(self, request):
        response = MagicMock()
        if (
            request
            == b"xU\305\212\376\330`\230\360\036\347\260\350W\245\322\345\033\020\024\255\361v\330.8\263\216\354\305\267\306"
        ):
            response = {}
            return response

    def WalletBalance(self, request):
        response = MagicMock()
        response.total_balance = 10_000_000
        response.confirmed_balance = 9_000_000
        response.unconfirmed_balance = 1_000_000
        response.reserved_balance_anchor_chan = 30_000
        response.account_balance = {}
        return response

    def ChannelBalance(self, request):
        response = MagicMock()
        response.balance: 10_000_000
        response.local_balance.sat = 10_000_000
        response.local_balance.msat = 10_000_000_000
        response.remote_balance.sat = 30_000_000
        response.remote_balance.msat = 30_000_000_000
        response.unsettled_local_balance.sat = 500_000
        response.unsettled_local_balance.msat = 500_000_000
        response.unsettled_remote_balance.sat = 100_000
        response.unsettled_remote_balance.msat = 100_000_000
        response.pending_open_local_balance = 2_000_000
        response.pending_open_local_balance = 2_000_000_000
        response.pending_open_remote_balance = 5_000_000
        response.pending_open_remote_balance = 5_000_000_000
        return response

    def SendCoins(self, request):
        response = MagicMock()
        return response


class MockInvoicesStub:
    def __init__(self, channel):
        pass

    def AddHoldInvoice(self, request):
        response = MagicMock()
        print(request)
        response.payment_request = "lntb17310n1pj552mdpp50p2utzh7mpsf3uq7u7cws4a96tj3kyq54hchdkpw8zecamx9klrqd2j2pshjmt9de6zqun9vejhyetwvdjn5gphxs6nsvfe893z6wphvfsj6dryvymj6wp5xvuz6wp5xcukvdec8yukgcf49cs9g6rfwvs8qcted4jkuapq2ay5cnpqgefy2326g5syjn3qt984253q2aq5cnz92skzqcmgv43kkgr0dcs9ymmzdafkzarnyp5kvgr5dpjjqmr0vd4jqampwvs8xatrvdjhxumxw4kzugzfwss8w6tvdssxyefqw4hxcmmrddjkggpgveskjmpfyp6kumr9wdejq7t0w5sxx6r9v96zqmmjyp3kzmnrv4kzqatwd9kxzar9wfskcmre9ccqz52xqzwzsp5hkzegrhn6kegr33z8qfxtcudaklugygdrakgyy7va0wt2qs7drfq9qyyssqc6rztchzl4m7mlulrhlcajszcl9fan8908k9n5x7gmz8g8d6ht5pj4l8r0dushq6j5s8x7yv9a5klz0kfxwy8v6ze6adyrrp4wu0q0sq3t604x"
        response.add_index = 1
        response.payment_addr = b'\275\205\224\016\363\325\262\201\306"8\022e\343\215\355\277\304\021\r\037l\202\023\314\353\334\265\002\036h\322'
        return response

    def CancelInvoice(self, request):
        response = MagicMock()
        return response

    def SettleInvoice(self, request):
        response = MagicMock()
        return response

    def LookupInvoiceV2(self, request):
        response = MagicMock()
        if request.payment_hash == bytes.fromhex(
            "7855c58afed86098f01ee7b0e857a5d2e51b1014adf176d82e38b38eecc5b7c6"
        ):
            response.memo = "Payment reference: ..."
            response.state = 3  # "ACCEPTED"
            return response


class MockRouterStub:
    def ResetMissionControl(self, request):
        response = MagicMock()
        return response

    def SendPaymentV2(self, request):
        response = MagicMock()
        return response

    def TrackPaymentV2(self, request):
        response = MagicMock()
        return response


class MockSignerStub:
    def SignMessage(self, request):
        response = MagicMock()
        return response


class MockVersionerStub:
    def __init__(self, channel):
        pass

    def GetVersion(self, request):
        response = MagicMock()
        response.commit = "v0.17.0-beta"
        response.commit_hash = "2fb150c8fe827df9df0520ef9916b3afb7b03a8d"
        response.version = "0.17.0-beta"
        response.app_minor = 17
        response.app_patch = 0
        response.app_pre_release = "beta"
        response.build_tags = [
            "autopilotrpc",
            "signrpc",
            "walletrpc",
            "chainrpc",
            "invoicesrpc",
            "watchtowerrpc",
            "neutrinorpc",
            "monitoring",
            "peersrpc",
            "kvdb_postgres",
            "kvdb_etcd",
            "kvdb_sqlite",
            "go1.20.3",
        ]
        response.go_version = "go1.21.0"
        return response