-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_database.py
40 lines (34 loc) · 1.33 KB
/
test_database.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from database import DataBase
from conftest import scraped_data
from settings import TEST_USER, TEST_URI, TEST_PASSWORD
import time
PAGE_NAME = 'GameSpot'
class TestDataBase:
def test_connection(self):
driver = DataBase(uri=TEST_URI, user=TEST_USER, password=TEST_PASSWORD)
verify = driver.check_connection()
assert verify is None
# create nodes and destroy it
def test_create_nodes(self):
driver = DataBase(uri=TEST_URI, user=TEST_USER, password=TEST_PASSWORD)
driver.add_page(PAGE_NAME)
for post in scraped_data:
driver.add_post(post_text=post['text'], time=post['time'], image=post['photo'])
driver.add_relationship(page_name=PAGE_NAME, time=post['time'])
assert driver.find_page(PAGE_NAME)['page_found'] == PAGE_NAME
time.sleep(4)
query = (
"MATCH (n)"
"DETACH DELETE n"
)
# clean database from the nodes we create in testing
driver.query(query=query)
driver.close()
# check existing nodes
def test_exist_nodes(self):
driver = DataBase(uri=TEST_URI, user=TEST_USER, password=TEST_PASSWORD)
x = driver.find_post(PAGE_NAME)
for post in scraped_data:
if x['time'] == post['time']:
print(x['time'], 'its my dict')
driver.close()