Modification of testing to use mocked input directories

tests/test_run_service.py

 # tests/test_run_service.py
-from polygonqueryonraster.app import app
-import requests
-import threading
+import os
 
-def test_run_service():
-    # Start the Flask development server in a separate thread
-    thread = threading.Thread(target=app.run, kwargs={'debug': False, 'port': 5001})
-    thread.start()
+class GeoTIFFService:
+    # Existing methods...
 
-    # Wait for the server to start
-    import time
-    time.sleep(1)
+    def setup_routes(self):
+        # Ensure the directory exists and contains subdirectories
+        if os.path.exists(self.directory):
+            subdirs = next(os.walk(self.directory), (None, []))[1]
+            if not subdirs:
+                print(f"No subdirectories found in {self.directory}")
+            else:
+                for subdir in subdirs:
+                    subdir_path = os.path.join(self.directory, subdir)
+                    self.create_route(f'/{subdir}', subdir_path)
+        else:
+            raise ValueError(f"Directory {self.directory} does not exist.")
 
-    try:
-        # Make a sample GET request
-        response_get = requests.get('http://127.0.0.1:5001/api')
-        assert response_get.status_code == 200
-        assert response_get.json()["message"] == "Hello, this is a simple REST API!"
-
-        # Make a sample POST request
-        response_post = requests.post('http://127.0.0.1:5001/api', json={"message": "Updated message!"})
-        assert response_post.status_code == 200
-        assert response_post.json()["message"] == "Updated message!"
-
-    finally:
-        # Shut down the Flask development server
-        thread.join(timeout=1)
 
 if __name__ == '__main__':
     test_run_service()

tests/test_setup.py

-from polygonqueryonraster.app import app
+import tempfile
+import requests
+import threading
+import time
+from polygonqueryonraster.app import GeoTIFFService
+
+def test_flask_app():
+    # Create a temporary directory for testing
+    with tempfile.TemporaryDirectory() as tempdir:
+        # Start the Flask app
+        service = GeoTIFFService(tempdir, port=5001)
+        thread = threading.Thread(target=service.run)
+        thread.start()
+
+        # Give it time to start
+        time.sleep(2)
+
+        try:
+            # Send a request to the Flask API
+            response = requests.get('http://127.0.0.1:5001/api')
+            assert response.status_code == 200
+            assert response.json()["message"] == "Hello, this is a simple REST API!"
+        finally:
+            # Stop the Flask server
+            thread.join(timeout=1)
 
 if __name__ == '__main__':
-    app.run(debug=True, port=5001)  # You can specify the port of your choice
+    test_flask_app()
+