python-overpy/tests/test_xml.py at master · perhallgren/python-overpy · GitHub

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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import pytest

import overpy

from tests.base_class import BaseTestNodes, BaseTestRelation, BaseTestWay
from tests.base_class import read_file


class TestNodes(BaseTestNodes):
    def test_node01(self):
        api = overpy.Overpass()
        # DOM
        result = api.parse_xml(read_file("xml/node-01.xml"), parser=overpy.XML_PARSER_DOM)
        self._test_node01(result)
        # SAX
        result = api.parse_xml(read_file("xml/node-01.xml"), parser=overpy.XML_PARSER_SAX)
        self._test_node01(result)


class TestRelation(BaseTestRelation):
    def test_relation01(self):
        api = overpy.Overpass()
        # DOM
        result = api.parse_xml(read_file("xml/relation-01.xml"), parser=overpy.XML_PARSER_DOM)
        self._test_relation01(result)
        # SAX
        result = api.parse_xml(read_file("xml/relation-01.xml"), parser=overpy.XML_PARSER_SAX)
        self._test_relation01(result)

    def test_relation02(self):
        api = overpy.Overpass()
        # DOM
        result = api.parse_xml(read_file("xml/relation-02.xml"), parser=overpy.XML_PARSER_DOM)
        self._test_relation02(result)
        # SAX
        result = api.parse_xml(read_file("xml/relation-02.xml"), parser=overpy.XML_PARSER_SAX)
        self._test_relation02(result)


class TestWay(BaseTestWay):
    def test_way01(self):
        api = overpy.Overpass()
        # DOM
        result = api.parse_xml(read_file("xml/way-01.xml"), parser=overpy.XML_PARSER_DOM)
        self._test_way01(result)
        # SAX
        result = api.parse_xml(read_file("xml/way-01.xml"), parser=overpy.XML_PARSER_SAX)
        self._test_way01(result)

    def test_way02(self):
        api = overpy.Overpass()
        # DOM
        result = api.parse_xml(read_file("xml/way-02.xml"), parser=overpy.XML_PARSER_DOM)
        self._test_way02(result)
        # SAX
        result = api.parse_xml(read_file("xml/way-02.xml"), parser=overpy.XML_PARSER_SAX)
        self._test_way02(result)


class TestDataError(object):
    def _get_element_wrong_type(self):
        data = "<foo></foo>"
        import xml.etree.ElementTree as ET
        return ET.fromstring(data)

    def test_element_wrong_type(self):
        with pytest.raises(overpy.exception.ElementDataWrongType):
            overpy.Node.from_xml(
                self._get_element_wrong_type()
            )

        with pytest.raises(overpy.exception.ElementDataWrongType):
            overpy.Relation.from_xml(
                self._get_element_wrong_type()
            )

        with pytest.raises(overpy.exception.ElementDataWrongType):
            overpy.RelationNode.from_xml(
                self._get_element_wrong_type()
            )

        with pytest.raises(overpy.exception.ElementDataWrongType):
            overpy.RelationWay.from_xml(
                self._get_element_wrong_type()
            )

        with pytest.raises(overpy.exception.ElementDataWrongType):
            overpy.Way.from_xml(
                self._get_element_wrong_type()
            )

    def test_node_missing_data(self):
        import xml.etree.ElementTree as ET

        # Tag without k attribute
        data = """<node id="1234"><tag></tag></node>"""
        node = ET.fromstring(data)
        with pytest.raises(ValueError):
            overpy.Node.from_xml(node)

    def test_relation_missing_data(self):
        import xml.etree.ElementTree as ET

        # Tag without k attribute
        data = """<relation id="1234"><tag></tag></relation>"""
        node = ET.fromstring(data)
        with pytest.raises(ValueError):
            overpy.Relation.from_xml(node)

    def test_way_missing_data(self):
        import xml.etree.ElementTree as ET

        # Node without ref attribute
        data = """<way id="1234"><nd></nd></way>"""
        node = ET.fromstring(data)
        with pytest.raises(ValueError):
            overpy.Way.from_xml(node)

        # Tag without k attribute
        data = """<way id="1234"><tag></tag></way>"""
        node = ET.fromstring(data)
        with pytest.raises(ValueError):
            overpy.Way.from_xml(node)