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)