diff --git a/britney.conf b/britney.conf index 940e796..827bd1f 100644 --- a/britney.conf +++ b/britney.conf @@ -62,7 +62,7 @@ SMOOTH_UPDATES = badgers REMOVE_OBSOLETE = no -ADT_ENABLE = no +ADT_ENABLE = yes ADT_DEBUG = no ADT_ARCHES = amd64 i386 diff --git a/tests/__init__.py b/tests/__init__.py index 6370632..81c0316 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -135,6 +135,11 @@ class TestBase(unittest.TestCase): def tearDown(self): del self.data + def restore_config(self, content): + """Helper for restoring configuration contents on cleanup.""" + with open(self.britney_conf, 'w') as fp: + fp.write(content) + def run_britney(self, args=[]): '''Run britney. diff --git a/tests/test_autopkgtest.py b/tests/test_autopkgtest.py index d77fabc..6112098 100644 --- a/tests/test_autopkgtest.py +++ b/tests/test_autopkgtest.py @@ -31,6 +31,16 @@ class TestAutoPkgTest(TestBase): def setUp(self): super(TestAutoPkgTest, self).setUp() + # Mofify configuration according to the test context. + with open(self.britney_conf, 'r') as fp: + original_config = fp.read() + # Disable boottests. + new_config = original_config.replace( + 'BOOTTEST_ENABLE = yes', 'BOOTTEST_ENABLE = no') + with open(self.britney_conf, 'w') as fp: + fp.write(new_config) + self.addCleanup(self.restore_config, original_config) + # fake adt-britney script self.adt_britney = os.path.join( self.data.home, 'auto-package-testing', 'jenkins', 'adt-britney') diff --git a/tests/test_boottest.py b/tests/test_boottest.py index 8a588b9..01772e8 100644 --- a/tests/test_boottest.py +++ b/tests/test_boottest.py @@ -117,18 +117,19 @@ class TestBoottestEnd2End(TestBase): def setUp(self): super(TestBoottestEnd2End, self).setUp() - self.britney_conf = os.path.join( - PROJECT_DIR, 'britney.conf') - old_config = None + + # Modify shared configuration file. with open(self.britney_conf, 'r') as fp: - self.old_config = fp.read() - config = self.old_config.replace( + original_config = fp.read() + # Disable autopkgtests. + new_config = original_config.replace( 'ADT_ENABLE = yes', 'ADT_ENABLE = no') # Disable TouchManifest auto-fetching. - config = config.replace( + new_config = new_config.replace( 'BOOTTEST_FETCH = yes', 'BOOTTEST_FETCH = no') with open(self.britney_conf, 'w') as fp: - fp.write(config) + fp.write(new_config) + self.addCleanup(self.restore_config, original_config) self.data.add('libc6', False, {'Architecture': 'armhf'}), @@ -147,11 +148,6 @@ class TestBoottestEnd2End(TestBase): 'pyqt5:armhf 1.0', ]) - def tearDown(self): - """ Replace the old_config. """ - with open(self.britney_conf, 'w') as fp: - fp.write(self.old_config) - def create_manifest(self, lines): """Create a manifest for this britney run context.""" path = os.path.join(