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