codex-storage/bittorrent-benchmarks
1
0
Fork 0
mirror of https://github.com/codex-storage/bittorrent-benchmarks.git synced 2025-02-20 23:08:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'feat/logstash'

Browse Source
This commit is contained in:
gmega 2025-01-21 19:19:24 -03:00
commit 2e8771fb72
No known key found for this signature in database
GPG Key ID: 6290D34EAD824B18
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
benchmarks/logging/sources/tests/fixtures.py
View File

@ -18,6 +18,9 @@ def _json_data(data: str) -> Dict[str, Any]:
def benchmark_logs_client() -> Elasticsearch: def benchmark_logs_client() -> Elasticsearch:
client = Elasticsearch(os.environ.get("ELASTICSEARCH_URL", "http://localhost:9200")) client = Elasticsearch(os.environ.get("ELASTICSEARCH_URL", "http://localhost:9200"))
# ES may take a while to boot up.
await_predicate(client.ping, timeout=15, polling_interval=0.5)
if client.indices.exists(index="benchmarks-2025.01.21"): if client.indices.exists(index="benchmarks-2025.01.21"):
client.indices.delete(index="benchmarks-2025.01.21") client.indices.delete(index="benchmarks-2025.01.21")
@ -33,6 +36,7 @@ def benchmark_logs_client() -> Elasticsearch:
def _is_indexed() -> bool: def _is_indexed() -> bool:
return client.count(index="benchmarks-2025.01.21")["count"] == len(documents) return client.count(index="benchmarks-2025.01.21")["count"] == len(documents)
# Indexed documents may not be immediately visible, hence the wait.
assert await_predicate( assert await_predicate(
_is_indexed, timeout=10, polling_interval=0.5 _is_indexed, timeout=10, polling_interval=0.5
), "Indexing failed" ), "Indexing failed"