#
|
# Copyright (C) 2016 Intel Corporation
|
#
|
# SPDX-License-Identifier: MIT
|
#
|
|
from oeqa.core.decorator import OETestDecorator, registerDecorator
|
from oeqa.core.utils.misc import strToSet
|
|
@registerDecorator
|
class OEHasPackage(OETestDecorator):
|
"""
|
Checks if image has packages (un)installed.
|
|
The argument must be a string, set, or list of packages that must be
|
installed or not present in the image.
|
|
The way to tell a package must not be in an image is using an
|
exclamation point ('!') before the name of the package.
|
|
If test depends on pkg1 or pkg2 you need to use:
|
@OEHasPackage({'pkg1', 'pkg2'})
|
|
If test depends on pkg1 and pkg2 you need to use:
|
@OEHasPackage('pkg1')
|
@OEHasPackage('pkg2')
|
|
If test depends on pkg1 but pkg2 must not be present use:
|
@OEHasPackage({'pkg1', '!pkg2'})
|
"""
|
|
attrs = ('need_pkgs',)
|
|
def setUpDecorator(self):
|
need_pkgs = set()
|
unneed_pkgs = set()
|
pkgs = strToSet(self.need_pkgs)
|
for pkg in pkgs:
|
if pkg.startswith('!'):
|
unneed_pkgs.add(pkg[1:])
|
else:
|
need_pkgs.add(pkg)
|
|
if unneed_pkgs:
|
msg = 'Checking if %s is not installed' % ', '.join(unneed_pkgs)
|
self.logger.debug(msg)
|
if not self.case.tc.image_packages.isdisjoint(unneed_pkgs):
|
msg = "Test can't run with %s installed" % ', or '.join(unneed_pkgs)
|
self._decorator_fail(msg)
|
|
if need_pkgs:
|
msg = 'Checking if at least one of %s is installed' % ', '.join(need_pkgs)
|
self.logger.debug(msg)
|
if self.case.tc.image_packages.isdisjoint(need_pkgs):
|
msg = "Test requires %s to be installed" % ', or '.join(need_pkgs)
|
self._decorator_fail(msg)
|
|
def _decorator_fail(self, msg):
|
self.case.skipTest(msg)
|
|
@registerDecorator
|
class OERequirePackage(OEHasPackage):
|
"""
|
Checks if image has packages (un)installed.
|
It is almost the same as OEHasPackage, but if dependencies are missing
|
the test case fails.
|
|
The argument must be a string, set, or list of packages that must be
|
installed or not present in the image.
|
|
The way to tell a package must not be in an image is using an
|
exclamation point ('!') before the name of the package.
|
|
If test depends on pkg1 or pkg2 you need to use:
|
@OERequirePackage({'pkg1', 'pkg2'})
|
|
If test depends on pkg1 and pkg2 you need to use:
|
@OERequirePackage('pkg1')
|
@OERequirePackage('pkg2')
|
|
If test depends on pkg1 but pkg2 must not be present use:
|
@OERequirePackage({'pkg1', '!pkg2'})
|
"""
|
|
def _decorator_fail(self, msg):
|
self.case.fail(msg)
|
