#   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 osc_lib.tests import utils as test_utils
from osc_lib.utils import columns as column_utils


class TestColumnUtils(test_utils.TestCase):

    def test_get_column_definitions(self):
        attr_map = (
            ('id', 'ID', column_utils.LIST_BOTH),
            ('tenant_id', 'Project', column_utils.LIST_LONG_ONLY),
            ('name', 'Name', column_utils.LIST_BOTH),
            ('summary', 'Summary', column_utils.LIST_SHORT_ONLY),
        )
        headers, columns = column_utils.get_column_definitions(
            attr_map, long_listing=False)
        self.assertEqual(['id', 'name', 'summary'], columns)
        self.assertEqual(['ID', 'Name', 'Summary'], headers)

    def test_get_column_definitions_long(self):
        attr_map = (
            ('id', 'ID', column_utils.LIST_BOTH),
            ('tenant_id', 'Project', column_utils.LIST_LONG_ONLY),
            ('name', 'Name', column_utils.LIST_BOTH),
            ('summary', 'Summary', column_utils.LIST_SHORT_ONLY),
        )
        headers, columns = column_utils.get_column_definitions(
            attr_map, long_listing=True)
        self.assertEqual(['id', 'tenant_id', 'name'], columns)
        self.assertEqual(['ID', 'Project', 'Name'], headers)

    def test_get_columns(self):
        item = {
            'id': 'test-id',
            'tenant_id': 'test-tenant_id',
            # 'name' is not included
            'foo': 'bar',  # unknown attribute
        }
        attr_map = (
            ('id', 'ID', column_utils.LIST_BOTH),
            ('tenant_id', 'Project', column_utils.LIST_LONG_ONLY),
            ('name', 'Name', column_utils.LIST_BOTH),
        )
        columns, display_names = column_utils.get_columns(item, attr_map)
        self.assertEqual(tuple(['id', 'tenant_id', 'foo']), columns)
        self.assertEqual(tuple(['ID', 'Project', 'foo']), display_names)
