nitter/tests/test_profile.py

79 lines
2.5 KiB
Python
Raw Normal View History

2019-06-28 02:56:02 +00:00
from base import BaseTestCase, Profile
2019-06-28 17:18:52 +00:00
from parameterized import parameterized
2019-06-28 02:56:02 +00:00
2019-06-28 17:18:52 +00:00
profiles = [
2019-06-28 22:42:44 +00:00
['mobile_test', 'Test account',
2019-06-28 17:18:52 +00:00
'Test Account. test test Testing username with @mobile_test_2 and a #hashtag'],
2019-06-28 22:42:44 +00:00
['mobile_test_2', 'mobile test 2', '']
2019-06-28 17:18:52 +00:00
]
2019-06-28 02:56:02 +00:00
2019-06-28 17:18:52 +00:00
verified = [['jack'], ['elonmusk']]
protected = [
['mobile_test_7', 'mobile test 7🔒', ''],
['Poop', 'Randy🔒', 'Social media fanatic.']
2019-06-28 17:18:52 +00:00
]
invalid = [['thisprofiledoesntexist'], ['%']]
2019-06-28 02:56:02 +00:00
2019-06-28 22:42:44 +00:00
banner_color = [
['TheTwoffice', '29, 161, 242'],
['profiletest', '80, 176, 58']
]
banner_image = [
['mobile_test', 'profile_banners%2F82135242%2F1384108037%2F1500x500']
]
2019-06-28 17:18:52 +00:00
2019-06-28 22:42:44 +00:00
class ProfileTest(BaseTestCase):
2019-06-28 17:18:52 +00:00
@parameterized.expand(profiles)
2019-06-28 22:42:44 +00:00
def test_data(self, username, fullname, bio):
2019-06-28 17:18:52 +00:00
self.open_nitter(username)
self.assert_exact_text(fullname, Profile.fullname)
self.assert_exact_text(f'@{username}', Profile.username)
if len(bio) > 0:
self.assert_exact_text(bio, Profile.bio)
else:
self.assert_element_absent(Profile.bio)
@parameterized.expand(verified)
def test_verified(self, username):
self.open_nitter(username)
2019-06-28 02:56:02 +00:00
self.assert_element_visible(Profile.verified)
2019-06-28 17:18:52 +00:00
@parameterized.expand(protected)
2019-06-28 22:42:44 +00:00
def test_protected(self, username, fullname, bio):
2019-06-28 17:18:52 +00:00
self.open_nitter(username)
2019-06-28 02:56:02 +00:00
self.assert_element_visible(Profile.protected)
2019-06-28 17:18:52 +00:00
self.assert_exact_text(fullname, Profile.fullname)
self.assert_exact_text(f'@{username}', Profile.username)
2019-06-28 02:56:02 +00:00
2019-06-28 17:18:52 +00:00
if len(bio) > 0:
self.assert_text(bio, Profile.bio)
else:
self.assert_element_absent(Profile.bio)
2019-06-28 02:56:02 +00:00
2019-06-28 17:18:52 +00:00
@parameterized.expand(invalid)
def test_invalid_username(self, username):
self.open_nitter(username)
self.assert_text(f'User "{username}" not found')
2019-06-28 02:56:02 +00:00
def test_suspended(self):
# TODO: detect suspended
self.open_nitter('test')
self.assert_text(f'User "test" not found')
2019-06-28 22:42:44 +00:00
@parameterized.expand(banner_color)
def test_banner_color(self, username, color):
self.open_nitter(username)
banner = self.find_element(Profile.banner + '-color')
self.assertIn(color, banner.value_of_css_property('background-color'))
@parameterized.expand(banner_image)
def test_banner_image(self, username, url):
self.open_nitter(username)
banner = self.find_element(Profile.banner + ' img')
self.assertIn(url, banner.get_attribute('src'))