Implement database config in project dir (#442)

Complements edgedb/edgedb-cli#1086
edgedb · Jun 18, 2023 · bee7327 · bee7327
1 parent 0f30b26
commit bee7327
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/edgedb/con_utils.py b/edgedb/con_utils.py
@@ -706,6 +706,12 @@ def _parse_connect_dsn_and_args(
                     f'project defined cloud profile ("{cloud_profile}")'
                 ),
             )
+
+            opt_database_file = os.path.join(stash_dir, 'database')
+            if os.path.exists(opt_database_file):
+                with open(opt_database_file, 'rt') as f:
+                    database = f.read().strip()
+                resolved_config.set_database(database, "project")
         else:
             raise errors.ClientConnectionError(
                 f'Found `edgedb.toml` but the project is not initialized. '

diff --git a/tests/shared-client-testcases b/tests/shared-client-testcases
diff --git a/tests/test_con_utils.py b/tests/test_con_utils.py
@@ -32,6 +32,7 @@
 
 
 class TestConUtils(unittest.TestCase):
+    maxDiff = 1000
 
     error_mapping = {
         'credentials_file_not_found': (
@@ -184,6 +185,9 @@ def run_testcase(self, testcase):
                             if 'cloud-profile' in v:
                                 profile = os.path.join(dir, 'cloud-profile')
                                 files[profile] = v['cloud-profile']
+                            if 'database' in v:
+                                database_file = os.path.join(dir, 'database')
+                                files[database_file] = v['database']
                             del files[f]
 
                     es.enter_context(