def set_context_scenario(self, context_scenario_id):
''' Get context slider settings from ETM scenario and set them in the Situation'''
self.context = context.get_context_values(
context_scenario_id,
self.CONTEXT_INPUTS,
self.CONTEXT_QUERIES
)
def test_get_context_values_with_no_inputs_and_no_queries(app, requests_mock):
mock_query_response(12345, app, requests_mock, {})
with app.app_context():
values = context.get_context_values(12345, [], [])
assert 'area_code' in values
assert 'end_year' in values
assert len(values.keys()) == 2
def test_get_context_values_with_just_inputs(app, requests_mock, happy_values):
scenario_id = 12345
inputs = ['input1', 'input2']
mock_inputs_response(scenario_id, app, requests_mock, happy_values)
mock_query_response(scenario_id, app, requests_mock, {})
with app.app_context():
values = context.get_context_values(scenario_id, inputs, [])
assert all((input in values for input in inputs))
def test_get_context_values_with_just_queries(app, requests_mock):
scenario_id = 12345
queries = ['query1', 'query2']
mock_query_response(
scenario_id, app, requests_mock,
{query: {
'future': 1,
'present': 0
}
for query in queries})
with app.app_context():
values = context.get_context_values(scenario_id, [], queries)
assert all((query in values for query in queries))
def test_get_context_values_with_both(app, requests_mock, happy_values):
scenario_id = 12345
queries = ['query1', 'query2']
inputs = ['input1', 'input2']
mock_inputs_response(scenario_id, app, requests_mock, happy_values)
mock_query_response(
scenario_id, app, requests_mock,
{query: {
'future': 1,
'present': 0
}
for query in queries})
with app.app_context():
values = context.get_context_values(scenario_id, inputs, queries)
assert all((input in values for input in inputs))
assert all((query in values for query in queries))