Skip to content

Commit

Permalink
Fix e2e tests for latest mkdocs version
Browse files Browse the repository at this point in the history
  • Loading branch information
lukasgeiter committed Dec 2, 2021
1 parent 5fde953 commit 821f997
Showing 1 changed file with 18 additions and 14 deletions.
32 changes: 18 additions & 14 deletions mkdocs_awesome_pages_plugin/tests/e2e/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,12 @@
from typing import Optional, List, Tuple, Union, Dict
from unittest import TestCase

import pkg_resources
import yaml
from bs4 import BeautifulSoup
from importlib_metadata import EntryPoint
from mkdocs import plugins
from mkdocs.commands.build import build
from mkdocs.config import load_config

from ...utils import cd

Expand Down Expand Up @@ -103,11 +106,7 @@ def _createFiles(self, directory: str, files: List[Union[str, Tuple[str, Union[s
self._createFiles(path, contents)

def _mkdocsBuild(self, **options):
# register project with pkg_resources so mkdocs picks it up as a plugin (before mkdocs module import!)
self._registerPluginDist()

from mkdocs.commands.build import build
from mkdocs.config import load_config
self._patchGetPlugins()

with warnings.catch_warnings():
# ignore deprecation warnings within mkdocs
Expand Down Expand Up @@ -147,14 +146,19 @@ def _parseNav(self, ul: BeautifulSoup):
return pages

@staticmethod
def _registerPluginDist():
distribution = pkg_resources.Distribution(__file__)
entry_point = pkg_resources.EntryPoint.parse(
'awesome-pages = mkdocs_awesome_pages_plugin.plugin:AwesomePagesPlugin',
dist=distribution
)
distribution._ep_map = {'mkdocs.plugins': {'awesome-pages': entry_point}}
pkg_resources.working_set.add(distribution)
def _patchGetPlugins():
_original_get_plugins = plugins.get_plugins

def _patched_get_plugins():
result = _original_get_plugins()
result['awesome-pages'] = EntryPoint(
name='awesome-pages',
value='mkdocs_awesome_pages_plugin.plugin:AwesomePagesPlugin',
group='mkdocs.plugins'
)
return result

plugins.get_plugins = _patched_get_plugins

@staticmethod
def _writeToFile(path: str, content: str):
Expand Down

0 comments on commit 821f997

Please sign in to comment.