def test_DBConnection_bad():
badConfig()
dbcm = db._DBConnectionManager()
with pytest.raises(db.BoardDbConfigurationError):
dbcm.get_connection( 'unittest_database' )
def test_get_past_months(): dbcm = db._DBConnectionManager() dbc = dbcm.get_connection( 'unittest_database' ) dao = pronto_statistics_dao.ProntoStatisticDAO() p = dao.get_past_months(7) assert len(p) == 7
def test_get_all_from_date(): dbcm = db._DBConnectionManager() dbc = dbcm.get_connection( 'unittest_database' ) dao = timeline_events_dao.TimelineEventDAO() cutoff_date = shiftByMonths(datetime.datetime(2016, 5,1,12,12,12), -3) t = dao.get_all_from_date(cutoff_date) assert len(t) == 11
def test_DBConnectionManager():
goodConfig()
dbcm = db._DBConnectionManager()
assert 'unittest_database' in dbcm._connections
assert dbcm.get_connection( 'unittest_database' ) is not None
with pytest.raises(db.BoardDbConfigurationError) as e:
dbcm.get_connection( 'NOT_EXISTS' )
def test_DBConnection_good():
goodConfig()
dbcm = db._DBConnectionManager()
dbc = dbcm.get_connection( 'unittest_database' )
assert dbc.conn is not None
with pytest.raises(db.BoardDbConfigurationError):
dbcm.get_connection( 'unittest_baddatabase' )
def test_get_by_type_and_severity():
dbcm = db._DBConnectionManager()
dbc = dbcm.get_connection( 'unittest_database' )
dao = severities_dao.SeverityDAO()
s = dao.get_by_type_and_severity('internal', 'A')
assert s.id == 7
assert s.type == 'internal'
assert s.severity == 'A'
assert s.assignment == 1
assert s.first_response == None
assert s.first_correction == 5
assert s.final_response == None
def test_execute():
goodConfig()
dbcm = db._DBConnectionManager()
dbc = dbcm.get_connection( 'unittest_database' )
sql = "SELECT `id`, `type`, `severity` from severity where id = 7"
cur = dbc.execute( sql )
assert cur is not None
row = cur.fetchone()
assert row == (7L, u'internal', u'A')
sql = "BLABLABLA"
with pytest.raises(db.MySQLdb.ProgrammingError):
cur = dbc.execute( sql )
