def test_print_queries_without_tracebacks(self): da.start_sql_logging() self.execute() result = da.stop_sql_logging() with stdout() as file: da.print_queries(result) self.assertEqual( '1-2@SQL-stub-database SELECT * FROM bar WHERE bing = 42\n' + "-" * 70 + "\n", file.getvalue())
def test_print_queries_with_tracebacks(self): da.start_sql_logging(tracebacks_if=True) self.execute() result = da.stop_sql_logging() with stdout() as file: da.print_queries(result) self.assertStartsWith(file.getvalue(), ' File "') self.assertEndsWith( file.getvalue(), "." * 70 + "\n" + '1-2@SQL-stub-database SELECT * FROM bar WHERE bing = 42\n' + "-" * 70 + "\n")
def test_print_queries_with_exceptions(self): def ow(sql): raise ValueError('rutebega') da.start_sql_logging(tracebacks_if=ow) self.execute() result = da.stop_sql_logging() with stdout() as file: da.print_queries(result) self.assertStartsWith( file.getvalue(), 'Error when determining whether to generate a stacktrace.\n' + 'Traceback (most recent call last):\n' + ' File "') self.assertEndsWith( file.getvalue(), "ValueError: rutebega\n" + "." * 70 + "\n" + '1-2@SQL-stub-database SELECT * FROM bar WHERE bing = 42\n' + "-" * 70 + "\n")