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

from octane import magic_consts
from octane.util import disk as disk_util


@pytest.mark.parametrize("disk", ["sda", "sdb", "sdc"])
@pytest.mark.parametrize("size,last_part,end_part", [
    (10, 1024, 1035),
    (0, 10, 11),
])
def test_create_partition(mocker, mock_ssh_call, mock_ssh_call_output, node,
                          size, last_part, end_part, disk):
    mock_part_end = mocker.patch("octane.util.disk.parse_last_partition_end")
    mock_part_end.return_value = last_part

    disk_util.create_partition(disk, size, node)
    mock_ssh_call_output.assert_called_once_with(
        ['parted', '/dev/%s' % disk, 'unit', 'MB', 'print'], node=node)
    mock_ssh_call.assert_called_once_with(
        ['parted', '/dev/%s' % disk, 'unit', 'MB', 'mkpart',
         'custom', 'ext4', str(last_part + 1), str(end_part)], node=node)


NODE_DISKS_ATTRIBUTE = [
    {
        'id': '1',
        'name': 'disk1',
    }, {
        'id': '2',
        'name': 'disk2',
    }
]


@pytest.mark.parametrize("disk_attrs", [
    NODE_DISKS_ATTRIBUTE,
    None,
])
def test_create_configdrive_partition(mocker, node, disk_attrs):
    name = 'disk1'
    node.mock_add_spec(['get_attribute'])
    node.data = {"id": "1"}
    node.get_attribute.return_value = disk_attrs
    mock_create_part = mocker.patch("octane.util.disk.create_partition")
    if disk_attrs:
        disk_util.create_configdrive_partition(node)
        mock_create_part.assert_called_once_with(
            name, magic_consts.CONFIGDRIVE_PART_SIZE, node)
    else:
        with pytest.raises(disk_util.NoDisksInfoError):
            disk_util.create_configdrive_partition(node)