class TestBoardBot(unittest.TestCase):
def setUp(self):
self.database = "test.db"
# Init database
self.conn = sqlite3.connect(self.database)
self.cursor = self.conn.cursor()
self.cursor.execute("CREATE TABLE board (title TEXT, author TEXT, time INT, url TEXT)")
self.conn.commit()
self.conn.close()
self.bot = BoardBot(self.database)
def tearDown(self):
self.conn.close()
os.remove(self.database)
def test_retrieve_posts(self):
self.bot.retrieve_posts()
self.assertEqual(25, len(self.bot.posts))
def test_save_posts(self):
self.bot.save_posts()
self.conn = sqlite3.connect(self.database)
self.cursor = self.conn.cursor()
self.cursor.execute("SELECT * FROM board")
self.assertEqual(25, len(self.cursor.fetchall()))
