# 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.

from novaclient.tests.unit.fixture_data import base


class Fixture(base.Fixture):

    base_url = 'os-server-groups'

    def setUp(self):
        super(Fixture, self).setUp()

        server_groups = [
            {
                "members": [],
                "metadata": {},
                "id": "2cbd51f4-fafe-4cdb-801b-cf913a6f288b",
                "policies": [],
                "name": "ig1"
            },
            {
                "members": [],
                "metadata": {},
                "id": "4473bb03-4370-4bfb-80d3-dc8cffc47d94",
                "policies": ["anti-affinity"],
                "name": "ig2"
            },
            {
                "members": [],
                "metadata": {"key": "value"},
                "id": "31ab9bdb-55e1-4ac3-b094-97eeb1b65cc4",
                "policies": [], "name": "ig3"
            },
            {
                "members": ["2dccb4a1-02b9-482a-aa23-5799490d6f5d"],
                "metadata": {},
                "id": "4890bb03-7070-45fb-8453-d34556c87d94",
                "policies": ["anti-affinity"],
                "name": "ig2"
            }
        ]

        other_project_server_groups = [
            {
                "members": [],
                "metadata": {},
                "id": "11111111-1111-1111-1111-111111111111",
                "policies": [],
                "name": "ig4"
            },
            {
                "members": [],
                "metadata": {},
                "id": "22222222-2222-2222-2222-222222222222",
                "policies": ["anti-affinity"],
                "name": "ig5"
            },
            {
                "members": [],
                "metadata": {"key": "value"},
                "id": "33333333-3333-3333-3333-333333333333",
                "policies": [], "name": "ig6"
            },
            {
                "members": ["2dccb4a1-02b9-482a-aa23-5799490d6f5d"],
                "metadata": {},
                "id": "44444444-4444-4444-4444-444444444444",
                "policies": ["anti-affinity"],
                "name": "ig5"
            }
        ]

        headers = self.json_headers

        self.requests_mock.get(self.url(),
                               json={'server_groups': server_groups},
                               headers=headers)

        self.requests_mock.get(self.url(all_projects=True),
                               json={'server_groups': server_groups
                                     + other_project_server_groups},
                               headers=headers)

        self.requests_mock.get(self.url(limit=2, offset=1),
                               json={'server_groups': server_groups[1:3]},
                               headers=headers)

        server = server_groups[0]

        def _register(method, *args):
            self.requests_mock.register_uri(method,
                                            self.url(*args),
                                            json={'server_group': server},
                                            headers=headers)

        _register('POST')
        _register('POST', server['id'])
        _register('GET', server['id'])
        _register('PUT', server['id'])
        _register('POST', server['id'], '/action')

        self.requests_mock.delete(self.url(server['id']),
                                  status_code=202,
                                  headers=headers)
