150 lines
4.5 KiB
Python
150 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, '.')
|
|
sys.path.insert(0, '../')
|
|
import unittest
|
|
import entropy.tools as et
|
|
from entropy.client.interfaces import Client
|
|
import tests._misc as _misc
|
|
import tempfile
|
|
import shutil
|
|
|
|
class SpmTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
sys.stdout.write("%s called\n" % (self,))
|
|
sys.stdout.flush()
|
|
self.Client = Client(noclientdb = 2, indexing = False, xcache = False,
|
|
repo_validation = False)
|
|
self.test_pkg = _misc.get_test_entropy_package()
|
|
self.test_pkg2 = _misc.get_test_entropy_package2()
|
|
self.test_pkg3 = _misc.get_test_entropy_package3()
|
|
self.test_pkgs = [self.test_pkg, self.test_pkg2, self.test_pkg3]
|
|
|
|
def tearDown(self):
|
|
"""
|
|
tearDown is run after each test
|
|
"""
|
|
sys.stdout.write("%s ran\n" % (self,))
|
|
sys.stdout.flush()
|
|
self.Client.shutdown()
|
|
|
|
def test_init(self):
|
|
spm = self.Client.Spm()
|
|
spm2 = self.Client.Spm()
|
|
self.assert_(spm is spm2)
|
|
spm_class = self.Client.Spm_class()
|
|
spm_class2 = self.Client.Spm_class()
|
|
self.assert_(spm_class is spm_class2)
|
|
|
|
def test_basic_methods(self):
|
|
spm = self.Client.Spm()
|
|
spm_class = self.Client.Spm_class()
|
|
|
|
path = spm.get_user_installed_packages_file()
|
|
self.assert_(path)
|
|
|
|
groups = spm_class.get_package_groups()
|
|
self.assert_(isinstance(groups, dict))
|
|
|
|
keys = spm.package_metadata_keys()
|
|
self.assert_(isinstance(keys, list))
|
|
|
|
cache_dir = spm.get_cache_directory()
|
|
self.assert_(cache_dir)
|
|
|
|
sys_pkgs = spm.get_system_packages()
|
|
self.assert_(sys_pkgs)
|
|
self.assert_(isinstance(sys_pkgs, list))
|
|
|
|
path1 = spm.get_merge_protected_paths_mask()
|
|
path2 = spm.get_merge_protected_paths()
|
|
self.assert_(isinstance(path1, list))
|
|
self.assert_(isinstance(path2, list))
|
|
|
|
pkg = spm.convert_from_entropy_package_name("app-foo/foo")
|
|
self.assert_(pkg)
|
|
|
|
def test_portage_xpak(self):
|
|
|
|
spm_class = self.Client.Spm_class()
|
|
if spm_class.PLUGIN_NAME != "portage":
|
|
return
|
|
|
|
sums = {}
|
|
paths = []
|
|
|
|
from entropy.spm.plugins.interfaces.portage_plugin import xpak
|
|
from entropy.spm.plugins.interfaces.portage_plugin import xpaktools
|
|
temp_unpack = tempfile.mkdtemp()
|
|
temp_unpack2 = tempfile.mkdtemp()
|
|
test_pkg = os.path.join(temp_unpack2, "test.pkg")
|
|
dbdir = _misc.get_entrofoo_test_spm_portage_dir()
|
|
|
|
for path in os.listdir(dbdir):
|
|
xpath = os.path.join(dbdir, path)
|
|
paths.append(xpath)
|
|
sums[path] = et.md5sum(xpath)
|
|
|
|
et.compress_files(test_pkg, paths)
|
|
comp_file = xpak.tbz2(test_pkg)
|
|
result = comp_file.recompose(dbdir)
|
|
|
|
shutil.rmtree(temp_unpack)
|
|
os.mkdir(temp_unpack)
|
|
|
|
# now extract xpak
|
|
new_sums = {}
|
|
xpaktools.extract_xpak(test_pkg, tmpdir = temp_unpack)
|
|
for path in os.listdir(temp_unpack):
|
|
xpath = os.path.join(temp_unpack, path)
|
|
new_sums[path] = et.md5sum(xpath)
|
|
|
|
self.assertEqual(sums, new_sums)
|
|
|
|
shutil.rmtree(temp_unpack)
|
|
shutil.rmtree(temp_unpack2)
|
|
|
|
def test_extract_xpak(self):
|
|
|
|
spm_class = self.Client.Spm_class()
|
|
if spm_class.PLUGIN_NAME != "portage":
|
|
return
|
|
|
|
from entropy.spm.plugins.interfaces.portage_plugin import xpaktools
|
|
tmp_path = tempfile.mkdtemp()
|
|
|
|
for test_pkg in self.test_pkgs:
|
|
out_path = xpaktools.extract_xpak(test_pkg, tmp_path)
|
|
self.assertNotEqual(out_path, None)
|
|
self.assert_(os.listdir(out_path))
|
|
|
|
shutil.rmtree(tmp_path, True)
|
|
|
|
|
|
def test_extract_xpak_only(self):
|
|
|
|
spm_class = self.Client.Spm_class()
|
|
if spm_class.PLUGIN_NAME != "portage":
|
|
return
|
|
|
|
from entropy.spm.plugins.interfaces.portage_plugin import xpaktools
|
|
pkg_path = _misc.get_test_xpak_empty_package()
|
|
tmp_path = tempfile.mkdtemp()
|
|
out_path = xpaktools.extract_xpak(pkg_path, tmp_path)
|
|
|
|
self.assertNotEqual(out_path, None)
|
|
self.assert_(os.listdir(out_path))
|
|
|
|
shutil.rmtree(tmp_path, True)
|
|
|
|
if __name__ == '__main__':
|
|
if "--debug" in sys.argv:
|
|
sys.argv.remove("--debug")
|
|
from entropy.const import etpUi
|
|
etpUi['debug'] = True
|
|
unittest.main()
|
|
et.kill_threads()
|
|
raise SystemExit(0)
|