def test_batch_distribute_helper( mock_perf_counter: MagicMock, mock_generate_description: MagicMock, ): # Setup base_permalink = MagicMock() seed_number = 5000 validate = MagicMock() output_dir = MagicMock() expected_permalink = Permalink( seed_number=seed_number, spoiler=True, patcher_configuration=base_permalink.patcher_configuration, layout_configuration=base_permalink.layout_configuration, ) mock_perf_counter.side_effect = [1000, 5000] # Run delta_time = batch_distribute.batch_distribute_helper( base_permalink, seed_number, validate, output_dir) # Assert mock_generate_description.assert_called_once_with(expected_permalink, None, validate) assert delta_time == 4000 output_dir.joinpath.assert_called_once_with("{}.json".format(seed_number)) mock_generate_description.return_value.save_to_file.assert_called_once_with( output_dir.joinpath.return_value)
def test_batch_distribute_helper( mock_perf_counter: MagicMock, mock_generate_description: MagicMock, ): # Setup base_permalink = MagicMock() seed_number = 5000 validate = MagicMock() output_dir = MagicMock() timeout = 67 expected_permalink = Permalink( seed_number=seed_number, spoiler=True, presets=base_permalink.presets, ) mock_perf_counter.side_effect = [1000, 5000] # Run delta_time = batch_distribute.batch_distribute_helper( base_permalink, seed_number, timeout, validate, output_dir) # Assert mock_generate_description.assert_called_once_with( permalink=expected_permalink, status_update=None, validate_after_generation=validate, timeout=timeout) assert delta_time == 4000 output_dir.joinpath.assert_called_once_with("{}.json".format(seed_number)) mock_generate_description.return_value.save_to_file.assert_called_once_with( output_dir.joinpath.return_value)
def test_batch_distribute_helper(mocker): # Setup description = MagicMock() mock_generate_description: AsyncMock = mocker.patch( "randovania.generator.generator.generate_and_validate_description", new_callable=AsyncMock, return_value=description) mock_perf_counter = mocker.patch("time.perf_counter", autospec=False) # TODO: pytest-qt bug base_permalink = MagicMock(spec=GeneratorParameters) base_permalink.presets = [MagicMock()] seed_number = 5000 validate = MagicMock() output_dir = MagicMock() timeout = 67 description.file_extension.return_value = "rdvgame" expected_permalink = GeneratorParameters( seed_number=seed_number, spoiler=True, presets=base_permalink.presets, ) mock_perf_counter.side_effect = [1000, 5000] # Run delta_time = batch_distribute.batch_distribute_helper( base_permalink, seed_number, timeout, validate, output_dir) # Assert mock_generate_description.assert_awaited_once_with( generator_params=expected_permalink, status_update=None, validate_after_generation=validate, timeout=timeout, attempts=0) assert delta_time == 4000 output_dir.joinpath.assert_called_once_with( "{}.rdvgame".format(seed_number)) description.save_to_file.assert_called_once_with( output_dir.joinpath.return_value)