def test_setup_custom_resources():
with mock.patch(
"paasta_tools.setup_kubernetes_cr.list_custom_resources", autospec=True
) as mock_list_cr, mock.patch(
"paasta_tools.setup_kubernetes_cr.reconcile_kubernetes_resource", autospec=True
) as mock_reconcile_kubernetes_resource:
mock_client = mock.Mock()
mock_kind = mock.Mock()
assert setup_kubernetes_cr.setup_custom_resources(
kube_client=mock_client,
kind=mock_kind,
version="v1",
config_dicts={},
group="yelp.com",
cluster="mycluster",
)
mock_reconcile_kubernetes_resource.side_effect = [True, False]
assert not setup_kubernetes_cr.setup_custom_resources(
kube_client=mock_client,
kind=mock_kind,
version="v1",
config_dicts={"kurupt": "something", "mc": "another"},
group="yelp.com",
cluster="mycluster",
)
mock_reconcile_kubernetes_resource.side_effect = [True, True]
assert setup_kubernetes_cr.setup_custom_resources(
kube_client=mock_client,
kind=mock_kind,
version="v1",
config_dicts={"kurupt": "something", "mc": "another"},
group="yelp.com",
cluster="mycluster",
)
mock_reconcile_kubernetes_resource.assert_has_calls(
[
mock.call(
kube_client=mock_client,
service="kurupt",
instance_configs="something",
cluster="mycluster",
instance=None,
kind=mock_kind,
custom_resources=mock_list_cr.return_value,
version="v1",
group="yelp.com",
),
mock.call(
kube_client=mock_client,
service="mc",
instance_configs="another",
cluster="mycluster",
instance=None,
kind=mock_kind,
custom_resources=mock_list_cr.return_value,
version="v1",
group="yelp.com",
),
]
)
def test_setup_custom_resources():
with mock.patch(
'paasta_tools.setup_kubernetes_cr.list_custom_resources',
autospec=True,
) as mock_list_cr, mock.patch(
'paasta_tools.setup_kubernetes_cr.reconcile_kubernetes_resource',
autospec=True,
) as mock_reconcile_kubernetes_resource:
mock_client = mock.Mock()
mock_kind = mock.Mock()
assert setup_kubernetes_cr.setup_custom_resources(
kube_client=mock_client,
kind=mock_kind,
version='v1',
config_dicts={},
group='yelp.com',
)
mock_reconcile_kubernetes_resource.side_effect = [True, False]
assert not setup_kubernetes_cr.setup_custom_resources(
kube_client=mock_client,
kind=mock_kind,
version='v1',
config_dicts={
'kurupt': 'something',
'mc': 'another'
},
group='yelp.com',
)
mock_reconcile_kubernetes_resource.side_effect = [True, True]
assert setup_kubernetes_cr.setup_custom_resources(
kube_client=mock_client,
kind=mock_kind,
version='v1',
config_dicts={
'kurupt': 'something',
'mc': 'another'
},
group='yelp.com',
)
mock_reconcile_kubernetes_resource.assert_has_calls([
mock.call(
kube_client=mock_client,
service='kurupt',
instance_configs='something',
kind=mock_kind,
custom_resources=mock_list_cr.return_value,
version='v1',
group='yelp.com',
),
mock.call(
kube_client=mock_client,
service='mc',
instance_configs='another',
kind=mock_kind,
custom_resources=mock_list_cr.return_value,
version='v1',
group='yelp.com',
),
])