Add validation to only download for supported domains

This commit is contained in:
Matthew Adams
2021-04-13 17:52:59 +10:00
parent 2b0c005f8b
commit 2c3df6416f
3 changed files with 69 additions and 1 deletions

17
tests.py Normal file
View File

@@ -0,0 +1,17 @@
import unittest
from validator import extractUrl, isSupportedUrl
class TestUrlParser(unittest.TestCase):
def test_supportedUrl(self):
url = "https://vm.tiktok.com/ZSJrgyXdt/"
supportedResponse = isSupportedUrl(url)
self.assertEqual(supportedResponse["supported"], 'true')
def test_unsupportedUrl(self):
url = "https://www.twitch.tv/robcdee/clip/AgileLivelyCucumberPartyTime"
supportedResponse = isSupportedUrl(url)
self.assertEqual(supportedResponse["supported"], 'false')
if __name__ == '__main__':
unittest.main()