diff --git a/test/test_extension.py b/test/test_extension.py index 3ae317a..6e8ca87 100644 --- a/test/test_extension.py +++ b/test/test_extension.py @@ -7,5 +7,11 @@ def test_extension(): assert MY_SECRET_SITE assert SocialSiteId.MY_SECRET_SITE - REGEXES.append(('test', MY_SECRET_SITE)) + REGEXES.append((r'test(\d+)', MY_SECRET_SITE)) + + url = 'test123' + result = determine_social_from_url(url) + assert result is not None + assert result.social_site_id == MY_SECRET_SITE + assert result.social_id == '123'