nitter/tests/base.py

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 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}) ')