def bragi_make_params_with_shape_scope_test():
"""
test of generate params with instance which includes a poi dataset
"""
instance = mock.MagicMock()
instance.name = 'bob'
instance.poi_dataset = 'priv.bob'
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20, "request_id": "1234", "shape_scope[]": ["poi", "street"]}
params = bragi.make_params(request=request, instances=[instance], timeout=1)
rsp = [
('q', 'aa'),
('limit', 20),
('pt_dataset[]', 'bob'),
('poi_dataset[]', 'priv.bob'),
('timeout', 1000),
("request_id", "1234"),
('shape_scope[]', 'poi'),
('shape_scope[]', 'street'),
]
params.sort()
rsp.sort()
assert rsp == params
def bragi_make_params_with_multiple_instances_test():
"""
test of generate params with instance
"""
instance1 = mock.MagicMock()
instance1.name = 'bib'
instance1.poi_dataset = None
instance2 = mock.MagicMock()
instance2.name = 'bob'
instance2.poi_dataset = None
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20, "request_id": "1234"}
params = bragi.make_params(request=request, instances=[instance1, instance2], timeout=1)
rsp = [
('q', 'aa'),
('limit', 20),
('pt_dataset[]', 'bib'),
('pt_dataset[]', 'bob'),
('timeout', 1000),
("request_id", "1234"),
('shape_scope[]', 'addr'),
('shape_scope[]', 'admin'),
('shape_scope[]', 'poi'),
('shape_scope[]', 'street'),
]
params.sort()
rsp.sort()
assert params == rsp
def bragi_make_params_with_places_proximity_radius_test():
"""
test of generate params with instance & places_proximity_radius
"""
instance = mock.MagicMock()
instance.name = 'bib'
instance.poi_dataset = None
bragi = GeocodeJson(host='http://bob.com/autocomplete', timeout=1)
request = {
"q": "aa",
"count": 20,
"request_id": "1234",
"places_proximity_radius": 5000
}
params = bragi.make_params(request=request, instances=[instance])
rsp = [
('q', 'aa'),
('limit', 20),
('pt_dataset[]', 'bib'),
('timeout', 1000),
("request_id", "1234"),
('shape_scope[]', 'addr'),
('shape_scope[]', 'admin'),
('shape_scope[]', 'poi'),
('shape_scope[]', 'street'),
('proximity_scale', 32.5),
('proximity_offset', 5.0),
('proximity_decay', 0.4),
]
params.sort()
rsp.sort()
assert rsp == params
def bragi_make_params_without_instance_test():
"""
test of generate params without instance
"""
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20}
params = bragi.make_params(request=request, instances=[], timeout=0.1)
rsp = [('q', 'aa'), ('limit', 20), ('timeout', 100)]
params.sort()
rsp.sort()
assert rsp == params
def bragi_make_params_without_instance_test():
"""
test of generate params without instance
"""
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20}
params = bragi.make_params(request=request, instances=[], timeout=0.1)
rsp = {'q': 'aa', 'limit': 20, 'timeout': 100}
assert len(list(rsp.keys())) == len(list(params.keys()))
for key, value in rsp.items():
assert key in params
assert value == params[key]
def bragi_make_params_with_instance_test():
"""
test of generate params with instance
"""
instance = mock.MagicMock()
instance.name = 'bib'
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20}
params = bragi.make_params(request=request, instances=[instance])
rsp = {'q': 'aa', 'limit': 20, 'pt_dataset': ['bib']}
len(list(rsp.keys())) == len(list(params.keys()))
for key, value in rsp.items():
assert key in params
assert value == params[key]
def bragi_make_params_with_instance_test():
"""
test of generate params with instance
"""
instance = mock.MagicMock()
instance.name = 'bib'
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20}
params = bragi.make_params(request=request,
instances=[instance],
timeout=1)
rsp = [('q', 'aa'), ('limit', 20), ('pt_dataset[]', 'bib'),
('timeout', 1000)]
params.sort()
rsp.sort()
assert rsp == params
def bragi_make_params_without_instance_test():
"""
test of generate params without instance
"""
bragi = GeocodeJson(host='http://bob.com/autocomplete')
request = {"q": "aa", "count": 20, "request_id": "1234"}
params = bragi.make_params(request=request, instances=[], timeout=0.1)
rsp = [
('q', 'aa'),
('limit', 20),
('timeout', 100),
("request_id", "1234"),
('shape_scope[]', 'addr'),
('shape_scope[]', 'admin'),
('shape_scope[]', 'poi'),
('shape_scope[]', 'street'),
]
params.sort()
rsp.sort()
assert rsp == params