29 lines
590 B
Python
29 lines
590 B
Python
import unittest
|
|
import os
|
|
import database
|
|
|
|
|
|
class TestMessageParsing(unittest.TestCase):
|
|
pass
|
|
|
|
|
|
class TestDatabase(unittest.TestCase):
|
|
dbname = 'test.db'
|
|
|
|
def setUp(self):
|
|
if os.path.exists(self.dbname):
|
|
os.remove(self.dbname)
|
|
|
|
def test_database_init(self):
|
|
db = database.DatabaseManager('test.db')
|
|
with db.conn:
|
|
cur = db.conn.execute('SELECT id, name FROM guilds;')
|
|
self.assertTrue(len(cur.fetchall()) == 0)
|
|
|
|
def tearDown(self):
|
|
os.remove(self.dbname)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|