def test_output_service_loop():
mock_output_service = MockOutputService()
# join and block, waiting for event loop to execute once
# and then exits. Mock manually controls the event loop.
gevent.joinall([mock_output_service.start()])
assert mock_output_service.event_loop_ack is True
assert mock_output_service.get_state() == BaseStates.Idle
def test_output_service_write():
# create ServiceManager and add services
service_manager = ServiceManager("service-manager")
# add worker
worker_name = "test-worker-1"
test_worker_1 = TestWorker(worker_name)
service_manager.add_service(test_worker_1, worker_name)
# add output service
mock_output_service = MockOutputService()
service_manager.add_service(mock_output_service, "mock-output-service")
assert test_worker_1.get_state() == BaseStates.Idle
assert mock_output_service.get_state() == BaseStates.Idle
# join and block, waiting for event loop to execute once
# and then exits. Mock manually controls the event loop.
# Manually starting services rather than using service_manager
gevent.joinall([test_worker_1.start(), mock_output_service.start()])
# Assert that state is started. State will not be stopped
# or idle because the services were manually called. Except
# for the Mock service which will be idle. See the code.
assert test_worker_1.get_state() == BaseStates.Started
assert mock_output_service.get_state() == BaseStates.Idle
# the services ran and I expect the test worker to have queried
# the directory proxy and found the output service, writing to it.
result = mock_output_service.get()
assert result is not None
assert result == "test-worker-work-result"
assert mock_output_service.size() == 0
