Cynosphere/nitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Improve test base

Browse source
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):
def __init__(self, tweet=''):
namerow = tweet + 'div.media-heading > div > .fullname-and-username > '
namerow = tweet + 'div.media-heading '
self.fullname = namerow + '.fullname'
self.username = namerow + '.username'
self.date = tweet + 'div.media-heading > div > .heading-right'
self.text = tweet + '.status-content-wrapper > .status-content.media-body'
self.date = tweet + 'div.media-heading .heading-right'
self.text = tweet + '.status-content-wrapper .status-content.media-body'
self.retweet = tweet = '.retweet'
class Profile(object):
fullname = '.profile-card-fullname'
username = '.profile-card-username'
bio = '.profile-bio'
protected = '.protected-icon'
verified = '.verified-icon'
banner = '.profile-banner'
bio = '.profile-bio'
class BaseTestCase(BaseCase):
@ -35,4 +37,4 @@ class BaseTestCase(BaseCase):
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
class TestSearch(BaseTestCase):
class SearchTest(BaseTestCase):
@parameterized.expand([['mobile_test'], ['mobile_test_2']])
def test_username_search(self, username):
self.search_username(username)