Add lite-mode

README.md

@@ -46,15 +46,34 @@ This command creates the project directory structure and installs `cairo-lang`,
 
 Run a local [`starknet-devnet`](https://github.com/Shard-Labs/starknet-devnet/) node:
 
-```sh
+```text
+nile node [--host HOST] [--port PORT] [--lite_mode]
+
+optional arguments:
+--host HOST         Specify the address to listen at; defaults to
+                    127.0.0.1 (use the address the program outputs on
+                    start)
+--port PORT         Specify the port to listen at; defaults to 5050
+--lite-mode         Applies all lite-mode optimizations by disabling
+                    features such as block hash calculation and deploy
+                    hash
+```
+
+```text
 nile node
 
- * Serving Flask app 'starknet_devnet.server' (lazy loading)
- * Environment: production
-   WARNING: This is a development server. Do not use it in a production deployment.
-   Use a production WSGI server instead.
- * Debug mode: off
- * Running on http://127.0.0.1:5050/ (Press CTRL+C to quit)
+Account #0
+Address: 0x877b050406a54adb5940227e51265a201e467e520ca85dc7f024abd03dcc61
+Public key: 0x256b8dc218586160ef80d3454a7cd51046271fbf091bd6779e3513304f22156
+Private key: 0xb204ff062d85674b467789f07826bb2
+
+...
+
+Initial balance of each account: 1000000000000000000000 WEI
+Seed to replicate this account sequence: 2128506880
+WARNING: Use these accounts and their keys ONLY for local testing. DO NOT use them on mainnet or other live networks because you will LOSE FUNDS.
+
+ * Listening on http://127.0.0.1:5050/ (Press CTRL+C to quit)
 ```
 
 ### `compile`

src/nile/cli.py

@@ -195,16 +195,20 @@ def clean():
 @cli.command()
 @click.option("--host", default="127.0.0.1")
 @click.option("--port", default=5050)
-def node(host, port):
+@click.option("--lite_mode", is_flag=True)
+def node(host, port, lite_mode):
     """Start StarkNet local network.
 
     $ nile node
       Start StarkNet local network at port 5050
 
     $ nile node --host HOST --port 5001
       Start StarkNet network on address HOST listening at port 5001
+
+    $ nile node --lite_mode
+      Start StarkNet network on lite-mode
     """
-    node_command(host, port)
+    node_command(host, port, lite_mode)
 
 
 @cli.command()

src/nile/core/node.py

@@ -1,11 +1,12 @@
 """Command to start StarkNet local network."""
 import json
+import logging
 import subprocess
 
 from nile.common import NODE_FILENAME
 
 
-def node(host="127.0.0.1", port=5050):
+def node(host="127.0.0.1", port=5050, lite_mode=False):
     """Start StarkNet local network."""
     try:
         # Save host and port information to be used by other commands
@@ -20,11 +21,16 @@ def node(host="127.0.0.1", port=5050):
         with open(file, "w+") as f:
             json.dump(gateway, f)
 
+        command = ["starknet-devnet", "--host", host, "--port", str(port)]
+
+        if lite_mode:
+            command.append("--lite-mode")
+
         # Start network
-        subprocess.check_call(["starknet-devnet", "--host", host, "--port", str(port)])
+        subprocess.check_call(command)
 
     except FileNotFoundError:
-        print("")
-        print("😰 Could not find starknet-devnet, is it installed? Try with:\n")
-        print("   pip install starknet-devnet")
-        print("")
+        logging.error(
+            "\n\n😰 Could not find starknet-devnet, is it installed? Try with:\n"
+            "    pip install starknet-devnet"
+        )

tests/commands/test_node.py

@@ -0,0 +1,48 @@
+"""Tests for node command."""
+import logging
+from unittest.mock import patch
+
+import pytest
+
+from nile.core.node import node
+
+HOSTS = ["127.0.0.1", "goerli"]
+PORTS = ["5050", "5001"]
+LITE = "--lite-mode"
+
+
+@pytest.fixture(autouse=True)
+def tmp_working_dir(monkeypatch, tmp_path):
+    monkeypatch.chdir(tmp_path)
+    return tmp_path
+
+
+@pytest.mark.parametrize(
+    "args, host, port, mode",
+    [
+        ([], HOSTS[0], PORTS[0], None),
+        ([HOSTS[1], PORTS[1]], HOSTS[1], PORTS[1], None),
+        ([HOSTS[1], PORTS[1], LITE], HOSTS[1], PORTS[1], LITE),
+    ],
+)
+@patch("nile.core.node.subprocess.check_call")
+def test_node_call(mock_subprocess, args, host, port, mode):
+    node(*args)
+
+    command = ["starknet-devnet", "--host", host, "--port", port]
+    if mode:
+        command.append(mode)
+    mock_subprocess.assert_called_once_with(command)
+
+
+@patch("nile.core.node.subprocess.check_call")
+def test_node_error(mock_subprocess, caplog):
+    logging.getLogger().setLevel(logging.INFO)
+
+    mock_subprocess.side_effect = FileNotFoundError
+
+    node()
+    assert (
+        "\n\n😰 Could not find starknet-devnet, is it installed? Try with:\n"
+        "    pip install starknet-devnet"
+    ) in caplog.text