def test_process_attribute_with_empty_local_name(self):
target = ClassFactory.create()
attr = AttrFactory.create()
attr.local_name = None
generator.process_attribute(target, attr, [])
self.assertIsNone(attr.local_name)
attr.local_name = ""
generator.process_attribute(target, attr, [])
self.assertEqual("", attr.local_name)
def test_process_attribute(
self,
mock_attribute_name,
mock_attribute_display_type,
mock_attribute_default,
mock_text_suffix,
):
type_bar = AttrTypeFactory.create(name="bar")
target = ClassFactory.create()
attr = AttrFactory.create(name="foo", types=[type_bar], default="thug")
self.assertIsNone(attr.display_type)
generator.process_attribute(target, attr, ["a", "b"])
self.assertEqual("oof", attr.name)
self.assertEqual("rab", attr.display_type)
self.assertEqual("nope", attr.local_name)
self.assertEqual("life", attr.default)
mock_attribute_name.assert_called_once_with("foo")
mock_attribute_display_type.assert_called_once_with(attr, ["a", "b"])
mock_attribute_default.assert_called_once_with(attr, target.ns_map)
mock_text_suffix.assert_called_once_with("foo")