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
|
import os
import pytest
import pkg_resources
# A mock setuptools dist object.
class MockDist():
def __init__(self, datafiles, module_name):
self.datafiles = datafiles
self.module_name = module_name
def get_resource_filename(self, *args, **kwargs):
return os.path.join(self.datafiles.dirname,
self.datafiles.basename,
self.module_name)
# A mock setuptools entry object.
class MockEntry():
def __init__(self, datafiles, module_name):
self.dist = MockDist(datafiles, module_name)
self.module_name = module_name
# Patch setuptools.get_entry_info
#
# Use result = entry_fixture(datafiles, entry_point, lookup_string) to
# patch setuptools for external plugin loading.
#
@pytest.fixture()
def entry_fixture(monkeypatch):
def patch(datafiles, entry_point, lookup_string):
dist, package = lookup_string.split(':')
def mock_entry(pdist, pentry_point, ppackage):
assert(pdist == dist)
assert(pentry_point == entry_point)
assert(ppackage == package)
return MockEntry(datafiles, package)
monkeypatch.setattr(pkg_resources, 'get_entry_info', mock_entry)
return patch
|
