qontract-reconcile 0.10.1rc1201__py3-none-any.whl → 0.10.2.dev1__py3-none-any.whl

reconcile/test/test_ocm_update_recommended_version.py

@@ -1,145 +0,0 @@
-import pytest
-
-from reconcile.ocm.types import (
-    OCMClusterNetwork,
-    OCMClusterSpec,
-    OCMSpec,
-)
-from reconcile.ocm_update_recommended_version import (
-    format_initial_version,
-    get_highest,
-    get_majority,
-    get_updated_recommended_versions,
-    get_version_weights,
-    recommended_version,
-)
-
-
-@pytest.fixture
-def versions() -> list[str]:
-    return ["1.0.0", "1.1.0", "1.1.1", "1.1.0"]
-
-
-@pytest.fixture
-def version_set(versions) -> set[str]:
-    return set(versions)
-
-
-def test_get_highest(version_set):
-    highest = get_highest(version_set)
-    assert highest == "1.1.1"
-
-    assert get_highest({"1.1.1"}) == "1.1.1"
-
-
-def test_get_majority(versions, version_set):
-    majority = get_majority(version_set, versions)
-    assert majority == "1.1.0"
-
-    assert get_majority({"1.1.1"}, ["1.1.1"]) == "1.1.1"
-
-
-def test_recommended_version(versions, version_set):
-    assert recommended_version(versions, high_weight=10, majority_weight=1) == "1.1.1"
-    assert recommended_version(versions, high_weight=0, majority_weight=1) == "1.1.0"
-    assert (
-        recommended_version(
-            versions=["1.1.0", "1.1.0", "1.1.1"], high_weight=10, majority_weight=5
-        )
-        == "1.1.1"
-    )
-
-
-def test_get_version_weights():
-    assert get_version_weights({}) == (1, 1)
-    assert get_version_weights({"recommendedVersionWeight": {"majority": 0}}) == (1, 0)
-    assert get_version_weights({
-        "recommendedVersionWeight": {"majority": 2, "highest": 4}
-    }) == (4, 2)
-
-
-def add_cluster(
-    clusters: dict[str, OCMSpec], cluster_name: str, version: str, channel: str
-):
-    clusters[cluster_name] = OCMSpec(
-        network=OCMClusterNetwork(vpc="", service="", pod=""),
-        spec=OCMClusterSpec(
-            version=version,
-            channel=channel,
-            instance_type="",
-            multi_az=True,
-            private=True,
-            product="",
-            provider="",
-            region="",
-        ),
-    )
-
-
-def test_get_updated_recommended_versions():
-    ocm_info = {
-        "recommendedVersions": [],
-        "upgradePolicyAllowedWorkloads": ["foo", "bar"],
-        "upgradePolicyClusters": [
-            {"name": "a", "upgradePolicy": {"workloads": ["foo"]}},
-        ],
-    }
-    clusters: dict[str, OCMSpec] = {}
-    add_cluster(clusters, "a", "2.0.0", "stable")
-
-    assert get_updated_recommended_versions(ocm_info, clusters) == [
-        {
-            "channel": "stable",
-            "recommendedVersion": "2.0.0",
-            "workload": "foo",
-            "initialVersion": "openshift-v2.0.0",
-        },
-    ]
-
-
-def test_get_updated_recommended_versions_multiple_channel():
-    ocm_info = {
-        "recommendedVersions": [
-            {"recommendedVersion": "1.0.0", "workload": "foo"},
-        ],
-        "upgradePolicyAllowedWorkloads": ["foo", "bar"],
-        "upgradePolicyClusters": [
-            {"name": "a", "upgradePolicy": {"workloads": ["foo"]}},
-            {"name": "c", "upgradePolicy": {"workloads": ["foo", "bar"]}},
-            {"name": "b", "upgradePolicy": {"workloads": ["bar"]}},
-            {"name": "b2", "upgradePolicy": {"workloads": ["bar"]}},
-            {"name": "b3", "upgradePolicy": {"workloads": ["bar"]}},
-        ],
-    }
-    clusters: dict[str, OCMSpec] = {}
-    add_cluster(clusters, "a", "2.1.0", "stable")
-    add_cluster(clusters, "c", "2.0.0", "stable")
-    add_cluster(clusters, "b", "3.0.0", "fast")
-    add_cluster(clusters, "b2", "3.0.0", "fast")
-    add_cluster(clusters, "b3", "2.0.0", "fast")
-
-    assert get_updated_recommended_versions(ocm_info, clusters) == [
-        {
-            "channel": "stable",
-            "recommendedVersion": "2.1.0",
-            "workload": "foo",
-            "initialVersion": "openshift-v2.1.0",
-        },
-        {
-            "channel": "stable",
-            "recommendedVersion": "2.0.0",
-            "workload": "bar",
-            "initialVersion": "openshift-v2.0.0",
-        },
-        {
-            "channel": "fast",
-            "recommendedVersion": "3.0.0",
-            "workload": "bar",
-            "initialVersion": "openshift-v3.0.0-fast",
-        },
-    ]
-
-
-def test_format_initial_version():
-    assert format_initial_version("2.0.0", "stable") == "openshift-v2.0.0"
-    assert format_initial_version("2.0.0", "fast") == "openshift-v2.0.0-fast"

