blankie
/
nitter
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improve test base

This commit is contained in:
Zed 2019-06-29 00:41:57 +02:00
parent ee1d6f7cf0
commit a5676e1c2c
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/base.py
View File

@ -3,19 +3,21 @@ from seleniumbase import BaseCase
class Tweet(object): class Tweet(object):
def __init__(self, tweet=''): def __init__(self, tweet=''):
namerow = tweet + 'div.media-heading > div > .fullname-and-username > ' namerow = tweet + 'div.media-heading '
self.fullname = namerow + '.fullname' self.fullname = namerow + '.fullname'
self.username = namerow + '.username' self.username = namerow + '.username'
self.date = tweet + 'div.media-heading > div > .heading-right' self.date = tweet + 'div.media-heading .heading-right'
self.text = tweet + '.status-content-wrapper > .status-content.media-body' self.text = tweet + '.status-content-wrapper .status-content.media-body'
self.retweet = tweet = '.retweet'
class Profile(object): class Profile(object):
fullname = '.profile-card-fullname' fullname = '.profile-card-fullname'
username = '.profile-card-username' username = '.profile-card-username'
bio = '.profile-bio'
protected = '.protected-icon' protected = '.protected-icon'
verified = '.verified-icon' verified = '.verified-icon'
banner = '.profile-banner'
bio = '.profile-bio'
class BaseTestCase(BaseCase): class BaseTestCase(BaseCase):
@ -35,4 +37,4 @@ class BaseTestCase(BaseCase):
def get_timeline_tweet(num=1): def get_timeline_tweet(num=1):
return Tweet(f'#tweets > div:nth-child({num}) > div > div ') return Tweet(f'#tweets > div:nth-child({num}) ')

2
tests/test_search.py
View File

@ -2,7 +2,7 @@ from base import BaseTestCase
from parameterized import parameterized from parameterized import parameterized
class TestSearch(BaseTestCase): class SearchTest(BaseTestCase):
@parameterized.expand([['mobile_test'], ['mobile_test_2']]) @parameterized.expand([['mobile_test'], ['mobile_test_2']])
def test_username_search(self, username): def test_username_search(self, username):
self.search_username(username) self.search_username(username)