[IMP] odoo: tests: allow to compare HtmlElement trees

addTypeEqualityFunc does not work for subclasses. Therefore if we want
to be able to compare html trees we have to add the equality check for
html elements which are children of _Element.
This commit is contained in:
Thibault Delavallée
2017-11-29 10:05:01 +01:00
parent 9cd1cae65f
commit ea6415bd90
+2 -1
View File
@@ -18,7 +18,7 @@ import itertools
import unittest
from contextlib import contextmanager
from datetime import datetime, timedelta
from lxml import etree
from lxml import etree, html
from pprint import pformat
import requests
@@ -93,6 +93,7 @@ class TreeCase(unittest.TestCase):
def __init__(self, methodName='runTest'):
super(TreeCase, self).__init__(methodName)
self.addTypeEqualityFunc(etree._Element, self.assertTreesEqual)
self.addTypeEqualityFunc(html.HtmlElement, self.assertTreesEqual)
def assertTreesEqual(self, n1, n2, msg=None):
self.assertEqual(n1.tag, n2.tag, msg)