Mini Shell

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.


import munch

from openstack.tests.unit import base

coe_cluster_obj = munch.Munch(
    status="CREATE_IN_PROGRESS",
    cluster_template_id="0562d357-8641-4759-8fed-8173f02c9633",
    uuid="731387cf-a92b-4c36-981e-3271d63e5597",
    links=[{}, {}],
    stack_id="31c1ee6c-081e-4f39-9f0f-f1d87a7defa1",
    keypair="my_keypair",
    master_count=3,
    create_timeout=60,
    node_count=10,
    name="k8s",
    created_at="2016-08-29T06:51:31+00:00",
    api_address="https://172.24.4.6:6443",
    discovery_url="https://discovery.etcd.io/cbeb580da58915809d59ee69348a84f3",
    updated_at="2016-08-29T06:53:24+00:00",
    coe_version="v1.2.0",
    master_addresses=["172.24.4.6"],
    node_addresses=["172.24.4.13"],
    status_reason="Stack CREATE completed successfully",
)


class TestCOEClusters(base.TestCase):

    def get_mock_url(
            self,
            service_type='container-infrastructure-management',
            base_url_append=None, append=None, resource=None):
        return super(TestCOEClusters, self).get_mock_url(
            service_type=service_type, resource=resource,
            append=append, base_url_append=base_url_append)

    def test_list_coe_clusters(self):
        self.register_uris([dict(
            method='GET',
            uri=self.get_mock_url(resource='clusters'),
            json=dict(clusters=[coe_cluster_obj.toDict()]))])
        cluster_list = self.cloud.list_coe_clusters()
        self.assertEqual(
            cluster_list[0],
            self.cloud._normalize_coe_cluster(coe_cluster_obj))
        self.assert_calls()

    def test_create_coe_cluster(self):
        json_response = dict(uuid=coe_cluster_obj.get('uuid'))
        kwargs = dict(name=coe_cluster_obj.name,
                      cluster_template_id=coe_cluster_obj.cluster_template_id,
                      master_count=coe_cluster_obj.master_count,
                      node_count=coe_cluster_obj.node_count)
        self.register_uris([dict(
            method='POST',
            uri=self.get_mock_url(resource='clusters'),
            json=json_response,
            validate=dict(json=kwargs)),
        ])
        expected = self.cloud._normalize_coe_cluster(json_response)
        response = self.cloud.create_coe_cluster(**kwargs)
        self.assertEqual(response, expected)
        self.assert_calls()

    def test_search_coe_cluster_by_name(self):
        self.register_uris([dict(
            method='GET',
            uri=self.get_mock_url(resource='clusters'),
            json=dict(clusters=[coe_cluster_obj.toDict()]))])

        coe_clusters = self.cloud.search_coe_clusters(
            name_or_id='k8s')

        self.assertEqual(1, len(coe_clusters))
        self.assertEqual(coe_cluster_obj.uuid, coe_clusters[0]['id'])
        self.assert_calls()

    def test_search_coe_cluster_not_found(self):
        self.register_uris([dict(
            method='GET',
            uri=self.get_mock_url(resource='clusters'),
            json=dict(clusters=[coe_cluster_obj.toDict()]))])

        coe_clusters = self.cloud.search_coe_clusters(
            name_or_id='non-existent')

        self.assertEqual(0, len(coe_clusters))
        self.assert_calls()

    def test_get_coe_cluster(self):
        self.register_uris([dict(
            method='GET',
            uri=self.get_mock_url(resource='clusters'),
            json=dict(clusters=[coe_cluster_obj.toDict()]))])

        r = self.cloud.get_coe_cluster(coe_cluster_obj.name)
        self.assertIsNotNone(r)
        self.assertDictEqual(
            r, self.cloud._normalize_coe_cluster(coe_cluster_obj))
        self.assert_calls()

    def test_get_coe_cluster_not_found(self):
        self.register_uris([dict(
            method='GET',
            uri=self.get_mock_url(resource='clusters'),
            json=dict(clusters=[]))])
        r = self.cloud.get_coe_cluster('doesNotExist')
        self.assertIsNone(r)
        self.assert_calls()

    def test_delete_coe_cluster(self):
        self.register_uris([
            dict(
                method='GET',
                uri=self.get_mock_url(resource='clusters'),
                json=dict(clusters=[coe_cluster_obj.toDict()])),
            dict(
                method='DELETE',
                uri=self.get_mock_url(
                    resource='clusters',
                    append=[coe_cluster_obj.uuid])),
        ])
        self.cloud.delete_coe_cluster(coe_cluster_obj.uuid)
        self.assert_calls()

    def test_update_coe_cluster(self):
        self.register_uris([
            dict(
                method='GET',
                uri=self.get_mock_url(resource='clusters'),
                json=dict(clusters=[coe_cluster_obj.toDict()])),
            dict(
                method='PATCH',
                uri=self.get_mock_url(
                    resource='clusters',
                    append=[coe_cluster_obj.uuid]),
                status_code=200,
                validate=dict(
                    json=[{
                        u'op': u'replace',
                        u'path': u'/node_count',
                        u'value': 3
                    }]
                )),
            dict(
                method='GET',
                uri=self.get_mock_url(resource='clusters'),
                # This json value is not meaningful to the test - it just has
                # to be valid.
                json=dict(clusters=[coe_cluster_obj.toDict()])),
        ])
        self.cloud.update_coe_cluster(
            coe_cluster_obj.uuid, 'replace', node_count=3)
        self.assert_calls()