def test_add_and_remove_endpoint(self): context = Context('c') endpoint = EndpointMockup('endpoint') context.add_endpoint(endpoint) self.assert_(endpoint in context.get_endpoints()) self.assert_(endpoint.get_context() is context) context.remove_endpoint(endpoint) self.assert_(not endpoint in context.get_endpoints()) self.assert_(endpoint.get_context() is None)