def test_generate_mms_config_properties_default_workers(env, read_file):
env.return_value.model_server_workers = None
mms_config_properties = model_server._generate_mms_config_properties()
workers = "default_workers_per_model={}".format(None)
read_file.assert_called_once_with(model_server.DEFAULT_MMS_CONFIG_FILE)
assert mms_config_properties.startswith(DEFAULT_CONFIGURATION)
assert workers not in mms_config_properties
def test_generate_mms_config_properties(env, read_file):
model_server_timeout = 'model_server_timeout'
model_server_workers = 'model_server_workers'
http_port = 'http_port'
env.return_value.model_server_timeout = model_server_timeout
env.return_value.model_server_workers = model_server_workers
env.return_value.inference_http_port = http_port
mms_config_properties = model_server._generate_mms_config_properties()
inference_address = 'inference_address=http://0.0.0.0:{}\n'.format(http_port)
server_timeout = 'default_response_timeout={}\n'.format(model_server_timeout)
workers = 'default_workers_per_model={}\n'.format(model_server_workers)
read_file.assert_called_once_with(model_server.DEFAULT_MMS_CONFIG_FILE)
assert mms_config_properties.startswith(DEFAULT_CONFIGURATION)
assert inference_address in mms_config_properties
assert server_timeout in mms_config_properties
assert workers in mms_config_properties