def setUp(self): super(DebugTracerTest, self).setUp() self.stream = self.mocker.mock(file) self.tracer = DebugTracer(self.stream) datetime_mock = self.mocker.replace("datetime.datetime") datetime_mock.now() self.mocker.result(datetime.datetime(1, 2, 3, 4, 5, 6, 7)) self.mocker.count(0, 1) self.variable = MockVariable("PARAM")
def debug_tracer(stream): return DebugTracer(stream)
def test_debug_tracer_uses_first_arg_as_stream(): marker = object() tracer = DebugTracer(marker) assert tracer._stream == marker
def test_debug_tracer_uses_stderr_by_default(): tracer = DebugTracer() assert tracer._stream == sys.stderr
class DebugTracerTest(TestHelper): def setUp(self): super(DebugTracerTest, self).setUp() self.stream = self.mocker.mock(file) self.tracer = DebugTracer(self.stream) datetime_mock = self.mocker.replace("datetime.datetime") datetime_mock.now() self.mocker.result(datetime.datetime(1, 2, 3, 4, 5, 6, 7)) self.mocker.count(0, 1) self.variable = MockVariable("PARAM") def tearDown(self): del _tracers[:] super(DebugTracerTest, self).tearDown() def test_wb_debug_tracer_uses_stderr_by_default(self): self.mocker.replay() tracer = DebugTracer() self.assertEqual(tracer._stream, sys.stderr) def test_wb_debug_tracer_uses_first_arg_as_stream(self): self.mocker.replay() marker = object() tracer = DebugTracer(marker) self.assertEqual(tracer._stream, marker) def test_connection_raw_execute(self): self.stream.write( "[04:05:06.000007] EXECUTE: 'STATEMENT', ('PARAM',)\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = [self.variable] self.tracer.connection_raw_execute(connection, raw_cursor, statement, params) def test_connection_raw_execute_with_non_variable(self): self.stream.write( "[04:05:06.000007] EXECUTE: 'STATEMENT', ('PARAM', 1)\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = [self.variable, 1] self.tracer.connection_raw_execute(connection, raw_cursor, statement, params) def test_connection_raw_execute_error(self): self.stream.write("[04:05:06.000007] ERROR: ERROR\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = "PARAMS" error = "ERROR" self.tracer.connection_raw_execute_error(connection, raw_cursor, statement, params, error) def test_connection_raw_execute_success(self): self.stream.write("[04:05:06.000007] DONE\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = "PARAMS" self.tracer.connection_raw_execute_success(connection, raw_cursor, statement, params)
def test_wb_debug_tracer_uses_first_arg_as_stream(self): self.mocker.replay() marker = object() tracer = DebugTracer(marker) self.assertEqual(tracer._stream, marker)
def test_wb_debug_tracer_uses_stderr_by_default(self): self.mocker.replay() tracer = DebugTracer() self.assertEqual(tracer._stream, sys.stderr)
class DebugTracerTest(TestHelper): def setUp(self): super(DebugTracerTest, self).setUp() self.stream = self.mocker.mock(file) self.tracer = DebugTracer(self.stream) datetime_mock = self.mocker.replace("datetime.datetime") datetime_mock.now() self.mocker.result(datetime.datetime(1, 2, 3, 4, 5, 6, 7)) self.mocker.count(0, 1) self.variable = MockVariable("PARAM") def tearDown(self): del _tracers[:] super(DebugTracerTest, self).tearDown() def test_wb_debug_tracer_uses_stderr_by_default(self): self.mocker.replay() tracer = DebugTracer() self.assertEqual(tracer._stream, sys.stderr) def test_wb_debug_tracer_uses_first_arg_as_stream(self): self.mocker.replay() marker = object() tracer = DebugTracer(marker) self.assertEqual(tracer._stream, marker) def test_connection_raw_execute(self): self.stream.write( "[04:05:06.000007] EXECUTE: 'STATEMENT', ('PARAM',)\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = [self.variable] self.tracer.connection_raw_execute(connection, raw_cursor, statement, params) def test_connection_raw_execute_with_non_variable(self): self.stream.write( "[04:05:06.000007] EXECUTE: 'STATEMENT', ('PARAM', 1)\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = [self.variable, 1] self.tracer.connection_raw_execute(connection, raw_cursor, statement, params) def test_connection_raw_execute_error(self): self.stream.write("[04:05:06.000007] ERROR: ERROR\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = "PARAMS" error = "ERROR" self.tracer.connection_raw_execute_error(connection, raw_cursor, statement, params, error) def test_connection_raw_execute_success(self): self.stream.write("[04:05:06.000007] DONE\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" raw_cursor = "RAW_CURSOR" statement = "STATEMENT" params = "PARAMS" self.tracer.connection_raw_execute_success(connection, raw_cursor, statement, params) def test_connection_commit(self): self.stream.write("[04:05:06.000007] COMMIT xid=None\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" self.tracer.connection_commit(connection) def test_connection_rollback(self): self.stream.write("[04:05:06.000007] ROLLBACK xid=None\n") self.stream.flush() self.mocker.replay() connection = "CONNECTION" self.tracer.connection_rollback(connection)