class HAAgentProcessControlMockTest(PyonTestCase):
def setUp(self):
pd_name = "fakepd"
resource_registry = "fakerr"
service_id = "fakeservice"
self.control = HAProcessControl(pd_name, resource_registry, service_id)
def test_broken_callback(self):
event = Mock()
def raises_socket_error(e):
# Not derived from Exception
import socket
raise socket.error("err")
def raises_exception(e):
raise Exception("something")
def raises_systemexit(e):
raise SystemExit("something")
self.control._inner_event_callback = raises_socket_error
self.control._event_callback(event)
self.control._inner_event_callback = raises_exception
self.control._event_callback(event)
self.control._inner_event_callback = raises_systemexit
self.assertRaises(SystemExit, self.control._event_callback, event)
class HAAgentProcessControlMockTest(PyonTestCase):
def setUp(self):
pd_name = "fakepd"
resource_registry = "fakerr"
service_id = "fakeservice"
self.control = HAProcessControl(pd_name, resource_registry, service_id)
def test_broken_callback(self):
event = Mock()
def raises_socket_error(e):
# Not derived from Exception
import socket
raise socket.error("err")
def raises_exception(e):
raise Exception("something")
def raises_systemexit(e):
raise SystemExit("something")
self.control._inner_event_callback = raises_socket_error
self.control._event_callback(event)
self.control._inner_event_callback = raises_exception
self.control._event_callback(event)
self.control._inner_event_callback = raises_systemexit
self.assertRaises(SystemExit, self.control._event_callback, event)
def setUp(self):
pd_name = "fakepd"
resource_registry = "fakerr"
service_id = "fakeservice"
self.control = HAProcessControl(pd_name, resource_registry, service_id)
def setUp(self):
pd_name = "fakepd"
resource_registry = "fakerr"
service_id = "fakeservice"
self.control = HAProcessControl(pd_name, resource_registry, service_id)