mirrors/core-lightning
1
0
Fork 0
mirror of https://github.com/ElementsProject/lightning.git synced 2025-03-12 02:08:15 +01:00
Code Issues 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
Show stats 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)