def test_find_by_id(self): ''' Test selection of specific data from pages table by id ''' DB.setup() DB.seed() value = self.exec.find_by_id(1) self.assertIsNotNone(value) self.assertEqual(type(value), tuple)
def test_get_url(self): ''' Test selection of specific url by id''' DB.setup() DB.seed() value = self.exec.get_url(1) self.assertIsNotNone(value) self.assertEqual(type(value), tuple)
def test_select_by_id(self): ''' Test selection of specific data from links table by id ''' DB.setup() DB.seed() self.exec.insert(1, 'https://www.google.com/') value = self.exec.select_by_id(1) self.assertIsNotNone(value) self.assertEqual(type(value), tuple)
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
class TestDatabase(TestCase): '''Class to test the database (db) functions''' def setUp(self): self.db = DB() def test_connection(self): '''tests that the connection function does it's work.''' connection = self.db.connect() self.assertIsNotNone(connection) def test_setup(self): '''tests that the setup function does what it was designed to do.''' self.db.setup() self.assertIsNone(self.db.setup()) def test_seed(self): '''tests that the seed function does what it was designed to do.''' self.db.connect() self.db.setup() self.db.seed() self.assertIsNone(self.db.seed()) def test_pages(self): '''tests that the pages function does what it was designed to do.''' self.db.connect() self.db.setup() self.db.seed() selecter = self.db.pages().select() self.assertIsNotNone(selecter) def test_links(self): '''tests that the links function does what it was designed to do.''' self.db.connect() self.db.setup() select_link = self.db.links().select() self.assertIsNotNone(select_link) def TearDown(self): '''the teardown function for all the tests.''' self.db.connect().close()
def test_select(self): ''' Test selection of entire pages table ''' DB.setup() DB.seed() value = self.exec.select() self.assertIsNotNone(value)
def test_delete_by_id(self): ''' Test selection from pages table ''' DB.setup() DB.seed() value = self.exec.delete_by_id(1) self.assertEqual(value, None)
def test_update_by_id(self): ''' Test update is_scraping value by id ''' DB.setup() DB.seed() value = self.exec.update_by_id(False, 1) self.assertEqual(value, None)
def test_seed(self): ''' Test table insertion ''' self.assertEqual(DB.seed(), None)
def test_find_url(self): # Test the find_url method of Pages class in pages.py DB.seed() self.assertIsNotNone(self.exec.find_url(1))
def test_insert(self): ''' Test insert into links table ''' DB.setup() DB.seed() value = self.exec.insert(1, 'https://www.google.com/') self.assertEqual(value, None)
def test_delete_by_page_id(self): ''' Test deletion of specific data in links table by page_id''' DB.setup() DB.seed() value = self.exec.delete_by_page_id(1) self.assertEqual(value, None)
def test_insert(self): DB().setup() DB.seed() self.assertEqual(self.links.insert(2, 'https://www.wikipedia.com'), None)
def test_select(self): # Test the select method of the Links class in links.py DB.setup() DB.seed() self.assertIsNotNone(self.exec.select())
def test_insert(self): # Test the insert method of the Links class in links.py DB.setup() DB.seed() self.assertIsNone(self.exec.insert(1, 'https://facebook.com'))
def test_seed(self): self.assertEqual(DB.seed(), None)
def test_find(self): # Test the find method of Pages class in pages.py DB.seed() result = self.exec.find(2) self.assertIsNotNone(result)
# 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 test_update(self): # Test the update method of Pages class in pages.py DB.seed() self.assertIsNone(self.exec.update(False, 1))
def test_select_urls(self): ''' Test selection of urls from pages table ''' DB.setup() DB.seed() value = self.exec.select_urls() self.assertIsNotNone(value)