def test_ignore_pg_error_other_exception(self): """ This isn't even a postgres error. Explode! """ err = Exception("Goodbye, cruel world.") d = ignore_pg_error(fail(err), errorcodes.UNDEFINED_TABLE) self.assertEqual(self.failureResultOf(d).value, err)
def test_ignore_pg_error_other_error(self): """ This is a different error. Explode! """ err = psycopg2.ProgrammingError() err.pgcode = errorcodes.INVALID_CATALOG_NAME d = ignore_pg_error(fail(err), errorcodes.UNDEFINED_TABLE) self.assertEqual(self.failureResultOf(d).value, err)
def test_ignore_pg_error_ignore(self): """ Ignore the error we're told to ignore. """ err = psycopg2.ProgrammingError() err.pgcode = errorcodes.UNDEFINED_TABLE d = ignore_pg_error(fail(err), errorcodes.UNDEFINED_TABLE) self.assertEqual(self.successResultOf(d), None)
def test_ignore_pg_error_no_err(self): """ Nothing to ignore, let's move on. """ d = ignore_pg_error(succeed("Yay"), errorcodes.UNDEFINED_TABLE) self.assertEqual(self.successResultOf(d), "Yay")
def _dropdb(self, plug, dbname): d = self.run_operation(plug, "DROP DATABASE %s;" % (dbname,)) ignore_pg_error(d, errorcodes.INVALID_CATALOG_NAME) return d
def cleanup_databases_table(self, plug): d = self.run_operation(plug, "DROP TABLE databases;") ignore_pg_error(d, errorcodes.UNDEFINED_TABLE) return d