base.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  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 Media(object):
  30. container = '.attachments'
  31. row = '.gallery-row'
  32. image = '.still-image'
  33. video = '.gallery-video'
  34. gif = '.gallery-gif'
  35. class BaseTestCase(BaseCase):
  36. def setUp(self):
  37. super(BaseTestCase, self).setUp()
  38. def tearDown(self):
  39. super(BaseTestCase, self).tearDown()
  40. def open_nitter(self, page=''):
  41. self.open(f'http://localhost:5000/{page}')
  42. def search_username(self, username):
  43. self.open_nitter()
  44. self.update_text('input', username)
  45. self.submit('form')
  46. def get_timeline_tweet(num=1):
  47. return Tweet(f'#tweets > div:nth-child({num}) ')