def test_evaluate_ok_nomock(phase, monkeypatch, snapshot):
monkeypatch.setattr(
'satellite.operations.operations.get_proxy_context',
Mock(return_value=ProxyContext(
mode=ProxyMode.FORWARD,
port=9099,
)),
)
mock_emit = Mock()
monkeypatch.setattr(
'satellite.operations.pipeline.audit_logs.emit',
mock_emit,
)
flow = load_flow('http_raw')
operation = LegitMockOperation(route_id='route-id', filter_id='filter-id')
operation.evaluate(flow, phase)
snapshot.assert_match(flow.request.get_state(), name='request')
mock_emit.assert_called_once_with(
OperationLogRecord(
flow_id=flow.id,
proxy_mode=ProxyMode.FORWARD,
route_id='route-id',
filter_id='filter-id',
phase=phase,
operation_name='legit-test-operation',
execution_time_ms=ANY,
execution_time_ns=ANY,
status=OperationStatus.OK,
error_message=None,
))
def test_evaluate_error(monkeypatch):
monkeypatch.setattr(
'satellite.operations.operations.get_proxy_context',
Mock(return_value=ProxyContext(
mode=ProxyMode.FORWARD,
port=9099,
)),
)
monkeypatch.setattr(
'satellite.operations.operations.evaluate',
Mock(side_effect=Exception('test error')),
)
mock_emit = Mock()
monkeypatch.setattr(
'satellite.operations.pipeline.audit_logs.emit',
mock_emit,
)
flow = load_flow('http_raw')
operation = MockOperation(route_id='route-id', filter_id='filter-id')
operation.evaluate(flow, Phase.REQUEST)
mock_emit.assert_called_once_with(
OperationLogRecord(
flow_id=flow.id,
proxy_mode=ProxyMode.FORWARD,
route_id='route-id',
filter_id='filter-id',
phase=Phase.REQUEST,
operation_name='test-operation',
execution_time_ms=ANY,
execution_time_ns=ANY,
status=OperationStatus.ERROR,
error_message='test error',
))
def test_evaluate_ok(phase, monkeypatch, snapshot):
monkeypatch.setattr(
'satellite.operations.operations.get_proxy_context',
Mock(return_value=ProxyContext(
mode=ProxyMode.FORWARD,
port=9099,
)),
)
mock_evaluate = Mock(wraps=lambda code, msg: HttpMessage(
url=msg.url,
headers=msg.headers,
data=msg.data.replace('bar', 'bar_processed'),
))
monkeypatch.setattr(
'satellite.operations.operations.evaluate',
mock_evaluate,
)
mock_emit = Mock()
monkeypatch.setattr(
'satellite.operations.pipeline.audit_logs.emit',
mock_emit,
)
flow = load_flow('http_raw')
operation = MockOperation(route_id='route-id', filter_id='filter-id')
operation.evaluate(flow, phase)
snapshot.assert_match(flow.request.get_state(), name='request')
snapshot.assert_match(flow.response.get_state(), name='response')
mock_emit.assert_called_once_with(
OperationLogRecord(
flow_id=flow.id,
proxy_mode=ProxyMode.FORWARD,
route_id='route-id',
filter_id='filter-id',
phase=phase,
operation_name='test-operation',
execution_time_ms=ANY,
execution_time_ns=ANY,
status=OperationStatus.OK,
error_message=None,
))