class TestDb(unittest.TestCase):
'''class that tests db class in _init_.py'''
def setUp(self):
'''function that sets up for testing '''
self.db = DB()
def test_connect(self):
'''function that tests the connect function'''
connection_object = self.db.connect()
self.assertIsNotNone(connection_object)
def test_new_connect(self):
'''function that tests the new_connect function'''
connection_object = self.db.new_connect()
self.assertIsNotNone(connection_object)
def test_setup(self):
'''function that tests the setup function'''
self.assertEqual(self.db.setup(), None)
cursor = self.db.new_connect().cursor()
query = cursor.execute('SELECT url FROM pages WHERE id=1 ')
self.assertEqual(query, None)
def test_seed(self):
'''function that tests the seed function'''
self.db.setup()
seed = self.db.seed()
self.assertIsNone(seed)
def tearDown(self):
self.db = None
# Show examples of how you would use ALL your implementations here
from src.db import DB
from src.spider import spider_scrap
from celery import Celery
from decouple import config
#
db = DB()
db.connect()
db.new_connect()
db.setup()
db.seed()
dd = DB.new_connect()
pages = DB.pages()
# pages.fetch_url(2)
print(pages.fetch_url(2))
print(pages.select())
print(pages.find(2))
# print(pages.update_id(1))
links = DB.links()
print(links.insert(1, 'www.goggle.com'))
print(links.delete(1))
print(links.select(1))
# #
# app = Celery('main', broker=config('CELERY_BROKER'), backend=config('CELERY_BACKEND'))
#
#
# @app.task
# def scrap_url():
# return spider_scrap(1)
# spider_scrap(1)
def setUp(self) -> None:
self.links = Links(DB.new_connect())
def setUp(self) -> None:
self.pages = Pages(DB.new_connect())
