| 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 |