mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-06-27 02:30:08 +00:00
31 lines
859 B
Python
31 lines
859 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from elasticsearch import Elasticsearch
|
||
|
|
from es_cluster_config import (
|
||
|
|
CLUSTER_URL,
|
||
|
|
INDEX_NAME,
|
||
|
|
PASSWORD,
|
||
|
|
USER,
|
||
|
|
)
|
||
|
|
|
||
|
|
N_ELEMENTS = 1404
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
print(f"Checking contents of index" f"{INDEX_NAME} at {CLUSTER_URL}")
|
||
|
|
|
||
|
|
print("Connecting to the Elasticsearch cluster.")
|
||
|
|
client = Elasticsearch(CLUSTER_URL, basic_auth=(USER, PASSWORD), request_timeout=30)
|
||
|
|
print(client.info())
|
||
|
|
|
||
|
|
count = int(client.cat.count(index=INDEX_NAME, format="json")[0]["count"])
|
||
|
|
try:
|
||
|
|
assert count == N_ELEMENTS
|
||
|
|
except AssertionError:
|
||
|
|
sys.exit(
|
||
|
|
"Elasticsearch dest check failed:"
|
||
|
|
f"got {count} items in index, expected {N_ELEMENTS} items in index."
|
||
|
|
)
|
||
|
|
print(f"Elasticsearch destination test was successful with {count} items being uploaded.")
|