t_plat.py revision 1715
1516N/A#!/usr/bin/python
290N/A#
290N/A# CDDL HEADER START
290N/A#
290N/A# The contents of this file are subject to the terms of the
290N/A# Common Development and Distribution License (the "License").
290N/A# You may not use this file except in compliance with the License.
290N/A#
290N/A# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
290N/A# or http://www.opensolaris.org/os/licensing.
290N/A# See the License for the specific language governing permissions
290N/A# and limitations under the License.
290N/A#
290N/A# When distributing Covered Code, include this CDDL HEADER in each
290N/A# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
290N/A# If applicable, add the following below this CDDL HEADER, with the
290N/A# fields enclosed by brackets "[]" replaced with your own identifying
290N/A# information: Portions Copyright [yyyy] [name of copyright owner]
290N/A#
290N/A# CDDL HEADER END
290N/A#
290N/A
1715N/A# Copyright 2010 Sun Microsystems, Inc. All rights reserved.
290N/A# Use is subject to license terms.
290N/A
1715N/Aimport testutils
1715N/Aif __name__ == "__main__":
1715N/A testutils.setup_environment("../../../proto")
1715N/Aimport pkg5unittest
1715N/A
526N/Aimport unittest
290N/Aimport os
402N/Aimport shutil
430N/Aimport sys
290N/Aimport tempfile
691N/Aimport pkg.pkgsubprocess as subprocess
290N/Aimport pkg.fmri as fmri
402N/Aimport pkg.client.image as image
290N/Aimport pkg.portable.util as util
290N/Aimport pkg.portable as portable
290N/A
430N/Aclass TestPlat(pkg5unittest.Pkg5TestCase):
402N/A
402N/A def testbasic(self):
402N/A portable.get_isainfo()
402N/A portable.get_release()
402N/A portable.get_platform()
290N/A
290N/A
402N/A def testAdmin(self):
402N/A if os.name == 'posix' and os.getuid() == 0:
402N/A self.assert_(portable.is_admin())
402N/A if os.name == 'posix' and os.getuid() != 0:
402N/A self.assert_(not portable.is_admin())
402N/A
402N/A def testUtils(self):
402N/A self.assertNotEqual("unknown", util.get_canonical_os_type())
402N/A self.assertNotEqual("unknown", util.get_canonical_os_name())
402N/A
402N/A def testRelease(self):
402N/A rel = util.get_os_release()
402N/A # make sure it can be used in an fmri
402N/A test_fmri = fmri.PkgFmri("testpkg", build_release = rel)
290N/A
402N/A def testForcibleRename(self):
402N/A # rename a file on top of another file which already exists
402N/A (fd1, path1) = tempfile.mkstemp()
402N/A os.write(fd1, "foo")
402N/A (fd2, path2) = tempfile.mkstemp()
402N/A os.write(fd2, "bar")
402N/A os.close(fd1)
402N/A os.close(fd2)
402N/A portable.rename(path1, path2)
402N/A self.failIf(os.path.exists(path1))
402N/A self.failUnless(os.path.exists(path2))
402N/A fd2 = os.open(path2, os.O_RDONLY)
402N/A self.assertEquals(os.read(fd2, 3), "foo")
402N/A os.close(fd2)
402N/A os.unlink(path2)
290N/A
402N/A def testRenameOfRunningExecutable(self):
402N/A if util.get_canonical_os_type() != 'windows':
402N/A return
402N/A import pkg.portable.os_windows as os_windows
1305N/A cwd = os.getcwdu()
402N/A exefilesrc = 'C:\\Windows\\system32\\more.com'
402N/A self.assert_(os.path.exists(exefilesrc))
290N/A
402N/A # create an image, copy an executable into it,
402N/A # run the executable, replace the executable
402N/A tdir1 = tempfile.mkdtemp()
1370N/A img1 = image.Image(tdir1, imgtype=image.IMG_USER,
1370N/A should_exist=False)
539N/A img1.history.client_name = "pkg-test"
1504N/A img1.set_attrs(False, "test",
1504N/A origins=["http://localhost:10000"], refresh_allowed=False)
402N/A exefile = os.path.join(tdir1, 'less.com')
402N/A shutil.copyfile(exefilesrc, exefile)
402N/A proc = subprocess.Popen([exefile], stdin = subprocess.PIPE)
402N/A self.assertRaises(OSError, os.unlink, exefile)
402N/A fd1, path1 = tempfile.mkstemp(dir = tdir1)
402N/A os.write(fd1, "foo")
402N/A os.close(fd1)
402N/A portable.rename(path1, exefile)
402N/A fd2 = os.open(exefile, os.O_RDONLY)
402N/A self.assertEquals(os.read(fd2, 3), "foo")
402N/A os.close(fd2)
402N/A proc.communicate()
290N/A
402N/A # Make sure that the moved executable gets deleted
1305N/A # This is a white-box test
1305N/A # To simulate running another process, we delete the cache
1305N/A # and call get_trashdir as if another file was being moved
1305N/A # to the trash.
1305N/A os_windows.cached_image_info = []
1305N/A os_windows.get_trashdir(exefile)
402N/A self.assert_(not os.path.exists(os.path.join(img1.imgdir,
402N/A os_windows.trashname)))
402N/A
402N/A # cleanup
1305N/A os.chdir(cwd)
1305N/A shutil.rmtree(tdir1)
481N/A
481N/A def testRemoveOfRunningExecutable(self):
481N/A if util.get_canonical_os_type() != 'windows':
481N/A return
481N/A import pkg.portable.os_windows as os_windows
1305N/A cwd = os.getcwdu()
481N/A exefilesrc = 'C:\\Windows\\system32\\more.com'
481N/A self.assert_(os.path.exists(exefilesrc))
481N/A
481N/A # create an image, copy an executable into it,
481N/A # run the executable, remove the executable
481N/A tdir1 = tempfile.mkdtemp()
1370N/A img1 = image.Image(tdir1, imgtype=image.IMG_USER,
1370N/A should_exist=False)
539N/A img1.history.client_name = "pkg-test"
1504N/A img1.set_attrs(False, "test",
1504N/A origins=["http://localhost:10000"], refresh_allowed=False)
481N/A exefile = os.path.join(tdir1, 'less.com')
481N/A shutil.copyfile(exefilesrc, exefile)
481N/A proc = subprocess.Popen([exefile], stdin = subprocess.PIPE)
481N/A self.assertRaises(OSError, os.unlink, exefile)
481N/A portable.remove(exefile)
481N/A self.assert_(not os.path.exists(exefile))
481N/A proc.communicate()
481N/A
1305N/A # Make sure that the moved executable gets deleted
1305N/A # This is a white-box test
1305N/A # To simulate running another process, we delete the cache
1305N/A # and call get_trashdir as if another file was being moved
1305N/A # to the trash.
1305N/A os_windows.cached_image_info = []
1305N/A os_windows.get_trashdir(exefile)
1305N/A self.assert_(not os.path.exists(os.path.join(img1.imgdir,
481N/A os_windows.trashname)))
481N/A
481N/A # cleanup
1305N/A os.chdir(cwd)
1305N/A shutil.rmtree(tdir1)
402N/A
290N/Aif __name__ == "__main__":
290N/A unittest.main()