From dfbecb6f266b319719c71effad4113f9c7091269 Mon Sep 17 00:00:00 2001
From: Ettore Leandro Tognoli <ettoreleandrotognoli@gmail.com>
Date: Sat, 10 Feb 2024 13:17:26 -0300
Subject: [PATCH] add MediaResolver

---
 ensta/MediaResolver.py                      | 20 ++++++
 test_ensta/resources/media/empty.txt        |  0
 test_ensta/resources/media/with-content.txt |  1 +
 test_ensta/test_media_resolver.py           | 78 +++++++++++++++++++++
 4 files changed, 99 insertions(+)
 create mode 100644 ensta/MediaResolver.py
 create mode 100644 test_ensta/resources/media/empty.txt
 create mode 100644 test_ensta/resources/media/with-content.txt
 create mode 100644 test_ensta/test_media_resolver.py

diff --git a/ensta/MediaResolver.py b/ensta/MediaResolver.py
new file mode 100644
index 0000000..6eebbd4
--- /dev/null
+++ b/ensta/MediaResolver.py
@@ -0,0 +1,20 @@
+
+import requests
+import tempfile
+from pathlib import Path
+from urllib.parse import urlparse
+
+class MediaResolver:
+    
+    def resolve_url(self, url: str):
+        parsed_url = urlparse(url)
+        if not parsed_url.scheme:
+            return url
+        filename = Path(parsed_url.path).name
+        _, file = tempfile.mkstemp(filename)
+        response = requests.get(url)
+        response.raise_for_status()
+        with open(file, 'wb') as output_stream:
+            for content in response.iter_content():
+                output_stream.write(content)
+        return file
\ No newline at end of file
diff --git a/test_ensta/resources/media/empty.txt b/test_ensta/resources/media/empty.txt
new file mode 100644
index 0000000..e69de29
diff --git a/test_ensta/resources/media/with-content.txt b/test_ensta/resources/media/with-content.txt
new file mode 100644
index 0000000..596218e
--- /dev/null
+++ b/test_ensta/resources/media/with-content.txt
@@ -0,0 +1 @@
+ensta
\ No newline at end of file
diff --git a/test_ensta/test_media_resolver.py b/test_ensta/test_media_resolver.py
new file mode 100644
index 0000000..fe5aa9c
--- /dev/null
+++ b/test_ensta/test_media_resolver.py
@@ -0,0 +1,78 @@
+from unittest import TestCase
+import time
+from urllib.parse import urlparse
+from ensta.MediaResolver import MediaResolver
+from pathlib import Path
+from http.server import HTTPServer, SimpleHTTPRequestHandler
+from threading import Thread
+from requests import HTTPError
+
+RESOURCES = Path(__file__).parent.joinpath('resources')
+MEDIA = RESOURCES.joinpath('media')
+
+class HttpServerTestCase(TestCase):
+
+    root: Path = None
+    server_thread: Thread = None
+    http_server: HTTPServer = None
+
+    def get_base_url(self):
+        return 'http://{}:{}/'.format(
+            *self.http_server.server_address
+        )
+        
+    def get_url(self, path: str):
+        return self.get_base_url() + path
+
+    @classmethod
+    def setUpClass(cls):
+        class HttpHandler(SimpleHTTPRequestHandler):
+            def __init__(self, *args, **kwargs) -> None:
+                super().__init__(*args, directory=cls.root and str(cls.root.absolute()), **kwargs)
+
+        cls.http_server = HTTPServer(
+            ('', 0,),
+            HttpHandler,
+            bind_and_activate=True,
+        )
+        cls.server_thread = Thread(target=cls.http_server.serve_forever)
+        cls.server_thread.daemon = True
+        cls.server_thread.start()
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.http_server.shutdown()
+        cls.server_thread.join()
+
+
+
+class DownloadTest(HttpServerTestCase):
+    
+    root = MEDIA
+    
+    media_resolver: MediaResolver
+    
+    def setUp(self) -> None:
+        self.media_resolver = MediaResolver()
+
+    def test_local(self):
+        local = './test.txt'
+        result = self.media_resolver.resolve_url(local)
+        self.assertEqual(result, local)
+        
+    def test_http(self):
+        url = self.get_url('with-content.txt')
+        result = self.media_resolver.resolve_url(url)
+        with open(result, 'rb') as input_stream:
+            self.assertEqual(input_stream.read(), b'ensta')
+
+    def test_empty_http(self):
+        url = self.get_url('empty.txt')
+        result = self.media_resolver.resolve_url(url)
+        with open(result, 'rb') as input_stream:
+            self.assertEqual(input_stream.read(), b'')
+            
+    def test_404(self):
+        url = self.get_url('not-found')
+        with self.assertRaises(HTTPError):
+            self.media_resolver.resolve_url(url)
\ No newline at end of file