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)