from __future__ import with_statement

#

# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

# All rights reserved.

# This component and the accompanying materials are made available

# under the terms of "Eclipse Public License v1.0"

# which accompanies this distribution, and is available

# at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

# Initial Contributors:

# Nokia Corporation - initial contribution.

#

# Contributors:

#

# Description: 

#

import unittest

import os

from cone.public import api

from cone.storage.filestorage import FileStorage

from testautomation import unzip_file

from testautomation.base_testcase import BaseTestCase

ROOT_PATH       = os.path.dirname(os.path.abspath(__file__))

TEMP_DIR        = os.path.join(ROOT_PATH, "temp/edit_data")

TEST_CPF        = os.path.join(ROOT_PATH, "testdata/test_project.cpf")

class open_config_and_get_dview(object):

    def __init__(self, project, mode, config):

        self.project_path = project

        self.mode = mode

        self.config = config

        self.project = None

    def __enter__(self):

        self.project = api.Project(FileStorage(self.project_path, self.mode))

        conf = self.project.get_configuration(self.config)

        dview = conf.get_default_view()

        return dview

        setting = dview.get_feature(self.setting_ref)

        return setting

    def __exit__(self, type, value, tb):

        if self.project:

            if self.mode == 'a':

                self.project.save()

            self.project.close()

class open_config_and_get_setting(open_config_and_get_dview):

    def __init__(self, project, mode, config, setting):

        open_config_and_get_dview.__init__(self, project, mode, config)

        self.setting_ref = setting

    def __enter__(self):

        dview = open_config_and_get_dview.__enter__(self)

        setting = dview.get_feature(self.setting_ref)

        return setting

