我正在为一些lxml自定义元素(例如ParentElement、ChildElement(实现测试，这些元素是通过来自自定义查找类(ModelLookup(的装饰器注册的。
pytest用于运行测试，而我使用的是conftest.py中定义的fixture。
问题是，当自定义元素的对象被创建为fixture的一部分时，测试函数上的类型会丢失，我会得到以下错误：
> assert ['eclios', 'ruby'] == sorted([e.name for e in simple_tidy_family.child_element])
E AttributeError: 'lxml.etree._Element' object has no attribute 'name'
另一方面，如果对象是作为测试函数的一部分创建的，那么所有对象都可以正常运行。
理想情况下，我希望在其他测试中重复使用这些固定装置，因此首先在conftest中定义它。
模块代码：
from __future__ import unicode_literals
from lxml import etree
class ModelLookup(etree.PythonElementClassLookup):
_lookup = {}
@classmethod
def register_node_handler_class(cls, handler_cls):
if handler_cls.cls_tag not in cls._lookup.keys():
cls._lookup[handler_cls.cls_tag] = handler_cls
return handler_cls
def lookup(self, doc, node):
if node.tag in self._lookup.keys():
print(node.tag)
return self._lookup[node.tag]
return etree.ElementBase
@ModelLookup.register_node_handler_class
class ParentElement(etree.ElementBase):
cls_tag = 'ParentElement'
@staticmethod
def tada():
return 'tada'
@property
def child_element(self):
return self.xpath('./ChildElement')
@child_element.setter
def child_element(self, value):
self.append(value)
@property
def name(self):
return self.get('name')
@name.setter
def name(self, value):
self.set('name', value)
@ModelLookup.register_node_handler_class
class ChildElement(etree.ElementBase):
cls_tag = 'ChildElement'
@property
def name(self):
return self.get('name')
@name.setter
def name(self, value):
self.set('name', value)
conftest.py
from __future__ import unicode_literals
import pytest
import xmlpal.xmlpal as xpal
import logging
@pytest.fixture()
def simple_family():
ruby = xpal.ChildElement(**{'name': 'ruby'})
eclios = xpal.ChildElement(**{'name': 'eclios'})
adam = xpal.ParentElement()
adam.append(ruby)
adam.append(eclios)
logging.info(f'my name is {ruby.name}')
return adam
测试代码：
from __future__ import unicode_literals
from lxml import etree
import xmlpal.xmlpal as xpal
def test_fixture_family(simple_family):
assert 'tada' == simple_family.tada()
assert isinstance(simple_family, xpal.ParentElement)
assert 2 == len(simple_family.child_element)
assert 2 == len([e for e in simple_family.child_element if isinstance(e, etree._Element)])
assert ['eclios', 'ruby'] == sorted([e.name for e in simple_family.child_element])
assert 2 == len([e for e in simple_family.child_element if isinstance(e, xpal.ChildElement)])
def test_local_family():
ruby = xpal.ChildElement(**{'name': 'ruby'})
eclios = xpal.ChildElement(**{'name': 'eclios'})
adam = xpal.ParentElement()
adam.append(ruby)
adam.append(eclios)
assert 'tada' == adam.tada()
assert isinstance(adam, xpal.ParentElement)
assert 2 == len(adam.child_element)
assert 2 == len([e for e in adam.child_element if isinstance(e, etree._Element)])
assert ['eclios', 'ruby'] == sorted([e.name for e in adam.child_element])
assert 2 == len([e for e in adam.child_element if isinstance(e, xpal.ChildElement)])
感谢您的帮助！
不能使用这样的自定义查找。查找仅在从源解析XML文档树时应用。因此，您根本没有在测试中调用自定义查找，也没有注册自定义元素类型。您遇到的其余问题只是两个函数调用之间的垃圾回收(产生效果是因为您没有退出fixture函数的作用域(。然而，当在解析文档时应用查找时，一切都如预期的那样工作：
@pytest.fixture
def source():
"""Construct the tree and serialize it to string."""
ruby = xpal.ChildElement(**{'name': 'ruby'})
eclios = xpal.ChildElement(**{'name': 'eclios'})
adam = xpal.ParentElement()
adam.append(ruby)
adam.append(eclios)
return etree.tostring(adam)
@pytest.fixture
def tree(source):
"""Parse the tree from a string using a parser with custom lookup registered."""
parser = etree.XMLParser()
parser.set_element_class_lookup(xpal.ModelLookup())
return etree.fromstring(source, parser)
def test_fixture_family(tree):
adam = tree
assert 'tada' == adam.tada()
...
这不是pytest fixture的问题。当simple_family是一个正常函数时，也会发生同样的错误。
哇，看起来你有内存管理问题：哦。如果从simple_familyreturn adam, eclios, ruby或使用yield返回所有对象，则一切正常。如果不使用yield:
ruby is not simple_family.child_element[0]
看起来图书馆里有个严重的错误。
编辑：如果ruby和eclios是全局变量，它也可以正常工作：D
我找到了一个有效的解决方案，但不太明白它为什么有效。
如果我将夹具中的return替换为yield，那么一切都会按预期进行。
@pytest.fixture()
def simple_tidy_family():
ruby = xpal.ChildElement()
ruby.name = 'ruby'
eclios = xpal.ChildElement()
eclios.name = 'eclios'
adam = xpal.ParentElement()
adam.append(ruby)
adam.append(eclios)
yield adam