base.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. from seleniumbase import BaseCase
  2. class Tweet(object):
  3. def __init__(self, tweet=''):
  4. namerow = tweet + 'div.media-heading '
  5. self.fullname = namerow + '.fullname'
  6. self.username = namerow + '.username'
  7. self.date = tweet + 'div.media-heading .heading-right'
  8. self.text = tweet + '.status-content-wrapper .status-content.media-body'
  9. self.retweet = tweet = '.retweet'
  10. class Profile(object):
  11. fullname = '.profile-card-fullname'
  12. username = '.profile-card-username'
  13. protected = '.protected-icon'
  14. verified = '.verified-icon'
  15. banner = '.profile-banner'
  16. bio = '.profile-bio'
  17. class Timeline(object):
  18. newest = 'div[class="show-more status-el"]'
  19. older = 'div[class="show-more"]'
  20. end = '.timeline-end'
  21. none = '.timeline-none'
  22. protected = '.timeline-protected'
  23. class Poll(object):
  24. votes = '.poll-info'
  25. choice = '.poll-meter'
  26. value = 'poll-choice-value'
  27. option = 'poll-choice-option'
  28. leader = 'leader'
  29. class BaseTestCase(BaseCase):
  30. def setUp(self):
  31. super(BaseTestCase, self).setUp()
  32. def tearDown(self):
  33. super(BaseTestCase, self).tearDown()
  34. def open_nitter(self, page=''):
  35. self.open(f'http://localhost:5000/{page}')
  36. def search_username(self, username):
  37. self.open_nitter()
  38. self.update_text('input', username)
  39. self.submit('form')
  40. def get_timeline_tweet(num=1):
  41. return Tweet(f'#tweets > div:nth-child({num}) ')