def test_dump_and_restore(self):
# dump the Gmf generated in setUpClass into a tarfile
hd = HazardDumper(self.conn)
hd.dump(self.hc.id)
tar = hd.mktar()
curs = self.conn.cursor()
try:
# delete the original hazard calculation
curs.execute("DELETE FROM uiapi.hazard_calculation " "WHERE id=%s", (self.hc.id,))
# restore the deleted Gmf
hazard_restore(self.conn, tar)
curs.execute("SELECT 1 FROM uiapi.hazard_calculation WHERE id=%s", (self.hc.id,))
self.assertEqual(curs.fetchall(), [(1,)])
finally:
os.remove(tar)
def test_dump_and_restore(self):
# dump the Gmf generated in setUpClass into a directory
hd = HazardDumper(self.conn)
hd.dump(self.hc.id)
curs = self.conn.cursor()
try:
# delete the original hazard calculation
curs.execute('DELETE FROM uiapi.hazard_calculation '
'WHERE id=%s', (self.hc.id,))
# restore the deleted Gmf
hazard_restore(self.conn, hd.outdir)
curs.execute('SELECT 1 FROM uiapi.hazard_calculation WHERE id=%s',
(self.hc.id,))
self.assertEqual(curs.fetchall(), [(1,)])
finally:
shutil.rmtree(hd.outdir)
def test_dump_and_restore(self):
# dump the Gmf generated in setUpClass into a directory
hd = HazardDumper(self.conn)
hd.dump(self.hc.id)
curs = self.conn.cursor()
try:
# delete the original hazard calculation
curs.execute('DELETE FROM uiapi.hazard_calculation '
'WHERE id=%s', (self.hc.id, ))
# restore the deleted Gmf
hazard_restore(self.conn, hd.outdir)
curs.execute('SELECT 1 FROM uiapi.hazard_calculation WHERE id=%s',
(self.hc.id, ))
self.assertEqual(curs.fetchall(), [(1, )])
finally:
shutil.rmtree(hd.outdir)