Add faucet auth token to tests (#39)

examples/common.py

@@ -13,6 +13,7 @@
     "APTOS_FAUCET_URL",
     "https://faucet.devnet.aptoslabs.com",
 )
+FAUCET_AUTH_TOKEN = os.getenv("FAUCET_AUTH_TOKEN")
 INDEXER_URL = os.getenv(
     "APTOS_INDEXER_URL",
     "https://api.devnet.aptoslabs.com/v1/graphql",

examples/fee_payer_transfer_coin.py

@@ -15,13 +15,15 @@
     TransactionPayload,
 )
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def main():
     # :!:>section_1
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)  # <:!:section_1
+    faucet_client = FaucetClient(
+        FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN
+    )  # <:!:section_1
 
     # :!:>section_2
     alice = Account.generate()

examples/hello_blockchain.py

@@ -30,7 +30,7 @@
     TransactionPayload,
 )
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 class HelloBlockchainClient(RestClient):
@@ -65,7 +65,7 @@ async def set_message(
 async def publish_contract(package_dir: str) -> AccountAddress:
     contract_publisher = Account.generate()
     rest_client = HelloBlockchainClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
     await faucet_client.fund_account(contract_publisher.address(), 10_000_000)
 
     AptosCLIWrapper.compile_package(
@@ -104,7 +104,7 @@ async def main(contract_address: AccountAddress):
     print(f"Bob: {bob.address()}")
 
     rest_client = HelloBlockchainClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     alice_fund = faucet_client.fund_account(alice.address(), 10_000_000)
     bob_fund = faucet_client.fund_account(bob.address(), 10_000_000)

examples/large_package_publisher.py

@@ -19,12 +19,12 @@
 from aptos_sdk.async_client import ClientConfig, FaucetClient, RestClient
 from aptos_sdk.package_publisher import MODULE_ADDRESS, PackagePublisher
 
-from .common import APTOS_CORE_PATH, FAUCET_URL, NODE_URL
+from .common import APTOS_CORE_PATH, FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def publish_large_packages(large_packages_dir) -> AccountAddress:
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     alice = Account.generate()
     await faucet_client.fund_account(alice.address(), 1_000_000_000)
@@ -42,7 +42,7 @@ async def main(
     client_config.transaction_wait_in_seconds = 120
     client_config.max_gas_amount = 1_000_000
     rest_client = RestClient(NODE_URL, client_config)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     alice = Account.generate()
     req0 = faucet_client.fund_account(alice.address(), 1_000_000_000)

examples/multikey.py

@@ -17,13 +17,15 @@
     TransactionPayload,
 )
 
-from .common import FAUCET_URL, INDEXER_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, INDEXER_URL, NODE_URL
 
 
 async def main():
     # :!:>section_1
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)  # <:!:section_1
+    faucet_client = FaucetClient(
+        FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN
+    )  # <:!:section_1
     if INDEXER_URL and INDEXER_URL != "none":
         IndexerClient(INDEXER_URL)
     else:

examples/multisig.py

@@ -22,7 +22,7 @@
 )
 from aptos_sdk.type_tag import StructTag, TypeTag
 
-from .common import APTOS_CORE_PATH, FAUCET_URL, NODE_URL
+from .common import APTOS_CORE_PATH, FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 should_wait = True
 
@@ -38,7 +38,7 @@ async def main(should_wait_input=True):
     should_wait = should_wait_input
 
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     # :!:>section_1
     alice = Account.generate()

examples/object_code_deployment.py

@@ -10,12 +10,12 @@
 from aptos_sdk.async_client import FaucetClient, RestClient
 from aptos_sdk.package_publisher import MODULE_ADDRESS, PackagePublisher, PublishMode
 
-from .common import APTOS_CORE_PATH, FAUCET_URL, NODE_URL
+from .common import APTOS_CORE_PATH, FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def main(package_dir):
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
     package_publisher = PackagePublisher(rest_client)
     alice = Account.generate()
 

examples/rotate_key.py

@@ -14,7 +14,7 @@
     TransactionPayload,
 )
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 WIDTH = 19
 
