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
|
#-----------------------------------------------------------------------------
# Copyright (c) 2005-2023, PyInstaller Development Team.
#
# Distributed under the terms of the GNU General Public License (version 2
# or later) with exception for distributing the bootloader.
#
# The full license is in the file COPYING.txt, distributed with this software.
#
# SPDX-License-Identifier: (GPL-2.0-or-later WITH Bootloader-exception)
#-----------------------------------------------------------------------------
# This contains tests for the class:``Tree``, see
# https://pyinstaller.readthedocs.io/en/latest/advanced-topics.html#the-tree-class
import os
import pytest
import PyInstaller.building.datastruct
class Tree(PyInstaller.building.datastruct.Tree):
# A stripped-down version of PyInstaller.building.datastruct.Tree that does not check the guts,
# but only the `assemble()` step.
def __postinit__(self):
self.assemble()
TEST_MOD = 'Tree_files'
_DATA_BASEPATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), TEST_MOD)
_TEST_FILES = sorted([
os.path.join('subpkg', 'twelve.py'),
os.path.join('subpkg', 'thirteen.txt'),
os.path.join('subpkg', 'init__.py'),
'two.py',
'dynamiclib.dylib',
os.path.join('py_files_not_in_package', 'sub_pkg', 'three.py'),
os.path.join('py_files_not_in_package', 'sub_pkg', 'init__.py'),
os.path.join('py_files_not_in_package', 'one.py'),
os.path.join('py_files_not_in_package', 'data', 'eleven.dat'),
os.path.join('py_files_not_in_package', 'ten.dat'),
'dynamiclib.dll',
'pyextension.pyd',
'nine.dat',
'init__.py',
'pyextension.so',
])
_PARAMETERS = (
(None, None, _TEST_FILES),
('abc', None, [os.path.join('abc', f) for f in _TEST_FILES]),
(None, ['*.py'], [f for f in _TEST_FILES if not f.endswith('.py')]),
(None, ['*.py', '*.pyd'], [f for f in _TEST_FILES if not f.endswith(('.py', '.pyd'))]),
(None, ['subpkg'], [f for f in _TEST_FILES if not f.startswith('subpkg')]),
(None, ['subpkg', 'sub_pkg'],
[f for f in _TEST_FILES if not (f.startswith('subpkg') or os.sep + 'sub_pkg' + os.sep in f)]),
('klm', ['subpkg', 'sub_pkg', '*.py', '*.pyd'], [
os.path.join('klm', f) for f in _TEST_FILES
if not (f.startswith('subpkg') or os.sep + 'sub_pkg' + os.sep in f or f.endswith(('.py', '.pyd')))
]),
) # yapf: disable
@pytest.mark.parametrize("prefix,excludes,result", _PARAMETERS)
def test_Tree(monkeypatch, prefix, excludes, result):
monkeypatch.setattr('PyInstaller.config.CONF', {'workpath': '.'})
tree = Tree(_DATA_BASEPATH, prefix=prefix, excludes=excludes)
files = sorted(f[0] for f in tree)
assert files == sorted(result)
|
