Changeset 516 for yum/trunk/test/testbase.py

Timestamp:

Feb 3, 2015, 9:39:54 AM (11 years ago)

Author:

Yuri Dario

Message:

yum: update trunk to 3.4.3.

Location:

yum/trunk

Files:

• . (modified) (1 prop)
• test/testbase.py (modified) (14 diffs)

yum/trunk/test/testbase.py

@@ -18 +18 @@
 import inspect
 from rpmUtils import arch
+from rpmUtils.transaction import initReadOnlyTransaction
+import rpmUtils.miscutils
+
 
 #############################################################
@@ -26 +29 @@
 def _(msg):
     return msg
+
+# dummy save_ts to avoid lots of errors
+def save_ts(*args, **kwargs):
+    pass
 
 class FakeConf(object):
@@ -45 +52 @@
         self.showdupesfromrepos = False
         self.uid = 0
+        self.groupremove_leaf_only = False
+        self.protected_packages = []
+        self.protected_multilib = False
+        self.clean_requirements_on_remove = True
+
+class FakeSack:
+    """ Fake PackageSack to use with FakeRepository"""
+    def __init__(self):
+        pass # This is fake, so do nothing
+    
+    def have_fastReturnFileEntries(self):
+        return True
 
 class FakeRepo(object):
 
+    __fake_sack = FakeSack()
     def __init__(self, id=None,sack=None):
         self.id = id
+        if sack is None:
+            sack = self.__fake_sack
         self.sack = sack
         self.cost = 1000
@@ -63 +85 @@
             return 0
 
+class FakeYumDBInfo(object):
+    """Simulate some functionality of RPMAdditionalDataPackage"""
+    _auto_hardlink_attrs = set(['checksum_type', 'reason',
+                                'installed_by', 'changed_by',
+                                'from_repo', 'from_repo_revision',
+                                'from_repo_timestamp', 'releasever',
+                                'command_line'])
+     
+    def __init__(self, conf=None, pkgdir=None, yumdb_cache=None):
+        self.db = {}
+        for attr in self._auto_hardlink_attrs:
+            self.db[attr] = ''
+            
+    def __getattr__(self, attr):
+        return self.db[attr]
+
+    def __setattr__(self, attr, value):
+        if not attr.startswith("db"):
+            self.db[attr] = value
+        else:
+            object.__setattr__(self, attr, value)
+            
+    def __iter__(self, show_hidden=False):
+        for item in self.db:
+            yield item
+
+    def get(self, attr, default=None):
+        try:
+            res = self.db[attr]
+        except AttributeError:
+            return default
+        return res
+        
 class FakePackage(packages.YumAvailablePackage):
 
@@ -81 +136 @@
         self.arch = arch
         self.pkgtup = (self.name, self.arch, self.epoch, self.version, self.release)
-        self.yumdb_info = {}
+        self.yumdb_info = FakeYumDBInfo()
 
         self.prco['provides'].append((name, 'EQ', (epoch, version, release)))
@@ -88 +143 @@
         self.id = self.__hash__()
         self.pkgKey = self.__hash__()
-
+        
+        self.required_pkgs = []
+        self.requiring_pkgs = []
     def addProvides(self, name, flag=None, evr=(None, None, None)):
         self.prco['provides'].append((name, flag, evr))
     def addRequires(self, name, flag=None, evr=(None, None, None)):
         self.prco['requires'].append((name, flag, evr))
+    def addRequiresPkg(self, pkg):
+        self.required_pkgs.append(pkg)
+    def addRequiringPkg(self, pkg):
+        self.requiring_pkgs.append(pkg)   
     def addConflicts(self, name, flag=None, evr=(None, None, None)):
         self.prco['conflicts'].append((name, flag, evr))
@@ -99 +160 @@
     def addFile(self, name, ftype='file'):
         self.files[ftype].append(name)
+    def required_packages(self):
+        return self.required_pkgs
+    def requiring_packages(self):
+        return self.requiring_pkgs
 
 class _Container(object):
@@ -114 +179 @@
     def pkgAdded(self, pkgtup, mode):
         modedict = { 'i': _('installed'),
-                     'u': _('updated'),
-                     'o': _('obsoleted'),
-                     'e': _('erased')}
+                     'u': _('an update'),
+                     'e': _('erased'),
+                     'r': _('reinstalled'),
+                     'd': _('a downgrade'),
+                     'o': _('obsoleting'),
+                     'ud': _('updated'),
+                     'od': _('obsoleted'),}
         (n, a, e, v, r) = pkgtup
         modeterm = modedict[mode]
         self.verbose_logger.log(logginglevels.INFO_2,
-            _('---> Package %s.%s %s:%s-%s set to be %s'), n, a, e, v, r,
+            _('---> Package %s.%s %s:%s-%s will be %s'), n, a, e, v, r,
             modeterm)
         
@@ -281 +350 @@
         return
 
+    def readOnlyTS(self):
+        #  Should probably be able to "fake" this, so we can provide different
+        # get_running_kernel_pkgtup(). Bah.
+        return initReadOnlyTransaction("/")
+
     def getProvides(self, name, flags=None, version=(None, None, None)):
         """return dict { packages -> list of matching provides }"""
@@ -288 +362 @@
         if not version:
             version=(None, None, None)
+        if type(version) in (str, type(None), unicode):
+            version = rpmUtils.miscutils.stringToVersion(version)
         if flags == '0':
             flags=None
@@ -338 +414 @@
         self.tsInfo = transactioninfo.TransactionData()
         
-
     def resolveCode(self):
         solver = YumBase()
+        solver.save_ts  = save_ts
         solver.conf = FakeConf()
         solver.arch.setup_arch('x86_64')
@@ -381 +457 @@
 
     def runOperation(self, args, installed=[], available=[],
-                     confs={}):
+                     confs={}, multi_cmds=False):
         """Sets up and runs the depsolver. args[0] must be a valid yum command
         ("install", "update", ...). It might be followed by pkg names as on the
@@ -389 +465 @@
         """
         depsolver = YumBaseCli()
+        depsolver.save_ts = save_ts
         depsolver.arch.setup_arch('x86_64')
         self.rpmdb = depsolver.rpmdb = FakeRpmDb()
@@ -410 +487 @@
             self.depsolver._pkgSack.addPackage(po)
 
-        self.depsolver.basecmd = args[0]
-        self.depsolver.extcmds = args[1:]
-        res, msg = self.depsolver.doCommands()
+        if not multi_cmds:
+            self.depsolver.basecmd = args[0]
+            self.depsolver.extcmds = args[1:]
+            res, msg = self.depsolver.doCommands()
+        else:
+            for nargs in args:
+                self.depsolver.basecmd = nargs[0]
+                self.depsolver.extcmds = nargs[1:]
+                res, msg = self.depsolver.doCommands()
+                if res != 2:
+                    return res, msg
+
         self.tsInfo = depsolver.tsInfo
         if res!=2: