Nominatim/test/python/test_tools_sp_wiki_loader.py

"""
    Tests for methods of the SPWikiLoader class.
"""
import pytest
from pathlib import Path
from nominatim.tools.special_phrases.sp_wiki_loader import SPWikiLoader

TEST_BASE_DIR = Path(__file__) / '..' / '..'

def test_parse_xml(sp_wiki_loader):
    """
        Test method parse_xml()
        Should return the right SpecialPhrase objects.
    """
    xml = get_test_xml_wiki_content()
    phrases = sp_wiki_loader.parse_xml(xml)
    assert check_phrases_content(phrases)


def test_next(sp_wiki_loader):
    """
        Test objects returned from the next() method.
        It should return all SpecialPhrases objects of
        the 'en' special phrases.
    """
    phrases = next(sp_wiki_loader)
    assert check_phrases_content(phrases)

def check_phrases_content(phrases):
    """
        Asserts that the given phrases list contains
        the right phrases of the 'en' special phrases.
    """
    return  len(phrases) > 1 \
            and any(p.p_label == 'Embassies' and p.p_class == 'amenity' and p.p_type == 'embassy'
                    and p.p_operator == '-' for p in phrases) \
            and any(p.p_label == 'Zip Line' and p.p_class == 'aerialway' and p.p_type == 'zip_line'
                    and p.p_operator == '-' for p in phrases)

@pytest.fixture
def sp_wiki_loader(monkeypatch, def_config):
    """
        Return an instance of SPWikiLoader.
    """
    loader = SPWikiLoader(def_config, ['en'])
    monkeypatch.setattr('nominatim.tools.special_phrases.sp_wiki_loader.SPWikiLoader._get_wiki_content',
                        mock_get_wiki_content)
    return loader

def mock_get_wiki_content(self, lang):
    """
        Mock the _get_wiki_content() method to return
        static xml test file content.
    """
    return get_test_xml_wiki_content()

def get_test_xml_wiki_content():
    """
        return the content of the static xml test file.
    """
    xml_test_content_path = (TEST_BASE_DIR / 'testdata' / 'special_phrases_test_content.txt').resolve()
    with open(xml_test_content_path) as xml_content_reader:
        return xml_content_reader.read()