update tests

Author: marshallweekley-eng

apps/bfd-pipeline-idr/test/test_pipeline.py

@@ -631,7 +631,7 @@ def _test_pipeline_load(postgres_db: tuple[PostgresContainer, str], load_type: L
     logger.remove()
 
 
-def _test_load_progress_concurrent(conn: Connection[DictRow]) -> None:
+def _test_load_progress_concurrent(conn: Connection) -> None:
     cur = conn.execute("select * from idr.load_progress where job_id = 1")
     rows = cur.fetchmany(1)
     assert rows[0]["max_run_ts"] is None
@@ -652,7 +652,13 @@ def _test_load_progress_concurrent(conn: Connection[DictRow]) -> None:
     )
     rows = cur.fetchmany(1)
     assert rows_2[0]["max_run_ts"] == rows[0]["last_ts"]
-
+    # test table counts
+    cur = conn.execute("select DISTINCT table_name from idr.load_progress where job_id = 1" \
+                        " EXCEPT " \
+                        "select DISTINCT table_name from idr.load_progress where job_id = 2")
+    rows = cur.fetchmany(1)
+    cur = conn.execute("select count(*) as row_count from idr.load_progress where job_id = 2")
+    assert len(rows) == 0
 
 def test_initial_pipeline_load(postgres_db: tuple[PostgresContainer, str]) -> None:
     _test_pipeline_load(postgres_db, LoadType.INITIAL)