# Copyright 2018 Intel, Inc.
# All Rights Reserved.
#
#    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 mock
from sushy import exceptions
import testtools

from rsd_lib import utils as rsd_lib_utils


class UtilsTestCase(testtools.TestCase):

    def test_get_resource_identity(self):
        self.assertIsNone(rsd_lib_utils.get_resource_identity(None))
        self.assertIsNone(rsd_lib_utils.get_resource_identity({}))
        self.assertEqual(
            '/redfish/v1/Systems/437XR1138R2/BIOS',
            rsd_lib_utils.get_resource_identity({
                "@odata.id": "/redfish/v1/Systems/437XR1138R2/BIOS"}))

    def test_num_or_none(self):
        self.assertIsNone(rsd_lib_utils.num_or_none(None))
        self.assertEqual(0, rsd_lib_utils.num_or_none('0'))
        self.assertEqual(1, rsd_lib_utils.num_or_none('1'))
        self.assertEqual(10, rsd_lib_utils.num_or_none('10.0'))
        self.assertEqual(12.5, rsd_lib_utils.num_or_none('12.5'))
        self.assertEqual(0, rsd_lib_utils.num_or_none(0))
        self.assertEqual(1, rsd_lib_utils.num_or_none(1))
        self.assertEqual(10, rsd_lib_utils.num_or_none(10.0))
        self.assertEqual(12.5, rsd_lib_utils.num_or_none(12.5))

    def test_get_sub_resource_path_list_by(self):
        sample = {
            "Links": {
                "PCIeDevices": [
                    {"@data.id": "/redfish/v1/Chassis/1/PCIeDevices/Device1"},
                    {"@data.id": "/redfish/v1/Chassis/1/PCIeDevices/Device2"}
                ]
            }
        }

        mock_resource = mock.Mock()
        mock_resource.json = sample

        self.assertRaises(
            ValueError,
            rsd_lib_utils.get_sub_resource_path_list_by,
            mock_resource,
            None)

        self.assertEqual(
            sorted([
                '/redfish/v1/Chassis/1/PCIeDevices/Device1',
                '/redfish/v1/Chassis/1/PCIeDevices/Device2'
            ]),
            sorted(rsd_lib_utils.get_sub_resource_path_list_by(
                mock_resource, ["Links", "PCIeDevices"]))
        )

        mock_resource.json = {'Links': {}}
        self.assertRaises(
            exceptions.MissingAttributeError,
            rsd_lib_utils.get_sub_resource_path_list_by,
            mock_resource,
            'Links'
        )

    def test_camelcase_to_underscore_joined(self):
        input_vs_expected = [
            ('GarbageCollection', 'garbage_collection'),
            ('DD', 'dd'),
            ('rr', 'rr'),
            ('AABbbC', 'aa_bbb_c'),
            ('AABbbCCCDd', 'aa_bbb_ccc_dd'),
            ('Manager', 'manager'),
            ('EthernetInterfaceCollection', 'ethernet_interface_collection'),
            (' ', ' '),
        ]
        for inp, exp in input_vs_expected:
            self.assertEqual(
                exp, rsd_lib_utils.camelcase_to_underscore_joined(inp))

    def test_camelcase_to_underscore_joined_fails_with_empty_string(self):
        self.assertRaisesRegex(
            ValueError,
            '"camelcase_str" cannot be empty',
            rsd_lib_utils.camelcase_to_underscore_joined, '')