Improve test base
This commit is contained in:
parent
ee1d6f7cf0
commit
a5676e1c2c
|
@ -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,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)
|
||||||
|
|
Loading…
Reference in New Issue