aljaz/lightning
1
0
Fork 0
mirror of https://github.com/aljazceru/lightning.git synced 2025-12-20 15:44:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

pyln: Delete psql DBs after testing

Browse Source
This commit is contained in:
Christian Decker
2022-01-28 22:53:10 +01:00
parent 0924b477b2
commit 95eb868047
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
contrib/pyln-testing/pyln/testing/db.py
View File

@@ -51,6 +51,9 @@ class Sqlite3Db(object):
c.close() c.close()
db.close() db.close()
def stop(self):
pass
class PostgresDb(object): class PostgresDb(object):
def __init__(self, dbname, port): def __init__(self, dbname, port):
@@ -88,6 +91,15 @@ class PostgresDb(object):
with self.conn, self.conn.cursor() as cur: with self.conn, self.conn.cursor() as cur:
cur.execute(query) cur.execute(query)
def stop(self):
"""Clean up the database.
"""
self.conn.close()
conn = psycopg2.connect("dbname=postgres user=postgres host=localhost port={self.port}")
cur = conn.cursor()
cur.execute("DROP DATABASE {};".format(self.dbname))
cur.close()
class SqliteDbProvider(object): class SqliteDbProvider(object):
def __init__(self, directory: str) -> None: def __init__(self, directory: str) -> None:
@@ -208,3 +220,4 @@ class PostgresDbProvider(object):
# [1] https://www.postgresql.org/docs/9.1/server-shutdown.html # [1] https://www.postgresql.org/docs/9.1/server-shutdown.html
self.proc.send_signal(signal.SIGINT) self.proc.send_signal(signal.SIGINT)
self.proc.wait() self.proc.wait()
shutil.rmtree(self.pgdir)