File: test_zip.py

package info (click to toggle)
python-importlib-metadata 8.7.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 448 kB
  • sloc: python: 2,021; sh: 7; makefile: 3
file content (97 lines) | stat: -rw-r--r-- 2,833 bytes parent folder | download
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
import multiprocessing
import os
import sys
import unittest

from importlib_metadata import (
    FastPath,
    PackageNotFoundError,
    distribution,
    distributions,
    entry_points,
    files,
    version,
)

from . import fixtures


class TestZip(fixtures.ZipFixtures, unittest.TestCase):
    def setUp(self):
        super().setUp()
        self._fixture_on_path('example-21.12-py3-none-any.whl')

    def test_zip_version(self):
        self.assertEqual(version('example'), '21.12')

    def test_zip_version_does_not_match(self):
        with self.assertRaises(PackageNotFoundError):
            version('definitely-not-installed')

    def test_zip_entry_points(self):
        scripts = entry_points(group='console_scripts')
        entry_point = scripts['example']
        self.assertEqual(entry_point.value, 'example:main')
        entry_point = scripts['Example']
        self.assertEqual(entry_point.value, 'example:main')

    def test_missing_metadata(self):
        self.assertIsNone(distribution('example').read_text('does not exist'))

    def test_case_insensitive(self):
        self.assertEqual(version('Example'), '21.12')

    def test_files(self):
        for file in files('example'):
            path = str(file.dist.locate_file(file))
            assert '.whl/' in path, path

    def test_one_distribution(self):
        dists = list(distributions(path=sys.path[:1]))
        assert len(dists) == 1

    @unittest.skipUnless(
        hasattr(os, 'register_at_fork')
        and 'fork' in multiprocessing.get_all_start_methods(),
        'requires fork-based multiprocessing support',
    )
    def test_fastpath_cache_cleared_in_forked_child(self):
        zip_path = sys.path[0]

        FastPath(zip_path)
        assert FastPath.__new__.cache_info().currsize >= 1

        ctx = multiprocessing.get_context('fork')
        parent_conn, child_conn = ctx.Pipe()

        def child(conn, root):
            try:
                before = FastPath.__new__.cache_info().currsize
                FastPath(root)
                after = FastPath.__new__.cache_info().currsize
                conn.send((before, after))
            finally:
                conn.close()

        proc = ctx.Process(target=child, args=(child_conn, zip_path))
        proc.start()
        child_conn.close()
        cache_sizes = parent_conn.recv()
        proc.join()

        self.assertEqual(cache_sizes, (0, 1))


class TestEgg(TestZip):
    def setUp(self):
        super().setUp()
        self._fixture_on_path('example-21.12-py3.6.egg')

    def test_files(self):
        for file in files('example'):
            path = str(file.dist.locate_file(file))
            assert '.egg/' in path, path

    def test_normalized_name(self):
        dist = distribution('example')
        assert dist._normalized_name == 'example'