Fix some annoyances with the smoketests

Author: coolreader18

smoketests/__init__.py

@@ -22,7 +22,7 @@
 exe_suffix = ".exe" if sys.platform == "win32" else ""
 SPACETIME_BIN = STDB_DIR / ("target/debug/spacetime" + exe_suffix)
 TEMPLATE_TARGET_DIR = STDB_DIR / "target/_stdbsmoketests"
-STDB_CONFIG = TEST_DIR / "config.toml"
+BASE_STDB_CONFIG_PATH = TEST_DIR / "config.toml"
 
 # the contents of files for the base smoketest project template
 TEMPLATE_LIB_RS = open(STDB_DIR / "crates/cli/templates/basic-rust/server/src/lib.rs").read()
@@ -50,6 +50,9 @@
 # default value can be overridden by `--compose-file` flag
 COMPOSE_FILE = "./docker-compose.yml"
 
+# this will be initialized by main()
+STDB_CONFIG = None
+
 # we need to late-bind the output stream to allow unittests to capture stdout/stderr.
 class CapturableHandler(logging.StreamHandler):
 
@@ -253,7 +256,7 @@ def publish_module(self, domain=None, *, clear=True, capture_stderr=True, num_re
 
     @classmethod
     def reset_config(cls):
-        shutil.copy(STDB_CONFIG, cls.config_path)
+        cls.config_path.write_text(STDB_CONFIG)
 
     def fingerprint(self):
         # Fetch the server's fingerprint; required for `identity list`.

smoketests/__main__.py

@@ -7,11 +7,12 @@
 import re
 import fnmatch
 import json
-from . import TEST_DIR, SPACETIME_BIN, exe_suffix, build_template_target
+from . import TEST_DIR, SPACETIME_BIN, BASE_STDB_CONFIG_PATH, exe_suffix, build_template_target
 import smoketests
 import sys
 import logging
 import itertools
+import tempfile
 
 def check_docker():
     docker_ps = smoketests.run_cmd("docker", "ps", "--format=json")
@@ -112,16 +113,23 @@ def main():
                 subprocess.Popen(["docker", "logs", "-f", docker_container])
         smoketests.HAVE_DOCKER = True
 
+    config_file = tempfile.NamedTemporaryFile(mode="w+b", suffix=".toml", buffering=0)
+    config_file.write(BASE_STDB_CONFIG_PATH.read_bytes())
+    config_file.flush()
+
     if args.remote_server is not None:
-        smoketests.spacetime("--config-path", TEST_DIR / 'config.toml', "server", "edit", "localhost", "--url", args.remote_server, "--yes")
+        smoketests.spacetime("--config-path", config_file.name, "server", "edit", "localhost", "--url", args.remote_server, "--yes")
         smoketests.REMOTE_SERVER = True
 
     if args.spacetime_login:
-        smoketests.spacetime("--config-path", TEST_DIR / 'config.toml', "logout")
-        smoketests.spacetime("--config-path", TEST_DIR / 'config.toml', "login")
+        smoketests.spacetime("--config-path", config_file.name, "logout")
+        smoketests.spacetime("--config-path", config_file.name, "login")
         smoketests.USE_SPACETIME_LOGIN = True
     else:
-        smoketests.new_identity(TEST_DIR / 'config.toml')
+        smoketests.new_identity(config_file.name)
+
+    config_file.seek(0)
+    smoketests.STDB_CONFIG = config_file.read().decode()
 
     if not args.skip_dotnet:
         smoketests.HAVE_DOTNET = check_dotnet()

smoketests/config.toml

@@ -1,5 +1,4 @@
 default_server = "localhost"
-spacetimedb_token = "eyJ0eXAiOiJKV1QiLCJhbGciOiJFUzI1NiJ9.eyJoZXhfaWRlbnRpdHkiOiJjMjAwMTU3NGEwMjgyNDRjNzZhNTE1MjU1NGMzY2ZjMWJiNmIzNzZlNjY4YmU1Yjg2MzE0MDAyYWRmOTMyYWVlIiwic3ViIjoiYzYwOWJkYjUtMDAyNS00YzZkLWIyZTktOGYyODEwM2IzNWUzIiwiaXNzIjoibG9jYWxob3N0IiwiYXVkIjpbInNwYWNldGltZWRiIl0sImlhdCI6MTc1NjkwOTQ3NywiZXhwIjpudWxsfQ.t6Aobx9fTe6kwvq7H01-2RO7vdK4SjQB7Uw-Lh4Daz0lG43WzIw3oVG_65txqlsFSkpx40wYElByj4jMolutpA"
 
 [[server_configs]]
 nickname = "localhost"

smoketests/tests/replication.py

@@ -223,23 +223,18 @@ class ReplicationTest(Smoketest):
 
     @classmethod
     def setUpClass(cls):
-        super().setUpClass()
         cls.root_config = cls.project_path / "root_config"
 
-    def tearDown(self):
-        # Ensure containers that were brought down during a test are back up.
-        self.docker.compose("up", "-d")
-        super().tearDown()
-
-    # TODO: This function seems to run even when `--docker` is not passed, leading to errors unless `-x replication` is passed, due to the docker-related code below.
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-
+    def setUp(self):
         self.docker = DockerManager(COMPOSE_FILE)
         self.root_token = self.docker.generate_root_token()
 
         self.cluster = Cluster(self.docker, self)
 
+    def tearDown(self):
+        # Ensure containers that were brought down during a test are back up.
+        self.docker.compose("up", "-d")
+
     def add_me_as_admin(self):
         """Add the current user as an admin account"""
         db_owner_id = str(self.spacetime("login", "show")).split()[-1]