#!/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

#

import testutils

if __name__ == "__main__":

	testutils.setup_environment("../../../proto")

import pkg5unittest

import os

import pkg.portable as portable

import unittest

class TestROption(pkg5unittest.SingleDepotTestCase):

        # Cleanup after every test.

        persistent_setup = False

        foo10 = """

            open [email protected],5.11-0

            close """

        def setUp(self):

                pkg5unittest.SingleDepotTestCase.setUp(self)

                self.pkgsend_bulk(self.rurl, self.foo10)

                self.image_create(self.rurl)

	def test_bad_cli_options(self):

                """Verify that pkg rejects invalid -R combos and values."""

		self.pkg("-@", exit=2)

		self.pkg("-s status", exit=2)

                self.pkg("-R status", exit=2)

                self.pkg("-R / version", exit=2)

        def test_1_explicit(self):

                """Ensure that pkg explicit image specification works as

                expected."""

                badpath = self.test_root

                # Verify that bad paths cause exit and good paths succeed.

                self.pkg("-R %s list" % badpath, exit=1)

                self.pkg("-R %s list" % imgpath, exit=1)

                self.pkg("-R %s install foo" % badpath, exit=1)

                self.pkg("-R %s install foo" % imgpath)

                self.pkg("-R %s list" % badpath, exit=1)

                self.pkg("-R %s list" % imgpath)

                self.pkgsend_bulk(self.rurl, self.foo10)

                self.pkg("-R %s refresh" % imgpath)

                self.pkg("-R %s update" % badpath, exit=1)

                self.pkg("-R %s update --be-name NEWBENAME" % imgpath, exit=1)

                self.pkg("-R %s update" % imgpath)

                self.pkg("-R %s uninstall foo" % badpath, exit=1)

                self.pkg("-R %s install foo" % imgpath, exit=4)

                self.pkg("-R %s info foo" % badpath, exit=1)

                self.pkg("-R %s info foo" % imgpath)

        def test_2_implicit(self):

                """Ensure that pkg implicit image finding works as expected."""

                # Should fail because $PKG_IMAGE is set to test root by default,

                # disabled.

                self.pkg("install foo", exit=1, use_img_root=False)

                # Unset unit testing default bogus image dir.

                del os.environ["PKG_IMAGE"]

                if portable.osname != "sunos":

                        # For other platforms, first install a package uses an

                        # explicit root, and then verify that an implicit find

                        # of the image results in the right image being found.

                        self.pkg("install foo")

                        self.pkg("info foo", use_img_root=False)

                        # Remaining tests are not valid on other platforms.

                        return

                # Should fail because image found is not at '/', but at cwd().

                self.pkg("install foo", exit=1, use_img_root=False)

                # Should succeed because image is found at simulated live root.

                self.pkg("-D simulate_live_root=%s install foo" %

if __name__ == "__main__":

        unittest.main()