48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from seleniumbase import BaseCase
|
|
|
|
|
|
class Tweet(object):
|
|
def __init__(self, tweet=''):
|
|
namerow = tweet + 'div.media-heading '
|
|
self.fullname = namerow + '.fullname'
|
|
self.username = namerow + '.username'
|
|
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'
|
|
protected = '.protected-icon'
|
|
verified = '.verified-icon'
|
|
banner = '.profile-banner'
|
|
bio = '.profile-bio'
|
|
|
|
|
|
class Timeline(object):
|
|
newest = 'div[class="show-more status-el"]'
|
|
older = 'div[class="show-more"]'
|
|
end = '.timeline-end'
|
|
none = '.timeline-none'
|
|
protected = '.timeline-protected'
|
|
|
|
|
|
class BaseTestCase(BaseCase):
|
|
def setUp(self):
|
|
super(BaseTestCase, self).setUp()
|
|
|
|
def tearDown(self):
|
|
super(BaseTestCase, self).tearDown()
|
|
|
|
def open_nitter(self, page=''):
|
|
self.open(f'http://localhost:5000/{page}')
|
|
|
|
def search_username(self, username):
|
|
self.open_nitter()
|
|
self.update_text('input', username)
|
|
self.submit('form')
|
|
|
|
|
|
def get_timeline_tweet(num=1):
|
|
return Tweet(f'#tweets > div:nth-child({num}) ')
|