def manage(self):
self.user = User(username='******')
self.user.set_password('pass')
self.user.set_id()
self.friend = User(username='******')
self.friend.set_password('pass')
self.friend.set_id()
db.session.add(self.user)
db.session.add(self.friend)
db.session.commit()
assert self.user in db.session
assert self.friend in db.session
driver.save_screenshot('test.png')
self.driver.get(self.baseURL)
self.driver.find_element_by_id("name").send_keys("test_user")
self.driver.find_element_by_id("password").send_keys("pass")
self.driver.find_element_by_id("submit").click()
self.driver.get(self.baseURL + "settings")
self.driver.find_element_by_id("new password").send_keys("pas")
self.driver.find_element_by_id("new password confirm").send_keys("pas")
assert self.user.verify_password("pas")
def friend(self):
self.user = User(username='******')
self.user.set_password('pass')
self.user.set_id()
self.friend = User(username='******')
self.friend.set_password('pass')
self.friend.set_id()
db.session.add(self.user)
db.session.add(self.friend)
db.session.commit()
assert self.user in db.session
assert self.friend in db.session
driver.save_screenshot('test.png')
self.driver.get(self.baseURL)
self.driver.find_element_by_id("name").send_keys("test_user")
self.driver.find_element_by_id("password").send_keys("pass")
self.driver.find_element_by_id("submit").click()
self.driver.get(self.baseURL + "user/test_friend")
self.driver.find_element_by_id("follow_button").click()
self.driver.get(self.baseURL + "/logout")
test1 = Friend.query.filter_by(a_id=self.user.id,
b_id=self.friend.id).first()
test2 = Friend.query.filter_by(b_id=self.user.id,
a_id=self.friend.id).first()
assert not test1 or not test2
test = Follow.query.filter_by(requester_id=self.user.id,
requestee_id=self.friend.id).first()
assert test
self.driver.find_element_by_id("name").send_keys("test_friend")
self.driver.find_element_by_id("password").send_keys("pass")
self.driver.find_element_by_id("submit").click()
self.driver.get(self.baseURL + "user/test_user")
self.driver.find_element_by_id("follow_button").click()
self.driver.get(self.baseURL + "/logout")
test1 = Friend.query.filter_by(a_id=self.user.id,
b_id=self.friend.id).first()
test2 = Friend.query.filter_by(b_id=self.user.id,
a_id=self.friend.id).first()
assert test1 or test2
def post(self):
self.user = User(username='******')
self.user.set_password('pass')
self.user.set_id()
db.session.add(self.user)
db.session.commit()
assert self.user in db.session
driver.save_screenshot('test.png')
self.driver.get(self.baseURL)
self.driver.find_element_by_id("name").send_keys("test_user")
self.driver.find_element_by_id("password").send_keys("pass")
self.driver.find_element_by_id("submit").click()
self.driver.find_element_by_id("post_title").send_keys("This is title")
self.driver.find_element_by_id("post_body").send_keys("This is a post")
self.driver.find_element_by_id("post_submit").click()
test = Post.query.filter_by(title="This is a title").first()
assert test
