#!/usr/bin/python

#

# CDDL HEADER START

#

# The contents of this file are subject to the terms of the

# Common Development and Distribution License (the "License").

# You may not use this file except in compliance with the License.

#

# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

# or http://www.opensolaris.org/os/licensing.

# See the License for the specific language governing permissions

# and limitations under the License.

#

# When distributing Covered Code, include this CDDL HEADER in each

# file and include the License file at usr/src/OPENSOLARIS.LICENSE.

# If applicable, add the following below this CDDL HEADER, with the

# fields enclosed by brackets "[]" replaced with your own identifying

# information: Portions Copyright [yyyy] [name of copyright owner]

#

# CDDL HEADER END

#

#

#

'''

Tests auto install to a specified XML Manifest

'''

import os

import sys

import unittest

from solaris_install.auto_install import auto_install

from solaris_install.engine.test.engine_test_utils import reset_engine

class TestAutoInstallManifest(unittest.TestCase):

    '''Tests to auto installation succeeds with -m specified manifest '''

    AI = None

    def setUp(self):

        '''

        Create a auto_install client for testing with.

        '''

        self.AI = None

        self.test_dir = os.path.dirname(os.path.abspath(__file__))

    def tearDown(self):

        '''

        Clean Up

        '''

        if self.AI is not None:

            self.AI = None

    def test_manifest_install_auto_reboot_true(self):

        '''

        Test installation with manifest containing auto_reboot set to true

        '''

        testmanifest = os.path.join(self.test_dir,

                                    "manifest_auto_reboot_true.xml")

        args = ["-n", "-s", "target-discovery", "-m", testmanifest]

        try:

            self.AI = auto_install.AutoInstall(args)

            self.assertNotEqual(self.AI, None)

            self.AI.perform_autoinstall()

            self.assertNotEqual(self.AI.exitval, self.AI.AI_EXIT_FAILURE)

        except KeyboardInterrupt:

            pass

    def test_manifest_install_auto_reboot_false(self):

        '''

        Test installation with manifest containing auto_reboot set to false

        '''

        testmanifest = os.path.join(self.test_dir,

                                    "manifest_auto_reboot_false.xml")

        args = ["-n", "-s", "target-discovery", "-m", testmanifest]

        try:

            self.AI = auto_install.AutoInstall(args)

            self.assertNotEqual(self.AI, None)

            self.AI.perform_autoinstall()

            self.assertNotEqual(self.AI.exitval, self.AI.AI_EXIT_FAILURE)

        except KeyboardInterrupt:

            pass

    def test_manifest_install_auto_reboot_not_set(self):

        '''

        Test installation with manifest containing auto_reboot set to not set

        '''

        testmanifest = os.path.join(self.test_dir,

                                    "manifest_auto_reboot_not_set.xml")

        args = ["-n", "-s", "target-discovery", "-m", testmanifest]

        try:

            self.AI = auto_install.AutoInstall(args)

            self.assertNotEqual(self.AI, None)

            self.AI.perform_autoinstall()

            self.assertNotEqual(self.AI.exitval, self.AI.AI_EXIT_FAILURE)

        except KeyboardInterrupt:

            pass

    def test_manifest_auto_reboot_invalid(self):

        '''

        Test installation with a manifest that fails to parse.

        Achieved by setting auto_reboot to an invalid value

        exitval should be set to AI_EXIT_FAILIRE

        '''

        testmanifest = os.path.join(self.test_dir,

                                    "manifest_auto_reboot_invalid.xml")

        args = ["-n", "-s", "target-discovery", "-m", testmanifest]

        try:

            self.AI = auto_install.AutoInstall(args)

            self.assertNotEqual(self.AI, None)

            self.AI.perform_autoinstall()

            self.assertNotEqual(self.AI.exitval, None)

            self.assertEqual(self.AI.exitval, self.AI.AI_EXIT_FAILURE)

        except:

            raise

if __name__ == '__main__':

    unittest.main()