diff --git a/tests/laborious/activities/test_opc.py b/tests/laborious/activities/test_opc.py index 669f82b..76afed5 100644 --- a/tests/laborious/activities/test_opc.py +++ b/tests/laborious/activities/test_opc.py @@ -276,13 +276,11 @@ async def test_write_opc_data_success(mock_dataframe, opc): opc.manage_output_tags = AsyncMock(return_value=(True, {'tag1': 0.1, 'tag2': 0.2})) opc.process_confidence = MagicMock(return_value={'data': 'data'}) - output = await opc.write_opc_data(input_data) + output_data, opc_metrics = await opc.write_opc_data(input_data) # Assert - assert output == { - 'data': {'data': 'data'}, - 'metrics': {'server1': {'tag1': 0.1, 'tag2': 0.2}}, - } + assert output_data == {'data': 'data'} + assert opc_metrics == {'server1': {'tag1': 0.1, 'tag2': 0.2}} opc.manage_output_tags.assert_called_once_with( 'server1', input_data['opc_output_config']['server1'],