def test_truncate(self): db = dbFunctions.databaseConnect(host, username, password) dbFunctions.tableCreate(db, tablename, primarykey, primaryType, secondarykey, secondaryType) values = ["var1", "var2", "var3"] dbFunctions.insertData(db, tablename, values) output = dbFunctions.truncate(db, tablename) self.assertIsNone(output) dbFunctions.tableDestroy(db, tablename)
import webScraperFunctions import dbFunctions host = '35.223.188.180' username = '******' password = '******' db = dbFunctions.databaseConnect(host, username, password) esToArtTi = "Es_To_Ar_Ti" arTiToArt = "Ar_Ti_To_Ar" topics = [ "Jobs", "Healthcare", "CJS", "Economy", "Environment", "Education", "Gun Violence", "Immigration", "LGBTQ", "Reproductive Issues" ] dbFunctions.truncate(db, esToArtTi) dbFunctions.truncate(db, arTiToArt) for essential in topics: for x in range(10): title = webScraperFunctions.getTitle(essential, x) if title == "couldn't find essential": print("failed to get article title: " + essential + " " + str(x)) article = webScraperFunctions.getContent( webScraperFunctions.getURL(essential, x)) if article is None or article == "bad url": print("failed to get article data: " + essential + " " + str(x)) print(dbFunctions.insertData(db, esToArtTi, [essential, title])) print(dbFunctions.insertData(db, arTiToArt, [title, article])) db.commit()
def test_insert_fake_table(self): db = dbFunctions.databaseConnect(host, username, password) tablename = "newTable" values = ["var1", "var2", "var3"] output = dbFunctions.insertData(db, tablename, values) self.assertTrue(output == "couldn't insert into table")
def test_destroy_fake(self): db = dbFunctions.databaseConnect(host, username, password) tablename = "newTable" output = dbFunctions.tableDestroy(db, tablename) self.assertTrue(output == "table doesn't exist")
def test_connect_wrong_host(self): username = "******" output = dbFunctions.databaseConnect(host, username, password) self.assertTrue(output == "couldn't connect to database")
def test_connect(self): self.assertIsNotNone( dbFunctions.databaseConnect(host, username, password))