# Copyright 2014 Mirantis, Inc.
#
# 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 copy

import unittest2

from bareon.drivers.data import ks_spaces_validator as kssv
from bareon import errors

SAMPLE_SCHEME = [
    {
        "id": {
            "type": "name",
            "value": "sda"
        },
        "name": "sda",
        "volumes": [
            {
                "type": "boot",
                "size": "300"
            },
            {
                "mount": "/boot",
                "size": "200",
                "type": "raid",
                "file_system": "ext2",
                "name": "Boot"
            },
            {
                "type": "lvm_meta_pool",
                "size": "0"
            },
            {
                "size": "19438",
                "type": "pv",
                "lvm_meta_size": "64",
                "vg": "os"
            },
            {
                "size": "45597",
                "type": "pv",
                "lvm_meta_size": "64",
                "vg": "image"
            }
        ],
        "type": "disk",
        "size": "65535"
    },
    {
        "id": {
            "type": "name",
            "value": "sdb"
        },
        "name": "sdb",
        "volumes": [
            {
                "type": "boot",
                "size": "300"
            },
            {
                "mount": "/boot",
                "size": "200",
                "type": "raid",
                "file_system": "ext2",
                "name": "Boot"
            },
            {
                "type": "lvm_meta_pool",
                "size": "64"
            },
            {
                "size": "0",
                "type": "pv",
                "lvm_meta_size": "0",
                "vg": "os"
            },
            {
                "size": "64971",
                "type": "pv",
                "lvm_meta_size": "64",
                "vg": "image"
            }
        ],
        "type": "disk",
        "size": "65535"
    },
    {
        "name": "sdc",
        "volumes": [
            {
                "type": "boot",
                "size": "300"
            },
            {
                "mount": "/boot",
                "size": "200",
                "type": "raid",
                "file_system": "ext2",
                "name": "Boot"
            },
            {
                "type": "lvm_meta_pool",
                "size": "64"
            },
            {
                "size": "0",
                "type": "pv",
                "lvm_meta_size": "0",
                "vg": "os"
            },
            {
                "size": "64971",
                "type": "pv",
                "lvm_meta_size": "64",
                "vg": "image"
            }
        ],
        "type": "disk",
        "id": {
            "type": "path",
            "value": "disk/by-path/pci-0000:00:0d.0-scsi-0:0:0:0"
        },
        "size": "65535"
    },
    {
        "_allocate_size": "min",
        "label": "Base System",
        "min_size": 19374,
        "volumes": [
            {
                "mount": "/",
                "size": "15360",
                "type": "lv",
                "name": "root",
                "file_system": "ext4"
            },
            {
                "mount": "swap",
                "size": "4014",
                "type": "lv",
                "name": "swap",
                "file_system": "swap"
            }
        ],
        "type": "vg",
        "id": "os"
    },
    {
        "_allocate_size": "all",
        "label": "Image Storage",
        "min_size": 5120,
        "volumes": [
            {
                "mount": "/var/lib/glance",
                "size": "175347",
                "type": "lv",
                "name": "glance",
                "file_system": "xfs"
            }
        ],
        "type": "vg",
        "id": "image"
    }
]


class TestKSSpacesValidator(unittest2.TestCase):
    def setUp(self):
        super(TestKSSpacesValidator, self).setUp()
        self.fake_scheme = copy.deepcopy(SAMPLE_SCHEME)

    def test_validate_ok(self):
        kssv.validate(self.fake_scheme, 'ironic')

    def test_validate_jsoschema_fail(self):
        self.assertRaises(errors.WrongPartitionSchemeError,
                          kssv.validate, [{}], 'ironic')

    def test_validate_no_disks_fail(self):
        self.assertRaises(errors.WrongPartitionSchemeError,
                          kssv.validate, self.fake_scheme[-2:], 'ironic')

    @unittest2.skip("Fix after cray rebase")
    def test_validate_16T_root_volume_fail(self):
        self.fake_scheme[3]['volumes'][0]['size'] = 16777216 + 1
        self.assertRaises(errors.WrongPartitionSchemeError,
                          kssv.validate, self.fake_scheme, 'ironic')

    def test_validate_volume_type_fail(self):
        incorrect_values_for_type = [
            False, True, 0, 1, None, object
        ]
        for value in incorrect_values_for_type:
            self.fake_scheme[0]['volumes'][1]['type'] = value
            self.assertRaises(errors.WrongPartitionSchemeError,
                              kssv.validate, self.fake_scheme, 'ironic')

    def test_validate_volume_size_fail(self):
        incorrect_values_for_size = [
            False, True, 0, 1, None, object
        ]
        for value in incorrect_values_for_size:
            self.fake_scheme[0]['volumes'][1]['size'] = value
            self.assertRaises(errors.WrongPartitionSchemeError,
                              kssv.validate, self.fake_scheme, 'ironic')

    def test_validate_device_id_fail(self):
        incorrect_values_for_id = [
            False, True, 0, 1, None, object
        ]
        for value in incorrect_values_for_id:
            self.fake_scheme[0]['id'] = value
            self.assertRaises(errors.WrongPartitionSchemeError,
                              kssv.validate, self.fake_scheme, 'ironic')

    def test_validate_missed_property(self):
        required = ['id', 'size', 'volumes', 'type']
        for prop in required:
            fake = copy.deepcopy(self.fake_scheme)
            del fake[0][prop]
            self.assertRaises(errors.WrongPartitionSchemeError,
                              kssv.validate, fake, 'ironic')

    def test_validate_missed_volume_property(self):
        required = ['type', 'size', 'vg']
        for prop in required:
            fake = copy.deepcopy(self.fake_scheme)
            del fake[0]['volumes'][3][prop]
            self.assertRaises(errors.WrongPartitionSchemeError,
                              kssv.validate, fake, 'ironic')