Skip to content

Commit 08b4ad8

Browse files
committed
fix: fix parameter name in smoke test and improve logging
1 parent 1c16b3a commit 08b4ad8

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

tests/smoke.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,12 @@
3131
parser.add_argument("sql", help="SQL query to execute")
3232
args = parser.parse_args()
3333

34-
logging.basicConfig(stream=sys.stdout, level=args.debug)
34+
logging.basicConfig(
35+
stream=sys.stdout,
36+
level=args.debug,
37+
format="%(asctime)s.%(msecs)03d %(levelname)s %(name)20s: %(message)s",
38+
datefmt="%Y-%m-%d %H:%M:%S",
39+
)
3540
logging.getLogger("websockets.protocol").setLevel(args.debug)
3641

3742
api_key = None
@@ -58,7 +63,7 @@
5863
api_key=api_key,
5964
runtime=Runtime.SEDONA,
6065
region=Region.AWS_US_WEST_2,
61-
wait_timeout_seconds=900,
66+
wait_timeout=900,
6267
)
6368

6469
with conn_func() as conn:

0 commit comments

Comments
 (0)