self = <swh.search.tests.test_in_memory.InmemorySearchTest testMethod=test_origin_nb_visit_update_search>
def test_origin_nb_visit_update_search(self):
origin_url = "http://foobar.baz"
self.search.origin_update([{"url": origin_url}])
self.search.flush()
def _update_nb_visit(nb_visit):
self.search.origin_update([{"url": origin_url, "nb_visit": nb_visit}])
self.search.flush()
def _check_nb_visit(nb_visit):
actual_page = self.search.origin_search(
url_pattern=origin_url, nb_visit=nb_visit,
)
assert actual_page.next_page_token is None
results = [r["url"] for r in actual_page.results]
expected_results = [origin_url]
assert sorted(results) == sorted(expected_results)
> _update_nb_visit(5)
.tox/py3/lib/python3.7/site-packages/swh/search/tests/test_search.py:208:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
.tox/py3/lib/python3.7/site-packages/swh/search/tests/test_search.py:196: in _update_nb_visit
self.search.origin_update([{"url": origin_url, "nb_visit": nb_visit}])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
self = <swh.search.in_memory.InMemorySearch object at 0x7f131b408208>
documents = [{'nb_visit': 5, 'url': 'http://foobar.baz'}]
def origin_update(self, documents: Iterable[OriginDict]) -> None:
for source_document in documents:
document: Dict[str, Any] = dict(source_document)
id_ = origin_identifier(document)
if "url" in document:
document["_url_tokens"] = set(
self._url_splitter.split(source_document["url"])
)
if "visit_types" in document:
document["visit_types"] = set(source_document["visit_types"])
if "visit_types" in self._origins[id_]:
document["visit_types"].update(self._origins[id_]["visit_types"])
if "nb_visit" in document:
document["nb_visit"] = max(
> document["nb_visit"], self._origins[id_]["nb_visit"]
)
E KeyError: 'nb_visit'
.tox/py3/lib/python3.7/site-packages/swh/search/in_memory.py:71: KeyError
TEST RESULT
TEST RESULT
- Run At
- Jun 10 2021, 1:41 PM