nitter/tests/test_embed.py

import requests
from base import BaseTestCase, Media
from parameterized import parameterized


class Embed:
    container = '.tweet-embed'
    footer = '.embed-footer'
    tweet_content = '.tweet-content'
    tweet_header = '.tweet-header'
    fullname = '.fullname'
    username = '.username'
    avatar = '.avatar'
    stats = '.tweet-stats'
    quote = '.quote'
    error_panel = '.error-panel'


class TweetEmbedTest(BaseTestCase):
    """Test tweet embed rendering."""
    tweet = 'elonmusk/status/1141367104702038016'

    def test_embed_container_visible(self):
        self.open_nitter(self.tweet + '/embed')
        self.assert_element_visible(Embed.container)

    def test_embed_has_footer(self):
        self.open_nitter(self.tweet + '/embed')
        self.assert_element_visible(Embed.footer)
        self.assert_text_visible('Read more on', Embed.footer)

    def test_embed_has_tweet_content(self):
        self.open_nitter(self.tweet + '/embed')
        self.assert_element_visible(Embed.tweet_content)

    def test_embed_has_avatar(self):
        self.open_nitter(self.tweet + '/embed')
        self.assert_element_visible(Embed.avatar)

    def test_embed_has_username(self):
        self.open_nitter(self.tweet + '/embed')
        self.assert_element_visible(Embed.username)

    def test_embed_has_stats(self):
        self.open_nitter(self.tweet + '/embed')
        self.assert_element_visible(Embed.stats)

    def test_embed_footer_links_to_tweet(self):
        self.open_nitter(self.tweet + '/embed')
        href = self.get_attribute(Embed.footer, 'href')
        self.assertIn('/elonmusk/status/1141367104702038016', href)


class TweetEmbedMediaTest(BaseTestCase):
    """Test embed rendering with various media types."""

    def test_embed_with_image(self):
        self.open_nitter('mobile_test/status/519364660823207936/embed')
        self.assert_element_visible(Embed.container)
        self.scroll_to(Media.container)
        self.assert_element_visible(Media.image)

    def test_embed_with_gif(self):
        self.open_nitter('elonmusk/status/1141367104702038016/embed')
        self.assert_element_visible(Embed.container)
        self.scroll_to(Media.container)
        self.assert_element_visible(Media.gif)

    def test_embed_with_video(self):
        self.open_nitter('d0m96/status/1078373829917974528/embed')
        self.assert_element_visible(Embed.container)
        self.scroll_to(Media.container)
        self.assert_element_visible(Media.video)

    def test_embed_with_gallery(self):
        self.open_nitter('mobile_test/status/451108446603980803/embed')
        self.assert_element_visible(Embed.container)
        self.scroll_to(Media.container)
        self.assert_element_visible(Media.row)


class TweetEmbedQuoteTest(BaseTestCase):
    """Test embed rendering with quoted tweets."""

    def test_embed_with_quote_shows_quote(self):
        self.open_nitter('elonmusk/status/1138827760107790336/embed')
        self.assert_element_visible(Embed.container)
        self.assert_element_visible(Embed.quote)

    def test_embed_quote_has_content(self):
        self.open_nitter('elonmusk/status/1138827760107790336/embed')
        quote = self.find_element(Embed.quote)
        self.assertIsNotNone(quote.text)


class EmbedErrorTest(BaseTestCase):
    """Test embed error handling."""

    def test_nonexistent_tweet_shows_error(self):
        self.open_nitter('nobody/status/1/embed')
        self.assert_element_visible('.tweet-embed.error-embed')
        self.assert_text_visible('not found', Embed.error_panel)

    def test_protected_account_embed_shows_error(self):
        self.open_nitter('mobile_test_7/status/1/embed')
        self.assert_element_visible('.tweet-embed.error-embed')

    def test_invalid_tweet_id_shows_error(self):
        self.open_nitter('jack/status/notanumber/embed')
        self.assert_element_visible('.tweet-embed.error-embed')


