diff --git a/python/src/AjouSlackProducerMySQL.py b/python/src/AjouSlackProducerMySQL.py index 8e413fa85c9d21ed57f9d599fe78aadcc1b257a0..f65b66e129c8523f173afd60c9245b6feae92c12 100644 --- a/python/src/AjouSlackProducerMySQL.py +++ b/python/src/AjouSlackProducerMySQL.py @@ -2,8 +2,6 @@ import datetime import json import os import time -from contextlib import contextmanager -from pathlib import Path import mysql.connector import requests @@ -22,23 +20,6 @@ def acked(err, msg): print("Message produced: {0}...".format(msg.value())) -@contextmanager -def OPEN_DB(): - # connect to my local MySQL instance using connection string - db = mysql.connector.connect( - host="localhost", - user=os.environ["MYSQL_USER"], - password=os.environ["MYSQL_PASSWORD"], - database="ajou_notices", - charset="utf8", - ) - cursor = db.cursor(buffered=True) - yield cursor - db.commit() # always commits - cursor.close() - db.close() - - def checkOldness(db): pass diff --git a/python/tests/test_mysql.py b/python/tests/test_mysql.py index 633858ddb72dfc55901661cd0d6644f5e4195626..0e531c2c7e81f7e80c22246ece068c49e0af340e 100644 --- a/python/tests/test_mysql.py +++ b/python/tests/test_mysql.py @@ -110,6 +110,17 @@ def test_updateDB(): assert row is not None +def test_printDB(): + with OPEN_DB() as cursor: + cursor.execute( + "SELECT * FROM notices ORDER BY id" + ) # query = SELECT * FROM testdb.notices LIMIT 3; + + for row in cursor: + print(row) + assert row is not None + + def test_lastparsed(): """Test LAST_PARSED update""" from datetime import datetime