def test_createStatefulSet(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
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)
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)