class TestEditConfigurationProjectData(BaseTestCase):

    def test_edit_sequence_data_on_last_layer_with_dview(self):

        PROJ = os.path.join(TEMP_DIR, "test_project_1")

        CONF = 'root3.confml'

        SETTING = 'Feature2.SequenceSetting'

        unzip_file.unzip_file(TEST_CPF, PROJ, delete_if_exists=True)

        # Open the temp project in append mode

        with open_config_and_get_setting(PROJ, 'a', CONF, SETTING) as setting:

            self.assertEquals(setting.value,

                [[333, 'layer3 (1)'],

                 [1, 'default 1'],

                 [2, 'default 2'],

                 [222, 'layer2 (1)'],

                 [222, 'layer2 (2)'],

                 [222, 'layer2 (3)']])

            # Change the value

            setting.value = [[123, 'foo1'],

                             [456, 'foo2']]

        # Reopen the project in read mode and check that the sequence data

        # was modified correctly

        with open_config_and_get_setting(PROJ, 'r', CONF, SETTING) as setting:

            self.assertEquals(setting.value,

                              [[123, 'foo1'],

                               [456, 'foo2']])

            self.assertEquals(setting.get_original_value(),

                              [['123', 'foo1'],

                               ['456', 'foo2']])

        # Modify the sequence data again by setting it empty

        with open_config_and_get_setting(PROJ, 'a', CONF, SETTING) as setting:

            setting.value = []

            self.assertEquals(setting.value, [])

            self.assertEquals(setting.get_original_value(), [])

        # Reopen and check again

        with open_config_and_get_setting(PROJ, 'r', CONF, SETTING) as setting:

            self.assertEquals(setting.value, [])

            self.assertEquals(setting.get_original_value(), [])

        # Do the same 'set empty' check with a more complex sequence setting

        SETTING = 'SequenceSettingTest.SequenceSetting'

        with open_config_and_get_setting(PROJ, 'a', CONF, SETTING) as setting:

            setting.value = []

            self.assertEquals(setting.value, [])

            self.assertEquals(setting.get_original_value(), [])

        with open_config_and_get_setting(PROJ, 'r', CONF, SETTING) as setting:

            self.assertEquals(setting.value, [])

            self.assertEquals(setting.get_original_value(), [])

    def test_edit_complex_sequence_data_on_last_layer_with_dview(self):

        PROJ = os.path.join(TEMP_DIR, "test_project_4")

        CONF = 'root3.confml'

        SETTING = 'SequenceSettingTest.SequenceSetting'

        unzip_file.unzip_file(TEST_CPF, PROJ, delete_if_exists=True)

        # Open the temp project in append mode

        with open_config_and_get_setting(PROJ, 'a', CONF, SETTING) as setting:

            self.assertEquals(setting.value,

                [[['seq/layer2_folder2', None], 2.1000000000000001, ['seq/layer2_file2.txt', None], 222, 'L22', True, '1', ('opt 1', 'opt 3'), '\x22\x22'],

                 [['seq/layer2_folder', None], 2.0, ['seq/layer2_file.txt', None], 22, 'L21', True, '2', ('opt 2',), '\x22\x11'],

                 [['seq/def1_folder', None], 1.25, ['seq/def1_file.txt', None], 128, 'def1', False, '1', ('opt 1',), '\x00\x11'],

                 [['seq/def2_folder', None], 1.5, ['seq/def2_file.txt', None], 256, 'def2', False, '1', ('opt 2',), '\x00\x22'],

                 [['seq/layer3_folder', None], 3.0, ['seq/layer3_file.txt', None], 33, 'L31', False, '0', ('opt 3', 'opt 2'), '\x33\x11'],

                 [['seq/layer3_folder', None], 3.3500000000000001, [None, None], 1, 'L32', True, '2', ('opt 1', 'opt 2'), '\x33\x22']])

            # Change the value

            setting.value = [[['seq/foofolder1', None], 5.6, ['seq/foofile1.txt', None], 1010, 'Lfoo', False, '3', ('opt 4', 'opt 0'), '\xaa\xbb'],

                             [['seq/foofolder2', None], 6.5, ['seq/foofile2.txt', None], 2020, 'Lfoo', True, '3', (), ''],

                             [['seq/foofolder1', None], 5.6, ['seq/foofile1.txt', None], 1010, 'Lfoo', False, '3', ('opt 4', 'opt 0'), '\xaa\xbb']]

        with open_config_and_get_setting(PROJ, 'r', CONF, SETTING) as setting:

            self.assertEquals(setting.value,

                [[['seq/foofolder1', None], 5.6, ['seq/foofile1.txt', None], 1010, 'Lfoo', False, '3', ('opt 4', 'opt 0'), '\xaa\xbb'],

                 [['seq/foofolder2', None], 6.5, ['seq/foofile2.txt', None], 2020, 'Lfoo', True, '3', (), ''],

                 [['seq/foofolder1', None], 5.6, ['seq/foofile1.txt', None], 1010, 'Lfoo', False, '3', ('opt 4', 'opt 0'), '\xaa\xbb']])

        # Modify the sequence data again by setting it empty

        with open_config_and_get_setting(PROJ, 'a', CONF, SETTING) as setting:

            setting.value = []

        # Reopen and check again

        with open_config_and_get_setting(PROJ, 'r', CONF, SETTING) as setting:

            self.assertEquals(setting.value, [])

    def test_edit_data_on_last_layer_with_dview(self):

        PROJ = os.path.join(TEMP_DIR, "test_project_2")

        CONF = 'root3.confml'

        unzip_file.unzip_file(TEST_CPF, PROJ, delete_if_exists=True)

        # Open the temp project in append mode

        with open_config_and_get_dview(PROJ, 'a', CONF) as dview:

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.IntSetting').value,          333)

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.BooleanSetting').value,      False)

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.RealSetting').value,         3.1456)

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.SelectionSetting').value,    '3')

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.StringSetting').value,       'layer 3 string')

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.MultiSelectionSetting').value, ('opt 1', 'opt 3'))

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.HexBinarySetting').value, '\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff')

            dview.get_feature('BasicSettingTypesTest.IntSetting').value         = 1500

            dview.get_feature('BasicSettingTypesTest.BooleanSetting').value     = True

            dview.get_feature('BasicSettingTypesTest.RealSetting').value        = 15.15

            dview.get_feature('BasicSettingTypesTest.SelectionSetting').value   = '1'

            dview.get_feature('BasicSettingTypesTest.StringSetting').value      = 'edit data test'

            dview.get_feature('BasicSettingTypesTest.MultiSelectionSetting').value = ('opt 4', 'opt 0')

            dview.get_feature('BasicSettingTypesTest.HexBinarySetting').value = '\x12\x34'

        # Reopen the project in read mode and check that the sequence data

        # was modified correctly

        with open_config_and_get_dview(PROJ, 'r', CONF) as dview:

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.IntSetting').value,          1500)

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.BooleanSetting').value,      True)

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.RealSetting').value,         15.15)

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.SelectionSetting').value,    '1')

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.StringSetting').value,       'edit data test')

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.MultiSelectionSetting').value, ('opt 4', 'opt 0'))

            self.assertEquals(dview.get_feature('BasicSettingTypesTest.HexBinarySetting').value, '\x12\x34')

        # Test that setting the multi-selection setting to empty works

        SETTING = 'BasicSettingTypesTest.MultiSelectionSetting'

        with open_config_and_get_setting(PROJ, 'a', 'root4.confml', SETTING) as setting:

            self.assertEquals(setting.value, ('opt 0', 'opt 4'))

            setting.value = ()

            self.assertEquals(setting.value, ())

        with open_config_and_get_setting(PROJ, 'r', 'root4.confml', SETTING) as setting:

            self.assertEquals(setting.value, ())

    def test_read_name_id_mapped_values(self):

        PROJ = os.path.join(TEMP_DIR, "test_project_5")

        CONF = 'root3.confml'

        unzip_file.unzip_file(TEST_CPF, PROJ, delete_if_exists=True)

        with open_config_and_get_dview(PROJ, 'r', CONF) as dview:

            def check(setting, expected_value):

                fea = dview.get_feature('NameIdMappingTestTargetSettings.' + setting)

                self.assertEquals(fea.get_value(), expected_value)

            check('Selection',       'seq1_item2')

            check('Selection2',      12)

            check('MultiSelection',  ('seq1_item1', 'seq1_item2', 'seq2_item2', 'seq2_item3'))

            check('MultiSelection2', (11, 12, True, False))

            check('String',          'seq1_item2')

            check('Int',             12)

            check('Real',            1.2)

            check('Boolean',         False)

            check('Sequence.Selection',       ['seq1_item2'])

            check('Sequence.Selection2',      [12])

            check('Sequence.MultiSelection',  [('seq1_item1', 'seq1_item2', 'seq2_item2', 'seq2_item3')])

            check('Sequence.MultiSelection2', [(11, 12, True, False)])

            check('Sequence.String',          ['seq1_item2'])

            check('Sequence.Int',             [12])

            check('Sequence.Real',            [1.2])

            check('Sequence.Boolean',         [False])

if __name__ == '__main__':

    unittest.main()