def test_on_class_methods(capsys):
with Probe(__name__) as probe, with_callback(PrintCallback()):
def assert_desired_state(desired_state, output):
result = probe.new_desired_state(desired_state)
obj = TestClass()
assert result is None
assert capsys.readouterr().out == ""
obj.sample(2)
traces = probe.traces[TRACE_COLUMNS].set_index('id')
assert all(traces == pd.DataFrame([
flatten_trace(trace) for trace in desired_state
]).set_index('id')), "traces is not set as the input desired_set"
assert capsys.readouterr().out == f"{output}\n"
assert_desired_state(
[
create_trace(
'inquest/test/sample.py',
'sample',
'{x}',
"3",
24,
'TestClass',
)
],
"2",
)
def test_on_function_simple(capsys):
with Probe(__name__) as probe, with_callback(PrintCallback()):
result = probe.new_desired_state([
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg1}',
"1",
1,
)
])
assert result is None
assert sample(2, 1) == 3
assert capsys.readouterr().out == "2\n"
assert sample(2, 1) == 3
assert capsys.readouterr().out == ""
def test_on_runtime_failure(capsys):
class TestCallback(Callback):
def log(self, value):
assert False
def error(self, trace_id, value):
assert value == "name 'arg0' is not defined\n"
with Probe(__name__) as probe, with_callback(TestCallback()):
result = probe.new_desired_state([
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg0}',
"1",
1,
)
])
assert result is None
assert sample(2, 1) == 3
assert capsys.readouterr().out == ""
def test_on_function_changes(capsys):
with Probe(__name__) as probe, with_callback(PrintCallback()):
def assert_desired_state(desired_state, output):
result = probe.new_desired_state(desired_state)
assert result is None
assert capsys.readouterr().out == ""
assert sample(2, 1) == 3
traces = probe.traces[TRACE_COLUMNS].set_index('id')
assert all(traces == pd.DataFrame([
flatten_trace(trace) for trace in desired_state
]).set_index('id')), "traces is not set as the input desired_set"
assert capsys.readouterr().out == f"{output}\n"
# testing duplicate
assert_desired_state([
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg1}',
"1",
1,
),
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg2}',
"2",
1,
)
], "1\n2")
assert_desired_state([
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg2}',
"2",
1,
)
], "1")
assert_desired_state([
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg2}',
"3",
1,
)
], "1")
assert_desired_state(
[
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg2} haha',
"3",
1,
)
],
"1 haha",
)
assert_desired_state(
[
create_trace(
'inquest/test/probe_test_module/test_imported_module.py',
'sample',
'{arg2}',
"3",
1,
)
],
"1",
)
assert all(
probe.traces == pd.DataFrame([], columns=TRACE_WITH_ERROR_COLUMNS))
assert sample(2, 1) == 3
captured = capsys.readouterr()
assert capsys.readouterr().out == ""
async def __aenter__(self):
await super().__aenter__()
self.log_queue = janus.Queue()
self.enter_context(with_callback(self.gen_callback()))
return self