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

from openstackclient.tests.functional.volume.v2 import common


class TransferRequestTests(common.BaseVolumeTests):
    """Functional tests for transfer request."""

    API_VERSION = '2'

    def test_volume_transfer_request_accept(self):
        volume_name = uuid.uuid4().hex
        xfer_name = uuid.uuid4().hex

        # create a volume
        cmd_output = self.openstack(
            'volume create ' + '--size 1 ' + volume_name,
            parse_output=True,
        )
        self.assertEqual(volume_name, cmd_output['name'])
        self.addCleanup(
            self.openstack,
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume delete '
            + volume_name,
        )
        self.wait_for_status("volume", volume_name, "available")

        # create volume transfer request for the volume
        # and get the auth_key of the new transfer request
        cmd_output = self.openstack(
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume transfer request create '
            + ' --name '
            + xfer_name
            + ' '
            + volume_name,
            parse_output=True,
        )
        self.assertEqual(xfer_name, cmd_output['name'])
        xfer_id = cmd_output['id']
        auth_key = cmd_output['auth_key']
        self.assertTrue(auth_key)
        self.wait_for_status("volume", volume_name, "awaiting-transfer")

        # accept the volume transfer request
        cmd_output = self.openstack(
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume transfer request accept '
            + '--auth-key '
            + auth_key
            + ' '
            + xfer_id,
            parse_output=True,
        )
        self.assertEqual(xfer_name, cmd_output['name'])
        self.wait_for_status("volume", volume_name, "available")

    def test_volume_transfer_request_list_show(self):
        volume_name = uuid.uuid4().hex
        xfer_name = uuid.uuid4().hex

        # create a volume
        cmd_output = self.openstack(
            'volume create ' + '--size 1 ' + volume_name,
            parse_output=True,
        )
        self.assertEqual(volume_name, cmd_output['name'])
        self.addCleanup(
            self.openstack,
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume delete '
            + volume_name,
        )
        self.wait_for_status("volume", volume_name, "available")

        cmd_output = self.openstack(
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume transfer request create '
            + ' --name '
            + xfer_name
            + ' '
            + volume_name,
            parse_output=True,
        )
        self.assertEqual(xfer_name, cmd_output['name'])
        xfer_id = cmd_output['id']
        auth_key = cmd_output['auth_key']
        self.assertTrue(auth_key)
        self.wait_for_status("volume", volume_name, "awaiting-transfer")

        cmd_output = self.openstack(
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume transfer request list',
            parse_output=True,
        )
        self.assertIn(xfer_name, [req['Name'] for req in cmd_output])

        cmd_output = self.openstack(
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume transfer request show '
            + xfer_id,
            parse_output=True,
        )
        self.assertEqual(xfer_name, cmd_output['name'])

        # NOTE(dtroyer): We need to delete the transfer request to allow the
        #                volume to be deleted. The addCleanup() route does
        #                not have a mechanism to wait for the volume status
        #                to become 'available' before attempting to  delete
        #                the volume.
        cmd_output = self.openstack(
            '--os-volume-api-version '
            + self.API_VERSION
            + ' '
            + 'volume transfer request delete '
            + xfer_id
        )
        self.wait_for_status("volume", volume_name, "available")