aljaz/lightning
1
0
Fork 0
mirror of https://github.com/aljazceru/lightning.git synced 2025-12-19 23:24:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

pytest: allow db_query to manipulate db.

Browse Source 
Needed for the next test.

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
Rusty Russell
2018-07-31 14:04:12 +09:30
committed by Christian Decker
parent c5b5f74965
commit 6e10237d10
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/utils.py
View File

@@ -379,13 +379,16 @@ class LightningNode(object):
def getactivechannels(self): def getactivechannels(self):
return [c for c in self.rpc.listchannels()['channels'] if c['active']] return [c for c in self.rpc.listchannels()['channels'] if c['active']]
def db_query(self, query): def db_query(self, query, use_copy=True):
from shutil import copyfile
orig = os.path.join(self.daemon.lightning_dir, "lightningd.sqlite3") orig = os.path.join(self.daemon.lightning_dir, "lightningd.sqlite3")
copy = os.path.join(self.daemon.lightning_dir, "lightningd-copy.sqlite3") if use_copy:
copyfile(orig, copy) from shutil import copyfile
copy = os.path.join(self.daemon.lightning_dir, "lightningd-copy.sqlite3")
copyfile(orig, copy)
db = sqlite3.connect(copy)
else:
db = sqlite3.connect(orig)
db = sqlite3.connect(copy)
db.row_factory = sqlite3.Row db.row_factory = sqlite3.Row
c = db.cursor() c = db.cursor()
c.execute(query) c.execute(query)
@@ -395,6 +398,7 @@ class LightningNode(object):
for row in rows: for row in rows:
result.append(dict(zip(row.keys(), row))) result.append(dict(zip(row.keys(), row)))
db.commit()
c.close() c.close()
db.close() db.close()
return result return result