diff options
| -rw-r--r-- | tests/integration_tests/conftest.py | 3 | ||||
| -rw-r--r-- | tests/integration_tests/test_upgrade.py | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/tests/integration_tests/conftest.py b/tests/integration_tests/conftest.py index 50b300c4..2d12045c 100644 --- a/tests/integration_tests/conftest.py +++ b/tests/integration_tests/conftest.py @@ -108,6 +108,7 @@ def session_cloud(): def get_validated_source( + session_cloud: IntegrationCloud, source=integration_settings.CLOUD_INIT_SOURCE ) -> CloudInitSource: if source == 'NONE': @@ -134,7 +135,7 @@ def setup_image(session_cloud: IntegrationCloud): So we can launch instances / run tests with the correct image """ - source = get_validated_source() + source = get_validated_source(session_cloud) if not source.installs_new_version(): return log.info('Setting up environment for %s', session_cloud.datasource) diff --git a/tests/integration_tests/test_upgrade.py b/tests/integration_tests/test_upgrade.py index 76ad055b..a115bea7 100644 --- a/tests/integration_tests/test_upgrade.py +++ b/tests/integration_tests/test_upgrade.py @@ -56,7 +56,7 @@ def _restart(instance): @pytest.mark.sru_2020_11 def test_upgrade(session_cloud: IntegrationCloud): - source = get_validated_source() + source = get_validated_source(session_cloud) if not source.installs_new_version(): pytest.skip("Install method '{}' not supported for this test".format( source |