"""Test module for runner.""" from src import runner import pytest @pytest.mark.parametrize( ('data_label', 'data_id', 'expected_index'), [ ('OrchardSoundRecording', 'xyz', 0), ('Product', 'abc', 1), ('Track', 123, 2), ('OrchardAsset', 'def', 3), ('Project', 456, 4) ] ) def test_process_input(data_label, data_id, expected_index): """Test id+label combo events.""" input_event = { 'id': data_id, 'label': data_label } result = runner._process_input(input_event) for idx, value in enumerate(result): if idx == expected_index: assert result[idx] == [data_id] else: assert result[idx] is None