Skip to content

Commit

Permalink
Merge pull request #10 from oamg/feature/tests-check-stdout
Browse files Browse the repository at this point in the history 
Add stoud check for main tests
  • Loading branch information
@andywaltlova
andywaltlova authored Feb 6, 2024
2 parents 3f12656 + d95bcd5 commit 4611548
Showing 1 changed file with 157 additions and 16 deletions.
173 changes: 157 additions & 16 deletions tests/test_main.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,18 @@
)


@patch("scripts.leapp_script.SCRIPT_TYPE", "TEST")
def test_main_invalid_script_type(capsys):
main()
captured = capsys.readouterr()

assert (
"Allowed values for RHC_WORKER_LEAPP_SCRIPT_TYPE are 'PREUPGRADE' and 'UPGRADE'."
in captured.out
)
assert "Exiting because RHC_WORKER_LEAPP_SCRIPT_TYPE='TEST'" in captured.out


@patch("scripts.leapp_script.SCRIPT_TYPE", "PREUPGRADE")
@patch("scripts.leapp_script.IS_PREUPGRADE", True)
@patch("scripts.leapp_script.get_rhel_version")
Expand All @@ -19,13 +31,17 @@ def test_main_non_eligible_release_preupgrade(
mock_setup_leapp,
mock_is_non_eligible_releases,
mock_get_rhel_version,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "6.9")
mock_is_non_eligible_releases.return_value = True
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])

main()

captured = capsys.readouterr()
assert 'Exiting because distribution="rhel" and version="6.9"' in captured.out

mock_get_rhel_version.assert_called_once()
mock_is_non_eligible_releases.assert_called_once()
mock_output_collector.assert_called_once()
Expand Down Expand Up @@ -56,6 +72,7 @@ def test_main_eligible_release_preupgrade(
mock_is_non_eligible_releases,
mock_get_rhel_version,
mock_parse_results,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "7.9")
mock_is_non_eligible_releases.return_value = False
Expand All @@ -64,6 +81,8 @@ def test_main_eligible_release_preupgrade(
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])

main()
captured = capsys.readouterr()
assert "Operation Preupgrade finished successfully." in captured.out

mock_setup_leapp.assert_called_once()
mock_should_use_no_rhsm_check.assert_called_once()
Expand All @@ -76,29 +95,52 @@ def test_main_eligible_release_preupgrade(

@patch("scripts.leapp_script.SCRIPT_TYPE", "UPGRADE")
@patch("scripts.leapp_script.IS_UPGRADE", True)
@patch("scripts.leapp_script.parse_results")
@patch("scripts.leapp_script.reboot_system")
@patch("scripts.leapp_script.get_rhel_version")
@patch("scripts.leapp_script.is_non_eligible_releases")
@patch("scripts.leapp_script.setup_leapp")
@patch("scripts.leapp_script.should_use_no_rhsm_check")
@patch("scripts.leapp_script.install_leapp_pkg_corresponding_to_installed_rhui")
@patch("scripts.leapp_script.remove_previous_reports")
@patch("scripts.leapp_script.execute_operation")
@patch("scripts.leapp_script.update_insights_inventory")
@patch("scripts.leapp_script.OutputCollector")
def test_main_non_eligible_release_upgrade(
def test_main_eligible_release_upgrade(
mock_output_collector,
mock_update_insights_inventory,
mock_execute_operation,
mock_remove_previous_reports,
mock_should_use_no_rhsm_check,
mock_install_rhui,
mock_setup_leapp,
mock_is_non_eligible_releases,
mock_get_rhel_version,
mock_reboot_system,
mock_parse_results,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "6.9")
mock_is_non_eligible_releases.return_value = True
mock_get_rhel_version.return_value = ("rhel", "7.9")
mock_is_non_eligible_releases.return_value = False
mock_setup_leapp.return_value = [{"leapp_pkg": "to_install"}]
mock_should_use_no_rhsm_check.return_value = True
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])
mock_execute_operation.return_value = (
"LOREM IPSUM\nTEST" + REBOOT_GUIDANCE_MESSAGE + "TEST\nDOLOR SIT AMET"
)

main()
captured = capsys.readouterr()
assert "Operation Upgrade finished successfully." in captured.out

