[Fix #710] Use network_id to identify faucet_url

tests/unit/asyn/wallet/test_wallet.py

@@ -3,6 +3,7 @@
 from xrpl.asyncio.wallet.wallet_generation import (
     _DEV_FAUCET_URL,
     _TEST_FAUCET_URL,
+    XRPLFaucetException,
     get_faucet_url,
     process_faucet_host_url,
 )
@@ -19,27 +20,45 @@ def test_wallet_get_xaddress(self):
     def test_get_faucet_wallet_dev(self):
         json_client_url = "https://s.devnet.rippletest.net:51234"
         ws_client_url = "wss://s.devnet.rippletest.net/"
-        expected_faucet = _DEV_FAUCET_URL
 
-        self.assertEqual(get_faucet_url(json_client_url), expected_faucet)
-        self.assertEqual(get_faucet_url(ws_client_url), expected_faucet)
+        self.assertEqual(get_faucet_url(2), _DEV_FAUCET_URL)
+        self.assertEqual(
+            get_faucet_url(network_id=None, faucet_host=json_client_url),
+            "https://s.devnet.rippletest.net:51234/accounts",
+        )
+        self.assertEqual(
+            get_faucet_url(network_id=None, faucet_host=ws_client_url),
+            "wss://s.devnet.rippletest.net/accounts",
+        )
 
     def test_get_faucet_wallet_custom(self):
-        json_client_url = "https://s.devnet.rippletest.net:51234"
-        ws_client_url = "wss://s.devnet.rippletest.net/"
         custom_host = "my_host.org"
         expected_faucet = "https://my_host.org/accounts"
 
-        self.assertEqual(get_faucet_url(json_client_url, custom_host), expected_faucet)
-        self.assertEqual(get_faucet_url(ws_client_url, custom_host), expected_faucet)
+        self.assertEqual(
+            get_faucet_url(network_id=12345, faucet_host=custom_host), expected_faucet
+        )
 
     def test_get_faucet_wallet_test(self):
         json_client_url = "https://testnet.xrpl-labs.com"
         ws_client_url = "wss://testnet.xrpl-labs.com"
-        expected_faucet = _TEST_FAUCET_URL
 
-        self.assertEqual(get_faucet_url(json_client_url), expected_faucet)
-        self.assertEqual(get_faucet_url(ws_client_url), expected_faucet)
+        self.assertEqual(get_faucet_url(1), _TEST_FAUCET_URL)
+        self.assertEqual(
+            get_faucet_url(network_id=None, faucet_host=json_client_url),
+            "https://testnet.xrpl-labs.com/accounts",
+        )
+        self.assertEqual(
+            get_faucet_url(network_id=None, faucet_host=ws_client_url),
+            "wss://testnet.xrpl-labs.com/accounts",
+        )
+
+    def test_get_faucet_wallet_invalid(self):
+        with self.assertRaises(XRPLFaucetException):
+            get_faucet_url(0)  # corresponds to mainnet
+
+        with self.assertRaises(XRPLFaucetException):
+            get_faucet_url(-1)  # network_id must be non-negative
 
 
 class TestProcessFaucetHostURL(TestCase):

xrpl/asyncio/clients/client.py

@@ -1,11 +1,14 @@
 """Interface for all network clients to follow."""
+
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
 from typing import Optional
 
 from typing_extensions import Final
 
+from xrpl.asyncio.clients.exceptions import XRPLRequestFailureException
+from xrpl.models.requests import ServerInfo
 from xrpl.models.requests.request import Request
 from xrpl.models.response import Response
 
@@ -51,3 +54,24 @@ async def _request_impl(
         :meta private:
         """
         pass
+
+
+async def _get_network_id_and_build_version(client: Client) -> None:
+    """
+    Get the network id and build version of the connected server.
+
+    Args:
+        client: The network client to use to send the request.
+
+    Raises:
+        XRPLRequestFailureException: if the rippled API call fails.
+    """
+    response = await client._request_impl(ServerInfo())
+    if response.is_successful():
+        if "network_id" in response.result["info"]:
+            client.network_id = response.result["info"]["network_id"]
+        if not client.build_version and "build_version" in response.result["info"]:
+            client.build_version = response.result["info"]["build_version"]
+        return
+
+    raise XRPLRequestFailureException(response.result)

xrpl/asyncio/transaction/main.py

@@ -1,17 +1,19 @@
 """High-level transaction methods with XRPL transactions."""
+
 import math
 from typing import Any, Dict, Optional, cast
 
 from typing_extensions import Final
 
 from xrpl.asyncio.account import get_next_valid_seq_number
 from xrpl.asyncio.clients import Client, XRPLRequestFailureException
+from xrpl.asyncio.clients.client import _get_network_id_and_build_version
 from xrpl.asyncio.ledger import get_fee, get_latest_validated_ledger_sequence
 from xrpl.constants import XRPLException
 from xrpl.core.addresscodec import is_valid_xaddress, xaddress_to_classic_address
 from xrpl.core.binarycodec import encode, encode_for_multisigning, encode_for_signing
 from xrpl.core.keypairs.main import sign as keypairs_sign
-from xrpl.models.requests import ServerInfo, ServerState, SubmitOnly
+from xrpl.models.requests import ServerState, SubmitOnly
 from xrpl.models.response import Response
 from xrpl.models.transactions import EscrowFinish
 from xrpl.models.transactions.transaction import Signer, Transaction
@@ -247,27 +249,6 @@ async def autofill(
     return Transaction.from_dict(transaction_json)
 
 
-async def _get_network_id_and_build_version(client: Client) -> None:
-    """
-    Get the network id and build version of the connected server.
-
-    Args:
-        client: The network client to use to send the request.
-
-    Raises:
-        XRPLRequestFailureException: if the rippled API call fails.
-    """
-    response = await client._request_impl(ServerInfo())
-    if response.is_successful():
-        if "network_id" in response.result["info"]:
-            client.network_id = response.result["info"]["network_id"]
-        if not client.build_version and "build_version" in response.result["info"]:
-            client.build_version = response.result["info"]["build_version"]
-        return
-
-    raise XRPLRequestFailureException(response.result)
-
-
 def _tx_needs_networkID(client: Client) -> bool:
     """
     Determines whether the transactions required network ID to be valid.

xrpl/asyncio/wallet/wallet_generation.py

@@ -1,4 +1,5 @@
 """Handles wallet generation from a faucet."""
+
 import asyncio
 from typing import Optional
 from urllib.parse import urlparse, urlunparse
@@ -8,6 +9,7 @@
 
 from xrpl.asyncio.account import get_balance, get_next_valid_seq_number
 from xrpl.asyncio.clients import Client, XRPLRequestFailureException
+from xrpl.asyncio.clients.client import _get_network_id_and_build_version
 from xrpl.constants import XRPLException
 from xrpl.wallet.main import Wallet
 
@@ -57,7 +59,9 @@ async def generate_faucet_wallet(
 
     .. # noqa: DAR402 exception raised in private method
     """
-    faucet_url = get_faucet_url(client.url, faucet_host)
+    if not client.network_id:
+        await _get_network_id_and_build_version(client)
+    faucet_url = get_faucet_url(client.network_id, faucet_host)
 
     if wallet is None:
         wallet = Wallet.create()
@@ -150,34 +154,40 @@ def process_faucet_host_url(input_url: str) -> str:
     return final_url
 
 
-def get_faucet_url(url: str, faucet_host: Optional[str] = None) -> str:
+def get_faucet_url(network_id: Optional[int], faucet_host: Optional[str] = None) -> str:
     """
-    Returns the URL of the faucet that should be used, based on whether the URL is from
-    a testnet or devnet client.
+    Returns the URL of the faucet that should be used, based on network_id and
+    faucet_host
 
     Args:
-        url: The URL that the client is using to access the ledger.
-        faucet_host: A custom host to use for funding a wallet.
+        network_id: The network_id corresponding to the XRPL network. This is parsed
+            from a ServerInfo() rippled response
+        faucet_host: A custom host to use for funding a wallet. This is useful because
+            network_id of sidechains might not be well-known. If faucet_url cannot be
+            gleaned from network_id alone, faucet_host is used
 
     Returns:
         The URL of the matching faucet.
 
     Raises:
-        XRPLFaucetException: if the provided URL is not for the testnet or devnet.
+        XRPLFaucetException: if the provided network_id does not correspond to testnet
+            or devnet.
     """
+    if network_id is not None:
+        if network_id == 1:
+            return _TEST_FAUCET_URL
+        elif network_id == 2:
+            return _DEV_FAUCET_URL
+        elif network_id == 0:
+            raise XRPLFaucetException("Cannot create faucet with a client on mainnet.")
+        elif network_id < 0:
+            raise XRPLFaucetException("network_id cannot be negative.")
+
     if faucet_host is not None:
         return process_faucet_host_url(faucet_host)
-    if "altnet" in url or "testnet" in url:  # testnet
-        return _TEST_FAUCET_URL
-    if "sidechain-net2" in url:  # sidechain issuing chain devnet
-        raise XRPLFaucetException(
-            "Cannot fund an account on an issuing chain. Accounts must be created via "
-            "the bridge."
-        )
-    if "devnet" in url:  # devnet
-        return _DEV_FAUCET_URL
+
     raise XRPLFaucetException(
-        "Cannot fund an account with a client that is not on the testnet or devnet."
+        "Cannot create faucet URL without network_id or the faucet_host information."
     )