ResolverPlayground: Add support for world sets

diff --git a/pym/portage/tests/resolver/ResolverPlayground.py b/pym/portage/tests/resolver/ResolverPlayground.py
index 5fb64cfbbb334f2925d0bb1b800d4656f356a944..adf03d2b7a88c39c5df5673c919b829d50c4fe3d 100644 (file)
--- a/pym/portage/tests/resolver/ResolverPlayground.py
+++ b/pym/portage/tests/resolver/ResolverPlayground.py
@@ -57,7 +57,7 @@ class ResolverPlayground(object):
 """
 
        def __init__(self, ebuilds={}, installed={}, profile={}, repo_configs={}, \
-               user_config={}, sets={}, world=[], distfiles={}, debug=False):
+               user_config={}, sets={}, world=[], world_sets=[], distfiles={}, debug=False):
                """
                ebuilds: cpv -> metadata mapping simulating available ebuilds. 
                installed: cpv -> metadata mapping simulating installed packages.
@@ -85,7 +85,7 @@ class ResolverPlayground(object):
                self._create_ebuilds(ebuilds)
                self._create_installed(installed)
                self._create_profile(ebuilds, installed, profile, repo_configs, user_config, sets)
-               self._create_world(world)
+               self._create_world(world, world_sets)
 
                self.settings, self.trees = self._load_config()
 
@@ -444,18 +444,24 @@ class ResolverPlayground(object):
                                f.write("%s\n" % line)
                        f.close()
 
-       def _create_world(self, world):
+       def _create_world(self, world, world_sets):
                #Create /var/lib/portage/world
                var_lib_portage = os.path.join(self.eroot, "var", "lib", "portage")
                os.makedirs(var_lib_portage)
 
                world_file = os.path.join(var_lib_portage, "world")
+               world_set_file = os.path.join(var_lib_portage, "world_sets")
 
                f = open(world_file, "w")
                for atom in world:
                        f.write("%s\n" % atom)
                f.close()
 
+               f = open(world_set_file, "w")
+               for atom in world_sets:
+                       f.write("%s\n" % atom)
+               f.close()
+
        def _load_config(self):
                portdir_overlay = []
                for repo_name in sorted(self.repo_dirs):