tests.py 文件源码

def test_setup_teardown_sequence_numbers(self):
        if list(self.kwargs.keys()) == ['binary_customisation_script']:
            pytest.skip('Test only applies to chroot hooks.')
        generate_build_config._write_cloud_config(
            open(self.output_file.strpath, 'w'), **self.kwargs)
        cloud_config = yaml.load(self.output_file.open())
        sequence_numbers = {}
        for stanza in cloud_config['write_files']:
            sequence_number = stanza['path'].rsplit('/')[-1].split('-')[0]
            content = base64.b64decode(stanza['content']).decode('utf-8')
            if '-- chroot --' in content:
                sequence_numbers['chroot'] = sequence_number
            elif '-- setup --' in content:
                sequence_numbers['setup'] = sequence_number
            elif '-- teardown --' in content:
                sequence_numbers['teardown'] = sequence_number
        assert sequence_numbers['setup'] < sequence_numbers['chroot']
        assert sequence_numbers['chroot'] < sequence_numbers['teardown']