class OEmbedApiTest(BaseTestCase):
    """Test oEmbed API endpoint."""
    base_url = 'http://localhost:8080'
    tweet_url = 'https://twitter.com/elonmusk/status/1141367104702038016'

    def test_oembed_returns_json(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url={self.tweet_url}')
        self.assertEqual(resp.status_code, 200)
        self.assertEqual(resp.headers['Content-Type'], 'application/json')

    def test_oembed_has_required_fields(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url={self.tweet_url}')
        data = resp.json()
        self.assertEqual(data['type'], 'rich')
        self.assertEqual(data['version'], '1.0')
        self.assertIn('html', data)
        self.assertIn('author_name', data)
        self.assertIn('provider_name', data)

    def test_oembed_html_contains_iframe(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url={self.tweet_url}')
        data = resp.json()
        self.assertIn('<iframe', data['html'])
        self.assertIn('/embed', data['html'])

    def test_oembed_has_cors_header(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url={self.tweet_url}')
        self.assertEqual(resp.headers.get('Access-Control-Allow-Origin'), '*')

    def test_oembed_missing_url_returns_400(self):
        resp = requests.get(f'{self.base_url}/api/oembed')
        self.assertEqual(resp.status_code, 400)

    def test_oembed_invalid_url_returns_400(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url=https://example.com')
        self.assertEqual(resp.status_code, 400)

    def test_oembed_supports_x_com_url(self):
        x_url = 'https://x.com/elonmusk/status/1141367104702038016'
        resp = requests.get(f'{self.base_url}/api/oembed?url={x_url}')
        self.assertEqual(resp.status_code, 200)

    def test_oembed_strips_query_params(self):
        url_with_params = 'https://twitter.com/elonmusk/status/1141367104702038016?s=20&t=abc'
        resp = requests.get(f'{self.base_url}/api/oembed?url={url_with_params}')
        self.assertEqual(resp.status_code, 200)
        data = resp.json()
        self.assertIn('html', data)

    def test_oembed_handles_trailing_slash(self):
        url_with_slash = 'https://twitter.com/elonmusk/status/1141367104702038016/'
        resp = requests.get(f'{self.base_url}/api/oembed?url={url_with_slash}')
        self.assertEqual(resp.status_code, 200)

    def test_oembed_handles_mobile_url(self):
        mobile_url = 'https://mobile.twitter.com/elonmusk/status/1141367104702038016'
        resp = requests.get(f'{self.base_url}/api/oembed?url={mobile_url}')
        self.assertEqual(resp.status_code, 200)

    def test_oembed_rejects_malformed_tweet_id(self):
        bad_url = 'https://twitter.com/elonmusk/status/notanumber'
        resp = requests.get(f'{self.base_url}/api/oembed?url={bad_url}')
        self.assertEqual(resp.status_code, 400)

    def test_oembed_maxwidth_param_accepted(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url={self.tweet_url}&maxwidth=400')
        self.assertEqual(resp.status_code, 200)

    def test_oembed_author_url_present(self):
        resp = requests.get(f'{self.base_url}/api/oembed?url={self.tweet_url}')
        data = resp.json()
        self.assertIn('author_url', data)
        self.assertIn('elonmusk', data['author_url'])


class VideoEmbedTest(BaseTestCase):
    """Test video embed route (/i/videos/tweet/{id})."""
    video_tweet_id = '1078373829917974528'

    def test_video_embed_has_video_element(self):
        self.open_nitter(f'i/videos/tweet/{self.video_tweet_id}')
        self.assert_element_visible('video')

    def test_video_embed_has_poster(self):
        self.open_nitter(f'i/videos/tweet/{self.video_tweet_id}')
        poster = self.get_attribute('video', 'poster')
        self.assertIsNotNone(poster)
        self.assertIn('pic/', poster)

    def test_video_embed_nonexistent_returns_error(self):
        self.open_nitter('i/videos/tweet/1')
        self.assert_element_visible('.error-embed')