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()