1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Tests the config file handling."""
import logging
import os
import time
import unittest
# Import gobject introspection to tigger the use of the correct client code
from gi.repository import GObject
import defer
import dbus
import aptdaemon.client
import aptdaemon.loop
import aptdaemon.enums
import aptdaemon.test
DEBUG = True
REPO_PATH = os.path.join(aptdaemon.test.get_tests_dir(), "repo")
class ConfigFilePromptTestCase(aptdaemon.test.AptDaemonTestCase):
"""Test the replacement of config files."""
def setUp(self):
"""Setup a chroot, run the aptdaemon and a fake PolicyKit daemon."""
# Setup chroot
self.chroot = aptdaemon.test.Chroot()
self.chroot.setup()
self.addCleanup(self.chroot.remove)
self.chroot.add_test_repository()
# Start aptdaemon with the chroot on the session bus
self.start_dbus_daemon()
self.bus = dbus.bus.BusConnection(self.dbus_address)
self.start_session_aptd(self.chroot.path)
# Start the fake PolikcyKit daemon
self.start_fake_polkitd()
time.sleep(1)
self.called = False
# Create a fake config file which gets overwritten by silly-config
self.config_path = os.path.join(self.chroot.path,
"etc/silly-packages.cfg")
with open(self.config_path, "w") as config:
config.write("BliBlaBlub")
@defer.inline_callbacks
def _on_config_file_conflict(self, trans, config_old, config_new, answer):
self.assertEqual(trans.config_file_conflict, (config_old, config_new))
if answer == "urgs":
# Check if cancelling is forbidden
try:
yield trans.cancel()
except aptdaemon.errors.AptDaemonError as error:
self.assertTrue(str(error),
"org.debian.apt: Could not cancel transaction")
# Check if we fail correctly on wrong answers
try:
yield trans.resolve_config_file_conflict(config_old,
"a&&dasmk")
except aptdaemon.errors.AptDaemonError as error:
self.assertTrue(str(error).index("Invalid value"))
yield trans.resolve_config_file_conflict(config_old, "replace")
else:
self.fail("Failed to detect invalid answer")
else:
yield trans.resolve_config_file_conflict(config_old, answer)
def _on_finished(self, trans, exit):
"""Callback to stop the mainloop after a transaction is done."""
aptdaemon.loop.mainloop.quit()
def test_keep(self):
"""Test keeping the current configuration file."""
@defer.inline_callbacks
def run():
self.client = aptdaemon.client.AptClient(self.bus)
trans = yield self.client.install_packages(["silly-config"])
trans.connect("finished", self._on_finished)
trans.connect("config-file-conflict",
self._on_config_file_conflict,
"keep")
yield trans.run()
defer.return_value(trans)
deferred = run()
aptdaemon.loop.mainloop.run()
self.assertEqual(deferred.result.exit, aptdaemon.enums.EXIT_SUCCESS)
with open(self.config_path) as config:
self.assertEqual(config.read(),
"BliBlaBlub")
with open("%s.dpkg-dist" % self.config_path) as config_dist:
self.assertEqual(config_dist.read(),
"#Just another config file.\n")
def test_replace(self):
"""Test replacing the current configuration file."""
@defer.inline_callbacks
def run():
self.client = aptdaemon.client.AptClient(self.bus)
trans = yield self.client.install_packages(["silly-config"])
trans.connect("finished", self._on_finished)
trans.connect("config-file-conflict",
self._on_config_file_conflict,
"replace")
yield trans.run()
defer.return_value(trans)
deferred = run()
aptdaemon.loop.mainloop.run()
self.assertEqual(deferred.result.exit, aptdaemon.enums.EXIT_SUCCESS)
with open(self.config_path) as config:
self.assertEqual(config.read(),
"#Just another config file.\n")
with open("%s.dpkg-old" % self.config_path) as config_old:
self.assertEqual(config_old.read(),
"BliBlaBlub")
def test_fail(self):
"""Test failing correctly."""
@defer.inline_callbacks
def run():
self.client = aptdaemon.client.AptClient(self.bus)
trans = yield self.client.install_packages(["silly-config"])
yield trans.set_locale("C")
trans.connect("finished", self._on_finished)
trans.connect("config-file-conflict",
self._on_config_file_conflict,
"urgs")
yield trans.run()
defer.return_value(trans)
deferred = run()
aptdaemon.loop.mainloop.run()
self.assertEqual(deferred.result.exit, aptdaemon.enums.EXIT_SUCCESS)
with open(self.config_path) as config:
self.assertEqual(config.read(),
"#Just another config file.\n")
with open("%s.dpkg-old" % self.config_path) as config_old:
self.assertEqual(config_old.read(),
"BliBlaBlub")
if __name__ == "__main__":
if DEBUG:
logging.basicConfig(level=logging.DEBUG)
unittest.main()
# vim: ts=4 et sts=4
|
