def test_bind_to_graph_unsupported_type(self):
sue = Entity(100, name="Sue")
self.assertRaises(
interfaces.UnknownEntityError,
self.graph.bind_to_graph,
sue
)
def test_set_property_unknown_type(self):
some_entity = Entity("SomeEntity")
self.assertRaises(TypeError,
self.graph.set_property,
some_entity,
name="should_not_be_set")
self.assertDictEqual(
some_entity.properties,
{},
)
def setUp(self):
self.marko = Entity(label="person", name="marko", age=29)
class TestEntityBase(unittest.TestCase):
def setUp(self):
self.marko = Entity(label="person", name="marko", age=29)
def test_is_bound(self):
graph = Graph()
self.marko.graph = graph # mock is bound
self.assertEqual(
self.marko.is_bound(),
True
)
def test_is_bound_not_bound(self):
self.assertEqual(
self.marko.is_bound(),
False
)
def test_remove_property(self):
self.marko.remove_property("name")
self.assertDictEqual(
self.marko.properties,
{
"age": 29,
}
)
def test_remove_property_no_such_key(self):
self.marko.remove_property("surname")
self.assertDictEqual(
self.marko.properties,
{
"name": "marko",
"age": 29,
}
)
def test_set_property_kwargs(self):
self.marko.set_property(surname="McDonald")
self.assertDictEqual(
self.marko.properties,
{
"name": "marko",
"surname": "McDonald",
"age": 29,
}
)
def test_set_property_no_kwargs(self):
self.assertRaises(
EntityUpdateError,
self.marko.set_property,
)
def test_set_property_args_and_kwargs(self):
self.marko.set_property(surname="McDonald", age=30)
self.assertDictEqual(
self.marko.properties,
{
"name": "marko",
"surname": "McDonald",
"age": 30,
}
)
def test_as_dict(self):
self.assertDictEqual(
self.marko.as_dict(),
{
"id": None,
"label": "person",
"properties": {
"name": "marko",
"age": 29
},
"metadata": {},
},
)
def test_as_dict_skip_private_keys(self):
self.marko.set_property(_private_name="Sasquatch")
self.assertDictEqual(
self.marko.as_dict(),
{
"id": None,
"label": "person",
"properties": {
"name": "marko",
"age": 29,
},
"metadata": {},
},
)
def test_as_dict_include_private_keys(self):
self.marko.set_property(_private_name="Sasquatch")
self.assertDictEqual(
self.marko.as_dict(include_privates=True),
{
"id": None,
"label": "person",
"properties": {
"name": "marko",
"age": 29,
"_private_name": "Sasquatch",
},
"metadata": {},
},
)
def test_get_property_as_attribute(self):
self.assertEqual(
self.marko.prop__name,
"marko"
)
def test_get_attribute(self):
self.assertEqual(
self.marko.label,
"person"
)
def test_get_attribute_no_propery_or_attr(self):
self.assertRaises(
AttributeError,
self.marko.__getattribute__,
"prop__bogus"
)
class TestEntityBase(unittest2.TestCase):
def setUp(self):
self.marko = Entity(label="person", name="marko", age=29)
def test_is_bound(self):
graph = Graph()
self.marko.graph = graph # mock is bound
self.assertEqual(
self.marko.is_bound(),
True
)
def test_is_bound_not_bound(self):
self.assertEqual(
self.marko.is_bound(),
False
)
def test_remove_property(self):
self.marko.remove_property("name")
self.assertDictEqual(
self.marko.properties,
{
"age": 29,
}
)
def test_remove_property_no_such_key(self):
self.marko.remove_property("surname")
self.assertDictEqual(
self.marko.properties,
{
"name": "marko",
"age": 29,
}
)
def test_set_property_kwargs(self):
self.marko.set_property(surname="McDonald")
self.assertDictEqual(
self.marko.properties,
{
"name": "marko",
"surname": "McDonald",
"age": 29,
}
)
def test_set_property_no_kwargs(self):
self.assertRaises(
EntityUpdateError,
self.marko.set_property,
)
def test_set_property_args_and_kwargs(self):
self.marko.set_property(surname="McDonald", age=30)
self.assertDictEqual(
self.marko.properties,
{
"name": "marko",
"surname": "McDonald",
"age": 30,
}
)
def test_as_dict(self):
self.assertDictEqual(
self.marko.as_dict(),
{
"id": None,
"label": "person",
"properties": {
"name": "marko",
"age": 29
},
"metadata": {},
},
)
