Example #1
0
def test_format_custom_resource():
    with mock.patch(
            'paasta_tools.setup_kubernetes_cr.get_config_hash',
            autospec=True,
    ) as mock_get_config_hash:
        expected = {
            'apiVersion': 'yelp.com/v1',
            'kind': 'flinkcluster',
            'metadata': {
                'name': 'kurupt--fm-radio--station',
                'labels': {
                    'yelp.com/paasta_service':
                    'kurupt_fm',
                    'yelp.com/paasta_instance':
                    'radio_station',
                    'yelp.com/paasta_config_sha':
                    mock_get_config_hash.return_value,
                },
            },
            'spec': {
                'dummy': 'conf'
            },
        }
        assert setup_kubernetes_cr.format_custom_resource(
            instance_config={'dummy': 'conf'},
            service='kurupt_fm',
            instance='radio_station',
            kind='flinkcluster',
            version='v1',
            group='yelp.com',
        ) == expected
def test_format_custom_resource():
    with mock.patch(
            "paasta_tools.setup_kubernetes_cr.get_config_hash",
            autospec=True) as mock_get_config_hash, mock.patch(
                "paasta_tools.setup_kubernetes_cr.load_system_paasta_config",
                autospec=True) as mock_load_system_paasta_config:
        mock_load_system_paasta_config.return_value = SystemPaastaConfig(
            {"dashboard_links": {}}, "")
        expected = {
            "apiVersion": "yelp.com/v1",
            "kind": "flink",
            "metadata": {
                "name": "kurupt--fm-radio--station",
                "namespace": "paasta-flinks",
                "labels": {
                    "yelp.com/paasta_service":
                    "kurupt_fm",
                    "yelp.com/paasta_instance":
                    "radio_station",
                    "yelp.com/paasta_cluster":
                    "mycluster",
                    "yelp.com/paasta_config_sha":
                    mock_get_config_hash.return_value,
                    "paasta.yelp.com/service":
                    "kurupt_fm",
                    "paasta.yelp.com/instance":
                    "radio_station",
                    "paasta.yelp.com/cluster":
                    "mycluster",
                    "paasta.yelp.com/config_sha":
                    mock_get_config_hash.return_value,
                },
                "annotations": {
                    "yelp.com/desired_state":
                    "running",
                    "yelp.com/dashboard_url":
                    "http://flink.k8s.paasta-mycluster.yelp:31080/kurupt--fm-radio--station",
                    "paasta.yelp.com/desired_state":
                    "running",
                    "paasta.yelp.com/dashboard_url":
                    "http://flink.k8s.paasta-mycluster.yelp:31080/kurupt--fm-radio--station",
                },
            },
            "spec": {
                "dummy": "conf"
            },
        }
        assert (setup_kubernetes_cr.format_custom_resource(
            instance_config={"dummy": "conf"},
            service="kurupt_fm",
            instance="radio_station",
            cluster="mycluster",
            kind="flink",
            version="v1",
            group="yelp.com",
            namespace="paasta-flinks",
        ) == expected)
def test_format_custom_resource():
    with mock.patch("paasta_tools.setup_kubernetes_cr.get_config_hash",
                    autospec=True) as mock_get_config_hash:
        expected = {
            "apiVersion": "yelp.com/v1",
            "kind": "flink",
            "metadata": {
                "name": "kurupt--fm-radio--station",
                "namespace": "paasta-flinks",
                "labels": {
                    "yelp.com/paasta_service":
                    "kurupt_fm",
                    "yelp.com/paasta_instance":
                    "radio_station",
                    "yelp.com/paasta_cluster":
                    "mycluster",
                    "yelp.com/paasta_config_sha":
                    mock_get_config_hash.return_value,
                },
                "annotations": {
                    "yelp.com/desired_state": "running"
                },
            },
            "spec": {
                "dummy": "conf"
            },
        }
        assert (setup_kubernetes_cr.format_custom_resource(
            instance_config={"dummy": "conf"},
            service="kurupt_fm",
            instance="radio_station",
            cluster="mycluster",
            kind="flink",
            version="v1",
            group="yelp.com",
            namespace="paasta-flinks",
        ) == expected)