@@ -409,18 +409,24 @@ def test_generate_api_success(mocker, caplog):
409409 SOURCE = "source"
410410 OUTPUT = "output"
411411
412- mock_run_post_processor = mocker .patch ("cli._read_bazel_build_py_rule" )
413- mock_run_post_processor = mocker .patch ("cli._get_api_generator_options" )
414- mock_copy_files_needed_for_post_processing = mocker .patch (
412+ mock_read_bazel_build_py_rule = mocker .patch ("cli._read_bazel_build_py_rule" )
413+ mock_get_api_generator_options = mocker .patch ("cli._get_api_generator_options" )
414+ mock_determine_generator_command = mocker .patch (
415415 "cli._determine_generator_command"
416416 )
417- mock_copy_files_needed_for_post_processing = mocker .patch (
417+ mock_run_generator_command = mocker .patch (
418418 "cli._run_generator_command"
419419 )
420- mock_clean_up_files_after_post_processing = mocker .patch ("shutil.copytree" )
420+ mock_shutil_copytree = mocker .patch ("shutil.copytree" )
421421
422422 _generate_api (API_PATH , LIBRARY_ID , SOURCE , OUTPUT )
423423
424+ mock_read_bazel_build_py_rule .assert_called_once ()
425+ mock_get_api_generator_options .assert_called_once ()
426+ mock_determine_generator_command .assert_called_once ()
427+ mock_run_generator_command .assert_called_once ()
428+ mock_shutil_copytree .assert_called_once ()
429+
424430
425431def test_handle_generate_success (
426432 caplog , mock_generate_request_file , mock_build_bazel_file , mocker
0 commit comments