Fix pytest

Author: YahooMing

Dockerfile

@@ -3,7 +3,6 @@ FROM python:3.9-slim
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONPATH="${PYTHONPATH}:/app"
-
 WORKDIR /app
 
 COPY requirements.txt .
@@ -21,5 +20,7 @@ COPY . .
 
 EXPOSE 8501
 
-CMD ["pytest", "-v", "app/tests/test_http.py", "app/tests/test_file.py", "app/tests/test_file_ext.py"]
+#RUN "pytest -v app/tests/test_http.py app/tests/test_file.py app/tests/test_file_ext.py" > logs.log
+
+
 CMD ["streamlit", "run", "app/main.py", "--server.port=8501", "--server.address=0.0.0.0"]

app/main.py

@@ -1,6 +1,5 @@
 # Main giga kox file
 import streamlit as st
-from tests import test_http, test_file, test_file_ext
 import time
 from src import FileProcessor, Summarizer
 from config.settings import settings
@@ -10,9 +9,9 @@ def main():
         st.session_state.tests_run = False
 
     if not st.session_state.tests_run:
-        test_http.test_port()
-        test_file_ext.test_file_ext("manu.txt")
-        test_file_ext.test_file_ext("manu.pdf")
+        # test_http.test_port()
+        # test_file_ext.test_file_ext("manu.txt")
+        # test_file_ext.test_file_ext("manu.pdf")
         st.session_state.tests_run = True
 
     st.title("Kaka - AI powered document summary")
@@ -25,7 +24,7 @@ def main():
         st.session_state.submitted = False
 
     if file is not None and not st.session_state.submitted:
-        test_file.test_file_name(file)
+        # test_file.test_file_name(file)
         if st.button("Summarize"):
             st.session_state.submitted = True
             st.rerun()

app/src/file_processor.py

@@ -2,7 +2,7 @@
 import streamlit as st
 import os
 from typing import Optional
-from tests import test_file_ext
+#from tests import test_file_ext
 
 class FileProcessor:
     def __init__(self):

app/tests/test_file.py

@@ -1,12 +1,14 @@
 from log import Logger
+import pytest
+#from ../main import main
+from ..src import FileProcessor, Summarizer
 
 logger = Logger(__name__).get_logger()
 
-def test_file_name(file):
-    try:
-        if file is not None:
-            logger.info(f"File name is: {file.name}")
-        else:
-            logger.error("There is an error in retrieving name")
-    except Exception as e:
-        logger.error(f"An error occurred: {e}")
+def test_file_name():
+    file_procesor = FileProcessor()
+    file_procesor.file = "manu.txt"
+    #file_procesor.upload_file()
+    ret = file_procesor.getFile()
+    assert ret != None
+

app/tests/test_file_ext.py

@@ -1,29 +1,34 @@
 from log import Logger
 import pytest
 import os
+from src import FileProcessor
 
 logger = Logger(__name__).get_logger()
 
-def test_file_ext(file_name: str) -> bool:
-    supported_extensions = {".txt"}
+# def test_file_ext(file_name: str) -> bool:
+#     supported_extensions = {".txt"}
 
-    _, ext = os.path.splitext(file_name)
+#     _, ext = os.path.splitext(file_name)
 
-    if any(file_name.endswith(ext) for ext in supported_extensions):
-        logger.info(f"Type {ext} is supported")
-        return True
-    else:
-        logger.error(f"Type {ext} isn't supported")
-        return False
+#     if any(file_name.endswith(ext) for ext in supported_extensions):
+#         logger.info(f"Type {ext} is supported")
+#         return True
+#     else:
+#         logger.error(f"Type {ext} isn't supported")
+#         return False
 
 @pytest.mark.parametrize("file_name, expected", [
-    ("image.jpeg", False),
-    ("document.pdf", False),
-    ("archive.zip", False),
-    ("notes.txt", True),
-    ("image.png", False),
+    (".jpeg", False),
+    (".pdf", False),
+    (".zip", False),
+    (".txt", True),
+    (".png", False),
 ])
 
 def test_is_supported_file(file_name, expected):
     logger.info(f"Testing file: {file_name}")
-    assert test_file_ext(file_name) == expected
+    file_processor = FileProcessor()
+    supp_files = file_processor.supported_types.keys()
+    supp_files = list(supp_files)
+    ret = file_name in supp_files
+    assert ret == expected

app/tests/test_http.py

@@ -1,19 +1,14 @@
 import socket
 from log import Logger
+import pytest
 
 logger = Logger(__name__).get_logger()
 
 def test_port():
     logger.info("Testing socket")
-    try:
-        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-            s.settimeout(2) 
-            result = s.connect_ex(('127.0.0.1', 8501))
-            if result == 0:
-                logger.info("8501 is open")
-            else:
-                logger.error("8501 is closed")
-            assert result == 0, "8501 is not available"
-    except Exception as e:
-        logger.exception(f"An error occured during the test {e}")
-        raise
+    s =socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.settimeout(2) 
+    result = s.connect_ex(('127.0.0.1', 8501))
+
+    assert result == 0, "8501 is not available"
+    logger.info("Socket test passed")