prehistoric-systems/omnivore
1
0
Fork 0
mirror of https://github.com/omnivore-app/omnivore.git synced 2026-03-11 08:54:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add success and failure count in assertions (#318)

Browse source
This commit is contained in:
Hongbo Wu 2022-03-25 12:16:43 +08:00 committed by GitHub
parent e7203bebb5
commit d595d4a8c4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
packages/api/elastic_migrate.py
View file

@ -78,6 +78,8 @@ UPDATE_HIGHLIGHT_SQL = f'''
def assertData(conn, client):
# get all users from postgres
try:
success = 0
failure = 0
cursor = conn.cursor(cursor_factory=RealDictCursor)
cursor.execute('''SELECT id FROM omnivore.user''')
result = cursor.fetchall()
@ -90,11 +92,14 @@ def assertData(conn, client):
index='pages', body={'query': {'term': {'userId': userId}}})['count']
if countInPostgres == countInElastic:
success += 1
print(f'User {userId} OK')
else:
failure += 1
print(
f'User {userId} ERROR: postgres: {countInPostgres}, elastic: {countInElastic}')
cursor.close()
print(f'Asserted data, success: {success}, failure: {failure}')
except Exception as err:
print('Assert data ERROR:', err)
exit(1)