def test_createMongoObjectDefinition(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
expected_def = V1beta1CustomResourceDefinition(
api_version="apiextensions.k8s.io/v1beta1",
kind="CustomResourceDefinition",
metadata=V1ObjectMeta(name="mongos.operators.ultimaker.com"),
spec=V1beta1CustomResourceDefinitionSpec(
group="operators.ultimaker.com",
version="v1",
scope="Namespaced",
names=V1beta1CustomResourceDefinitionNames(plural="mongos",
singular="mongo",
kind="Mongo",
short_names=["mng"
])))
client_mock.ApiextensionsV1beta1Api.return_value.list_custom_resource_definition.return_value = \
V1beta1CustomResourceDefinitionList(items=[])
client_mock.ApiextensionsV1beta1Api.return_value.create_custom_resource_definition.return_value = expected_def
result = service.createMongoObjectDefinition()
self.assertEquals(expected_def, result)
expected_calls = [
call.ApiextensionsV1beta1Api().list_custom_resource_definition(),
call.ApiextensionsV1beta1Api().create_custom_resource_definition(
expected_def),
]
self.assertEqual(expected_calls, client_mock.mock_calls)
def test_createMongoObjectDefinition_existing(self, client_mock):
service = KubernetesService()
client_mock.reset_mock()
item = MagicMock()
item.spec.names.plural = "mongos"
client_mock.ApiextensionsV1beta1Api.return_value.list_custom_resource_definition.return_value.items = [
item
]
result = service.createMongoObjectDefinition()
self.assertEquals(item, result)
expected_calls = [
call.ApiextensionsV1beta1Api().list_custom_resource_definition()
]
self.assertEqual(expected_calls, client_mock.mock_calls)