@@ -125,7 +125,7 @@ def rotation_payload(
 async def main():
     # Initialize the clients used to interact with the blockchain
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     # Generate random accounts Alice and Bob
     alice = Account.generate()

examples/secp256k1_ecdsa_transfer_coin.py

@@ -6,13 +6,15 @@
 from aptos_sdk.account import Account
 from aptos_sdk.async_client import FaucetClient, RestClient
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def main():
     # :!:>section_1
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)  # <:!:section_1
+    faucet_client = FaucetClient(
+        FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN
+    )  # <:!:section_1
 
     # :!:>section_2
     alice = Account.generate_secp256k1_ecdsa()

examples/simple_aptos_token.py

@@ -16,7 +16,7 @@
 )
 from aptos_sdk.async_client import FaucetClient, RestClient
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 def get_owner(obj: ReadObject) -> AccountAddress:
@@ -54,7 +54,9 @@ async def main():
     # Create API and faucet clients.
     # :!:>section_1a
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)  # <:!:section_1a
+    faucet_client = FaucetClient(
+        FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN
+    )  # <:!:section_1a
 
     # Create client for working with the token module.
     # :!:>section_1b

examples/simple_nft.py

@@ -8,12 +8,12 @@
 from aptos_sdk.aptos_tokenv1_client import AptosTokenV1Client
 from aptos_sdk.async_client import FaucetClient, RestClient
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def main():
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
     token_client = AptosTokenV1Client(rest_client)
 
     # :!:>section_2

examples/simulate_transfer_coin.py

@@ -14,12 +14,14 @@
 )
 from aptos_sdk.type_tag import StructTag, TypeTag
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def main():
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)  # <:!:section_1
+    faucet_client = FaucetClient(
+        FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN
+    )  # <:!:section_1
 
     alice = Account.generate()
     bob = Account.generate()

examples/transaction_batching.py

@@ -24,7 +24,7 @@
     TransactionPayload,
 )
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 def generate_rest_client(node_url: str) -> RestClient:
@@ -295,7 +295,7 @@ def load(path: str, num_accounts: int) -> Accounts:
 
 
 async def fund_from_faucet(rest_client: RestClient, source: Account):
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     fund_txns = []
     for _ in range(40):

examples/transfer_coin.py

@@ -6,13 +6,15 @@
 from aptos_sdk.account import Account
 from aptos_sdk.async_client import FaucetClient, IndexerClient, RestClient
 
-from .common import FAUCET_URL, INDEXER_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, INDEXER_URL, NODE_URL
 
 
 async def main():
     # :!:>section_1
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)  # <:!:section_1
+    faucet_client = FaucetClient(
+        FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN
+    )  # <:!:section_1
     if INDEXER_URL and INDEXER_URL != "none":
         indexer_client = IndexerClient(INDEXER_URL)
     else:

examples/transfer_two_by_two.py

@@ -8,12 +8,12 @@
 from aptos_sdk.async_client import FaucetClient, RestClient
 from aptos_sdk.transactions import Script, ScriptArgument, TransactionPayload
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 async def main():
     rest_client = RestClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     alice = Account.generate()
     bob = Account.generate()

examples/your_coin.py

@@ -28,7 +28,7 @@
 )
 from aptos_sdk.type_tag import StructTag, TypeTag
 
-from .common import FAUCET_URL, NODE_URL
+from .common import FAUCET_AUTH_TOKEN, FAUCET_URL, NODE_URL
 
 
 class CoinClient(RestClient):
@@ -88,7 +88,7 @@ async def main(moon_coin_path: str):
     print(f"Bob: {bob.address()}")
 
     rest_client = CoinClient(NODE_URL)
-    faucet_client = FaucetClient(FAUCET_URL, rest_client)
+    faucet_client = FaucetClient(FAUCET_URL, rest_client, FAUCET_AUTH_TOKEN)
 
     alice_fund = faucet_client.fund_account(alice.address(), 20_000_000)
     bob_fund = faucet_client.fund_account(bob.address(), 20_000_000)