def test_deleteStatefulSet(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
result = service.deleteStatefulSet(self.name, self.namespace)
expected_calls = [
call.AppsV1beta1Api().delete_namespaced_stateful_set(
self.name, self.namespace, V1DeleteOptions())
]
self.assertEqual(expected_calls, client_mock.mock_calls)
self.assertEqual(
client_mock.AppsV1beta1Api().delete_namespaced_stateful_set.
return_value, result)
def test_updateStatefulSet(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
result = service.updateStatefulSet(self.cluster_object)
expected_calls = [
call.AppsV1beta1Api().patch_namespaced_stateful_set(
self.name, self.namespace, self.stateful_set)
]
self.assertEqual(expected_calls, client_mock.mock_calls)
self.assertEqual(
client_mock.AppsV1beta1Api().patch_namespaced_stateful_set.
return_value, result)
def test_listAllStatefulSetsWithLabels_custom(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
labels = {"operated-by": "me", "heritage": "mongo", "name": "name"}
result = service.listAllStatefulSetsWithLabels(labels)
expected_calls = [
call.AppsV1beta1Api().list_stateful_set_for_all_namespaces(
label_selector="operated-by=me,heritage=mongo,name=name")
]
self.assertEqual(expected_calls, client_mock.mock_calls)
self.assertEqual(
client_mock.AppsV1beta1Api().list_stateful_set_for_all_namespaces.
return_value, result)
def test_listAllStatefulSetsWithLabels_default(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
result = service.listAllStatefulSetsWithLabels()
expected_calls = [
call.AppsV1beta1Api().list_stateful_set_for_all_namespaces(
label_selector=
"operated-by=operators.ultimaker.com,heritage=mongos")
]
self.assertEqual(expected_calls, client_mock.mock_calls)
self.assertEqual(
client_mock.AppsV1beta1Api().list_stateful_set_for_all_namespaces.
return_value, result)
def test___init__(self, client_mock):
KubernetesService()
config = Configuration()
config.debug = False
expected = [
call.ApiClient(config),
call.CoreV1Api(client_mock.ApiClient.return_value),
call.CustomObjectsApi(client_mock.ApiClient.return_value),
call.ApiextensionsV1beta1Api(client_mock.ApiClient.return_value),
call.AppsV1beta1Api(client_mock.ApiClient.return_value),
]
with patch("kubernetes.client.configuration.Configuration.__eq__",
dict_eq):
self.assertEqual(expected, client_mock.mock_calls)
def test_createStatefulSet_no_optional_fields(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
del self.cluster_dict["spec"]["mongodb"]["cpu_limit"]
del self.cluster_dict["spec"]["mongodb"]["memory_limit"]
self.cluster_object = V1MongoClusterConfiguration(**self.cluster_dict)
expected_calls = [
call.AppsV1beta1Api().create_namespaced_stateful_set(
self.namespace, self.stateful_set)
]
result = service.createStatefulSet(self.cluster_object)
self.assertEqual(expected_calls, client_mock.mock_calls)
self.assertEqual(
client_mock.AppsV1beta1Api().create_namespaced_stateful_set.
return_value, result)
