Fix the smoke test so it take the server URL from the config.toml (#3227)

# Description of Changes

Closes
https://github.com/clockworklabs/SpacetimeDBPrivate/issues/1998#event-19488677475

Now the smoke test parse the `config.toml` for the correct address for
connecting.

# API and ABI breaking changes

<!-- If this is an API or ABI breaking change, please apply the
corresponding GitHub label. -->

# Expected complexity level and risk
1
# Testing

- [x] Run the smoke both on public / private

Author: mamcx

smoketests/__init__.py

@@ -270,11 +270,7 @@ def run():
         # and **not raise any exceptions to the caller**.
         return ReturnThread(run).join
 
-    # Make an HTTP call with `method` to `path`.
-    #
-    # If the response is 200, return the body.
-    # Otherwise, throw an `Exception` constructed with two arguments, the response object and the body.
-    def api_call(self, method, path, body = None, headers = {}):
+    def get_server_address(self):
         with open(self.config_path, "rb") as f:
             config = tomllib.load(f)
             token = config['spacetimedb_token']
@@ -284,24 +280,35 @@ def api_call(self, method, path, body = None, headers = {}):
                 raise Exception(f"Unable to find server in config with nickname {server_name}")
             host = server_config['host']
             protocol = server_config['protocol']
-            conn = None
-            if protocol == "http":
-                conn = http.client.HTTPConnection(host)
-            elif protocol == "https":
-                conn = http.client.HTTPSConnection(host)
-            else:
-                raise Exception(f"Unknown protocol: {protocol}")
-            auth = {"Authorization": f'Bearer {token}'}
-            headers.update(auth)
-            log_cmd([method, path])
-            conn.request(method, path, body, headers)
-            resp = conn.getresponse()
-            body = resp.read()
-            logging.debug(f"{resp.status} {body}")
-            if resp.status != 200:
-                raise Exception(resp, body)
-            return body
 
+            return dict(host=host, protocol=protocol, token=token)
+
+    # Make an HTTP call with `method` to `path`.
+    #
+    # If the response is 200, return the body.
+    # Otherwise, throw an `Exception` constructed with two arguments, the response object and the body.
+    def api_call(self, method, path, body=None, headers={}):
+        server = self.get_server_address()
+        host = server["host"]
+        protocol = server["protocol"]
+        token = server["token"]
+        conn = None
+        if protocol == "http":
+            conn = http.client.HTTPConnection(host)
+        elif protocol == "https":
+            conn = http.client.HTTPSConnection(host)
+        else:
+            raise Exception(f"Unknown protocol: {protocol}")
+        auth = {"Authorization": f'Bearer {token}'}
+        headers.update(auth)
+        log_cmd([method, path])
+        conn.request(method, path, body, headers)
+        resp = conn.getresponse()
+        body = resp.read()
+        logging.debug(f"{resp.status} {body}")
+        if resp.status != 200:
+            raise Exception(resp, body)
+        return body
 
     @classmethod
     def write_module_code(cls, module_code):

smoketests/tests/quickstart.py

@@ -81,7 +81,6 @@ def _publish(self) -> Path:
         base_path = Path(self.enterClassContext(tempfile.TemporaryDirectory()))
         server_path = base_path / "server"
         self.project_path = server_path
-        self.config_path = server_path / "config.toml"
 
         self.generate_server(server_path)
         self.publish_module(f"quickstart-chat-{self.lang}", capture_stderr=True, clear=True)
@@ -125,6 +124,10 @@ def _test_quickstart(self):
         for src, dst in self.replacements.items():
             main = main.replace(src, dst)
         main += "\n" + self.extra_code
+        server = self.get_server_address()
+        host = server["host"]
+        protocol = server["protocol"]
+        main = main.replace("http://localhost:3000", f"{protocol}://{host}")
         _write_file(client_path / self.client_file, main)
 
         self.check("", client_path, self.connected_str)
@@ -142,9 +145,11 @@ class Rust(BaseQuickstart):
     run_cmd = ["cargo", "run"]
     build_cmd = ["cargo", "build"]
 
-    # Replace the interactive user input to allow direct testing
     replacements = {
-        "user_input_loop(&ctx)": "user_input_direct(&ctx)"
+        # Replace the interactive user input to allow direct testing
+        "user_input_loop(&ctx)": "user_input_direct(&ctx)",
+        # Don't cache the token, because it will cause the test to fail if we run against a non-default server (because  we don't cache the corresponding signing keypair)
+        ".with_token(creds_store()": "//.with_token(creds_store()"
     }
     extra_code = """
 fn user_input_direct(ctx: &DbConnection) {
@@ -194,6 +199,8 @@ class CSharp(BaseQuickstart):
         "InputLoop();": "UserInputDirect();",
         ".OnConnect(OnConnected)": ".OnConnect(OnConnectedSignal)",
         ".OnConnectError(OnConnectError)": ".OnConnectError(OnConnectErrorSignal)",
+        # Don't cache the token, because it will cause the test to fail if we run against a non-default server (because  we don't cache the corresponding signing keypair)
+        ".WithToken(AuthToken.Token)": "//.WithToken(AuthToken.Token)",
         "Main();": ""  # To put the main function at the end so it can see the new functions
     }
     # So we can wait for the connection to be established...