mock_get_rhel_version.assert_called_once()
mock_is_non_eligible_releases.assert_called_once()
mock_output_collector.assert_called_once()
mock_setup_leapp.assert_not_called()
mock_update_insights_inventory.assert_not_called()
mock_setup_leapp.assert_called_once()
mock_should_use_no_rhsm_check.assert_called_once()
mock_install_rhui.assert_called_once()
mock_remove_previous_reports.assert_called_once()
mock_execute_operation.assert_called_once()
mock_parse_results.assert_called_once()
mock_update_insights_inventory.assert_called_once()
mock_reboot_system.assert_called_once()


@patch("scripts.leapp_script.SCRIPT_TYPE", "UPGRADE")
Expand All @@ -114,7 +156,7 @@ def test_main_non_eligible_release_upgrade(
@patch("scripts.leapp_script.execute_operation")
@patch("scripts.leapp_script.update_insights_inventory")
@patch("scripts.leapp_script.OutputCollector")
def test_main_eligible_release_upgrade(
def test_main_upgrade_not_sucessfull(
mock_output_collector,
mock_update_insights_inventory,
mock_execute_operation,
Expand All @@ -126,17 +168,18 @@ def test_main_eligible_release_upgrade(
mock_get_rhel_version,
mock_reboot_system,
mock_parse_results,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "7.9")
mock_is_non_eligible_releases.return_value = False
mock_setup_leapp.return_value = [{"leapp_pkg": "to_install"}]
mock_should_use_no_rhsm_check.return_value = True
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])
mock_execute_operation.return_value = (
"LOREM IPSUM\nTEST" + REBOOT_GUIDANCE_MESSAGE + "TEST\nDOLOR SIT AMET"
)
mock_execute_operation.return_value = "LOREM IPSUM\n" + "\nDOLOR SIT AMET"

main()
captured = capsys.readouterr()
assert "Operation Upgrade finished successfully." in captured.out

mock_setup_leapp.assert_called_once()
mock_should_use_no_rhsm_check.assert_called_once()
Expand All @@ -145,7 +188,7 @@ def test_main_eligible_release_upgrade(
mock_execute_operation.assert_called_once()
mock_parse_results.assert_called_once()
mock_update_insights_inventory.assert_called_once()
mock_reboot_system.assert_called_once()
mock_reboot_system.assert_not_called()


@patch("scripts.leapp_script.SCRIPT_TYPE", "UPGRADE")
Expand All @@ -154,40 +197,138 @@ def test_main_eligible_release_upgrade(
@patch("scripts.leapp_script.reboot_system")
@patch("scripts.leapp_script.get_rhel_version")
@patch("scripts.leapp_script.is_non_eligible_releases")
@patch("scripts.leapp_script.setup_leapp")
@patch("scripts.leapp_script.should_use_no_rhsm_check")
@patch("scripts.leapp_script.install_leapp_pkg_corresponding_to_installed_rhui")
@patch("scripts.leapp_script.remove_previous_reports")
@patch("scripts.leapp_script.execute_operation")
@patch("scripts.leapp_script.update_insights_inventory")
@patch("scripts.leapp_script.OutputCollector")
def test_main_upgrade_not_sucessfull(
@patch("scripts.leapp_script.run_subprocess")
def test_main_setup_leapp_not_sucessfull(
mock_run_subprocess,
mock_output_collector,
mock_update_insights_inventory,
mock_execute_operation,
mock_remove_previous_reports,
mock_should_use_no_rhsm_check,
mock_install_rhui,
mock_is_non_eligible_releases,
mock_get_rhel_version,
mock_reboot_system,
mock_parse_results,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "7.9")
mock_is_non_eligible_releases.return_value = False
mock_run_subprocess.return_value = ("Installation failed", 1)
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])

main()
captured = capsys.readouterr()
assert "Installation of leapp failed with code '1'" in captured.out

mock_should_use_no_rhsm_check.assert_not_called()
mock_install_rhui.assert_not_called()
mock_remove_previous_reports.assert_not_called()
mock_execute_operation.assert_not_called()
mock_parse_results.assert_not_called()
mock_update_insights_inventory.assert_not_called()
mock_reboot_system.assert_not_called()


@patch("scripts.leapp_script.SCRIPT_TYPE", "UPGRADE")
@patch("scripts.leapp_script.IS_UPGRADE", True)
@patch("scripts.leapp_script.parse_results")
@patch("scripts.leapp_script.reboot_system")
@patch("scripts.leapp_script.get_rhel_version")
@patch("scripts.leapp_script.is_non_eligible_releases")
@patch("scripts.leapp_script.should_use_no_rhsm_check")
@patch("scripts.leapp_script.setup_leapp")
@patch("scripts.leapp_script.remove_previous_reports")
@patch("scripts.leapp_script.execute_operation")
@patch("scripts.leapp_script.update_insights_inventory")
@patch("scripts.leapp_script.OutputCollector")
@patch("scripts.leapp_script.run_subprocess")
def test_main_install_corresponding_pkgs_not_sucessfull(
mock_run_subprocess,
mock_output_collector,
mock_update_insights_inventory,
mock_execute_operation,
mock_remove_previous_reports,
mock_setup_leapp,
mock_should_use_no_rhsm_check,
mock_is_non_eligible_releases,
mock_get_rhel_version,
mock_reboot_system,
mock_parse_results,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "7.9")
mock_is_non_eligible_releases.return_value = False
mock_run_subprocess.return_value = ("Installation failed", 1)
mock_setup_leapp.return_value = [{"leapp_pkg": "to_install"}]
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])

