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
