add ?assertEquals around wait_for_pg() calls