reconcile/test/test_ocm_upgrade_scheduler_org_updater.py

@@ -1,125 +0,0 @@
-import json
-
-import pytest
-
-from reconcile.ocm.types import OCMClusterNetwork, OCMSpec, OSDClusterSpec
-from reconcile.ocm_upgrade_scheduler_org_updater import render_policy
-from reconcile.utils.jinja2.utils import Jinja2TemplateError
-
-
-@pytest.fixture
-def cluster_ocm_spec():
-    n = OCMClusterNetwork(
-        vpc="10.112.0.0/16",
-        service="10.120.0.0/16",
-        pod="10.128.0.0/14",
-    )
-    spec = OSDClusterSpec(
-        product="osd",
-        channel="stable",
-        instance_type="m5.xlarge",
-        multi_az=False,
-        private=False,
-        region="us-east-1",
-        version="4.10.0",
-        load_balancers=5,
-        storage=1100,
-        provider="aws",
-    )
-    obj = OCMSpec(
-        spec=spec,
-        network=n,
-    )
-    yield obj
-
-
-@pytest.fixture
-def labels():
-    return {"key1": "value1", "key2": "value2"}
-
-
-@pytest.fixture
-def variables():
-    return json.dumps({"var1": "value1", "var2": "value2"})
-
-
-def create_template_info(content, type="jinja2", variables=None):
-    return {
-        "path": {
-            "content": content,
-        },
-        "type": type,
-        "variables": variables,
-    }
-
-
-def test_render_policy_basic(cluster_ocm_spec, labels):
-    template_info = create_template_info("hello: world")
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": "world"}
-
-
-def test_render_policy_cluster_spec(cluster_ocm_spec, labels):
-    template_info = create_template_info("hello: {{ cluster.spec.region }}")
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": cluster_ocm_spec.spec.region}
-
-
-def test_render_policy_cluster_spec_bad_attribute(cluster_ocm_spec, labels):
-    template_info = create_template_info("hello: {{ cluster.unkown_attribute }}")
-    with pytest.raises(Jinja2TemplateError):
-        render_policy(template_info, cluster_ocm_spec, labels, settings={})
-
-
-def test_render_policy_cluster_labels(cluster_ocm_spec, labels):
-    template_info = create_template_info("hello: {{ labels.key1 }}")
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": labels["key1"]}
-
-    template_info = create_template_info("hello: {{ labels['key1'] }}")
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": labels["key1"]}
-
-
-def test_render_policy_cluster_labels_missing(cluster_ocm_spec, labels):
-    template_info = create_template_info("hello: {{ labels.unknown_key }}")
-    with pytest.raises(Jinja2TemplateError):
-        render_policy(template_info, cluster_ocm_spec, labels, settings={})
-
-    template_info = create_template_info("hello: {{ labels['unknown_key'] }}")
-    with pytest.raises(Jinja2TemplateError):
-        render_policy(template_info, cluster_ocm_spec, labels, settings={})
-
-
-def test_render_policy_cluster_labels_default(cluster_ocm_spec, labels):
-    template_info = create_template_info(
-        "hello: {{ labels.unknown_key | default('OK') }}"
-    )
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": "OK"}
-
-    template_info = create_template_info(
-        "hello: {{ labels['unknown_key'] | default('OK') }}"
-    )
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": "OK"}
-
-
-def test_render_policy_variables(cluster_ocm_spec, labels, variables):
-    template_info = create_template_info("hello: {{ var1 }}", variables=variables)
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": "value1"}
-
-
-def test_render_policy_variables_missing(cluster_ocm_spec, labels, variables):
-    template_info = create_template_info("hello: {{ unkown_var }}", variables=variables)
-    with pytest.raises(Jinja2TemplateError):
-        render_policy(template_info, cluster_ocm_spec, labels, settings={})
-
-
-def test_render_policy_variables_default(cluster_ocm_spec, labels, variables):
-    template_info = create_template_info(
-        "hello: {{ unkown_var | default('OK') }}", variables=variables
-    )
-    rendered = render_policy(template_info, cluster_ocm_spec, labels, settings={})
-    assert rendered == {"hello": "OK"}