main()
captured = capsys.readouterr()
assert (
"Installation of to_install (coresponding pkg to '{'leapp_pkg': 'to_install'}') failed with exit code 1 and output: Installation failed."
in captured.out
)

mock_setup_leapp.assert_called_once()
mock_should_use_no_rhsm_check.assert_called_once()
mock_remove_previous_reports.assert_not_called()
mock_execute_operation.assert_not_called()
mock_parse_results.assert_not_called()
mock_update_insights_inventory.assert_not_called()
mock_reboot_system.assert_not_called()


@patch("scripts.leapp_script.SCRIPT_TYPE", "UPGRADE")
@patch("scripts.leapp_script.IS_UPGRADE", True)
@patch("scripts.leapp_script.parse_results")
@patch("scripts.leapp_script.reboot_system")
@patch("scripts.leapp_script.get_rhel_version")
@patch("scripts.leapp_script.is_non_eligible_releases")
@patch("scripts.leapp_script.setup_leapp")
@patch("scripts.leapp_script.should_use_no_rhsm_check")
@patch("scripts.leapp_script.install_leapp_pkg_corresponding_to_installed_rhui")
@patch("scripts.leapp_script.remove_previous_reports")
@patch("scripts.leapp_script.execute_operation")
@patch("scripts.leapp_script.OutputCollector")
@patch("scripts.leapp_script.run_subprocess")
def test_main_update_inventory_not_sucessfull(
mock_run_subprocess,
mock_output_collector,
mock_execute_operation,
mock_remove_previous_reports,
mock_should_use_no_rhsm_check,
mock_install_rhui,
mock_setup_leapp,
mock_is_non_eligible_releases,
mock_get_rhel_version,
mock_reboot_system,
mock_parse_results,
capsys,
):
mock_get_rhel_version.return_value = ("rhel", "7.9")
mock_is_non_eligible_releases.return_value = False
mock_setup_leapp.return_value = [{"leapp_pkg": "to_install"}]
mock_should_use_no_rhsm_check.return_value = True
mock_output_collector.return_value = OutputCollector(entries=["non-empty"])
mock_execute_operation.return_value = "LOREM IPSUM\n" + "\nDOLOR SIT AMET"
mock_run_subprocess.return_value = ("Installation failed", 1)

main()
captured = capsys.readouterr()
assert "Updating system status in Red Hat Insights." in captured.out
assert "System registration failed with exit code 1." in captured.out

mock_setup_leapp.assert_called_once()
mock_should_use_no_rhsm_check.assert_called_once()
mock_install_rhui.assert_called_once()
mock_remove_previous_reports.assert_called_once()
mock_execute_operation.assert_called_once()
mock_parse_results.assert_called_once()
mock_update_insights_inventory.assert_called_once()
mock_reboot_system.assert_not_called()

0 comments on commit 4611548

Please sign in to comment.