diff --git a/warden_server/test_warden_server.py b/warden_server/test_warden_server.py index bee3433cf9627751788fb6c779914846b5133398..e76e35645e926c7536c472f87bcd7ea1948dccbe 100755 --- a/warden_server/test_warden_server.py +++ b/warden_server/test_warden_server.py @@ -412,7 +412,7 @@ def init_user(): conn = None try: conn = my.connect(user='root', passwd=getpass.getpass('Enter MySQL Root password:')) - with conn as cur: # Not a canonical connector implementation, for sure + with conn.cursor() as cur: cur.execute("GRANT SELECT, INSERT, UPDATE, CREATE, DELETE, DROP ON *.* TO %s@'localhost' IDENTIFIED BY %s", (USER, PASSWORD)) conn.commit() print("DB User set up successfuly")