Modify analyser tests to reflect the use of extractor

2 files changed, 9 insertions, 0 deletions

diff --git a/tests/utils/get_random_vid_info.py b/tests/utils/get_random_vid_info.py
new file mode 100644
index 0000000..f85a226
--- /dev/null
+++ b/tests/utils/get_random_vid_info.py
@@ -0,0 +1,5 @@
+from hashlib import md5
+
+def get_random_vid_duration(url: str) -> int:
+    hash = md5(url.encode()).hexdigest()
+    return int(hash[:4], 16)
diff --git a/tests/utils/vid_url_to_html.py b/tests/utils/vid_url_to_html.py
index 355e830..60f6ccc 100644
--- a/tests/utils/vid_url_to_html.py
+++ b/tests/utils/vid_url_to_html.py
@@ -1,6 +1,7 @@
 from urllib.parse import urlparse, parse_qs
 
 from components.extractor.check_url import is_video
+from .get_random_vid_info import get_random_vid_duration
 
 def extract_vid_id(url: str) -> str:
     if not is_video(url):
@@ -16,3 +17,6 @@ def extract_vid_id(url: str) -> str:
 def get_vid_html_from_url(url: str) -> str:
     with open(f'tests/data/video@{extract_vid_id(url)}.html', 'r') as file:
         return file.read()
+
+def obtain_vid_duration(url:str, html: str = '') -> int:
+    return get_random_vid_duration(url)