def test_prepare_children(self, input, output, p_output):
model = _search_model.PreparedSearchTextRequestParams()
model.prepare_children(input)
assert model.children == output
assert model.prepared_children == p_output
def test_prepare_city_limit(self, input, output, p_output):
model = _search_model.PreparedSearchTextRequestParams()
model.prepare_city_limit(input)
assert model.city_limit == output
assert model.prepared_city_limit == p_output
def test_generate_params(self, input, output):
model = _search_model.PreparedSearchTextRequestParams()
model.prepare(**input)
for k, v in iteritems(output):
assert model.params[k] == v
assert len(model.params) == len(output)
def test_prepare_others(self):
model = _search_model.PreparedSearchTextRequestParams()
model.prepare_building('building')
model.prepare_floor('floor')
assert model.building == model.prepared_building == 'building'
assert model.floor == model.prepared_floor == 'floor'
def test_prepare_city(self, input, output):
model = _search_model.PreparedSearchTextRequestParams()
model.prepare_city(input)
if input is None:
assert model.city is None
assert model.prepared_city is None
else:
assert model.city == model.prepared_city == output
assert isinstance(model.city, unicode)
assert isinstance(model.prepared_city, unicode)
def test_prepare_ok(self, mocker):
model = _search_model.PreparedSearchTextRequestParams()
model.prepare_keywords = lambda x: 'keywords'
model.prepare_location = lambda x: 'location'
model.prepare_types = lambda x: 'types'
model.prepare_city = lambda x: 'city'
model.prepare_city_limit = lambda x: 'city_limit'
model.prepare_children = lambda x: 'children'
model.prepare_offset = lambda x: 'offset'
model.prepare_page = lambda x: 'page'
model.prepare_building = lambda x: 'building'
model.prepare_floor = lambda x: 'floor'
model.prepare_sort_rule = lambda x: 'sort_rule'
model.prepare_extension = lambda x: 'extensions'
for i in [
'prepare_keywords', 'prepare_types', 'prepare_city',
'prepare_city_limit', 'prepare_children', 'prepare_offset',
'prepare_page', 'prepare_building', 'prepare_floor',
'prepare_sort_rule', 'prepare_extension', 'prepare_location'
]:
mocker.spy(model, i)
model.prepare('keywords', 'types', 'city', True, True, 1, 1,
'building', 'floor', 'all', True, '12,34')
model.prepare_keywords.assert_called_once_with('keywords')
model.prepare_types.assert_called_once_with('types')
model.prepare_city.assert_called_once_with('city')
model.prepare_city_limit.assert_called_once_with(True)
model.prepare_children.assert_called_once_with(True)
model.prepare_offset.assert_called_once_with(1)
model.prepare_page.assert_called_once_with(1)
model.prepare_building.assert_called_once_with('building')
model.prepare_floor.assert_called_once_with('floor')
model.prepare_sort_rule.assert_called_once_with('all')
model.prepare_extension.assert_called_once_with(True)
model.prepare_location.assert_called_once_with('12,34')
def test_init_ok(self):
model = _search_model.PreparedSearchTextRequestParams()
assert model.key is None