return temporary dir object, not just the string, so that it doesnt get garbage collected early.

Author: michelp

postgresql/__init__.py

@@ -34,24 +34,24 @@ def f(cmdline):
 
 def setup(pgdata=None, log="db_test.log", user="postgres"):
     if pgdata is None:
-        pgdata = TemporaryDirectory().name
+        pgdata = TemporaryDirectory()
 
     log = Path(log)
     try:
         log.unlink()
     except FileNotFoundError:
         pass
 
-    initdb(f"-D {pgdata} --auth-local=trust --no-sync -U postgres")
-    pg_ctl(f'-D {pgdata} -o "-k {pgdata} -h \\"\\"" -l {log} start')
+    initdb(f"-D {pgdata.name} --auth-local=trust --no-sync -U postgres")
+    pg_ctl(f'-D {pgdata.name} -o "-k {pgdata.name} -h \\"\\"" -l {log} start')
     sleep(3)
-    con_str = f"host={pgdata} user={user}"
+    con_str = f"host={pgdata.name} user={user}"
     return pgdata, con_str
 
 
 def teardown(pgdata):
-    msg = pg_ctl(f"-D {pgdata} stop")
-    shutil.rmtree(pgdata)
+    msg = pg_ctl(f"-D {pgdata.name} stop")
+    pgdata.cleanup()
     return msg