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