8
from mock import Mock,patch
10
sys.path.insert(0,"../")
11
from DistUpgrade.DistUpgradeCache import MyCache
12
from DistUpgrade.DistUpgradeConfigParser import DistUpgradeConfig
14
class TestKernelBaseinstaller(unittest.TestCase):
16
def test_kernel_from_baseinstaller(self):
17
# the upgrade expects this
18
os.chdir("../DistUpgrade")
20
config = DistUpgradeConfig(".")
21
config.set("Files", "LogDir", "/tmp")
22
cache = MyCache(config, None, None, lock=False)
23
cache._has_kernel_headers_installed = Mock()
24
cache._has_kernel_headers_installed.return_value = True
25
cache.getKernelsFromBaseInstaller = Mock()
26
cache.getKernelsFromBaseInstaller.return_value = \
27
["linux-generic2-pae", "linux-generic2"]
28
cache.mark_install = Mock()
29
cache.mark_install.return_value = True
30
cache._selectKernelFromBaseInstaller()
31
#print cache.mark_install.call_args
32
calls = cache.mark_install.call_args_list
33
self.assertEqual(len(calls), 2)
34
cache.mark_install.assert_any_call(
35
"linux-generic2-pae", "Selecting new kernel from base-installer")
36
cache.mark_install.assert_any_call(
37
"linux-headers-generic2-pae", "Selecting new kernel headers from base-installer")
38
if __name__ == "__main__":
40
logging.basicConfig(level=logging.DEBUG)