Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add stoud check for main tests #10

Merged
merged 2 commits into from
Feb 6, 2024
Merged

Add stoud check for main tests #10

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
11be946
Add stoud check for main tests
Jan 24, 2024
d95bcd5
Add more tests cases for main function
Feb 2, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
173 changes: 157 additions & 16 deletions tests/test_main.py
View file
Open in desktop
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()
Loading