class ProcessDispatcherSimpleAPIClientTest(PyonTestCase):
def setUp(self):
self.mock_real_client = DotDict()
self.mock_real_client.read_process_definition = Mock()
self.mock_real_client.create_process = Mock()
self.mock_real_client.schedule_process = Mock()
self.mock_real_client.read_process = Mock()
self.mock_eventpub = DotDict()
self.mock_eventpub.publish_event = Mock()
self.mock_container = Mock()
self.client = ProcessDispatcherSimpleAPIClient('fake',
real_client=self.mock_real_client, container=self.mock_container)
self.client.event_pub = self.mock_eventpub
def test_schedule(self):
upid = 'my_pid'
definition_id = 'my_def'
configuration = {'some': 'value'}
self.client.schedule_process(upid, definition_id, configuration=configuration)
self.assertEqual(self.mock_real_client.schedule_process.call_count, 1)
args, kwargs = self.mock_real_client.schedule_process.call_args
self.assertEqual(args[0], definition_id)
self.assertEqual(kwargs['configuration'], configuration)
class ProcessDispatcherSimpleAPIClientTest(PyonTestCase):
def setUp(self):
self.mock_real_client = DotDict()
self.mock_real_client.read_process_definition = Mock()
self.mock_real_client.create_process = Mock()
self.mock_real_client.schedule_process = Mock()
self.mock_real_client.read_process = Mock()
self.mock_eventpub = DotDict()
self.mock_eventpub.publish_event = Mock()
self.mock_container = Mock()
self.client = ProcessDispatcherSimpleAPIClient('fake',
real_client=self.mock_real_client, container=self.mock_container)
self.client.event_pub = self.mock_eventpub
def test_schedule(self):
upid = 'my_pid'
definition_id = 'my_def'
configuration = {'some': 'value'}
self.client.schedule_process(upid, definition_id, configuration=configuration)
self.assertEqual(self.mock_real_client.schedule_process.call_count, 1)
args, kwargs = self.mock_real_client.schedule_process.call_args
self.assertEqual(args[0], definition_id)
self.assertEqual(kwargs['configuration'], configuration)
def setUp(self):
self.mock_real_client = DotDict()
self.mock_real_client.read_process_definition = Mock()
self.mock_real_client.create_process = Mock()
self.mock_real_client.schedule_process = Mock()
self.mock_real_client.read_process = Mock()
self.mock_eventpub = DotDict()
self.mock_eventpub.publish_event = Mock()
self.mock_container = Mock()
self.client = ProcessDispatcherSimpleAPIClient('fake',
real_client=self.mock_real_client, container=self.mock_container)
self.client.event_pub = self.mock_eventpub
def setUp(self):
self.mock_real_client = DotDict()
self.mock_real_client.read_process_definition = Mock()
self.mock_real_client.create_process = Mock()
self.mock_real_client.schedule_process = Mock()
self.mock_real_client.read_process = Mock()
self.mock_eventpub = DotDict()
self.mock_eventpub.publish_event = Mock()
self.mock_container = Mock()
self.client = ProcessDispatcherSimpleAPIClient('fake',
real_client=self.mock_real_client, container=self.mock_container)
self.client.event_pub = self.mock_eventpub
