more tests

ssciwr · Sep 4, 2024 · 1f2f113 · 1f2f113
1 parent a26a83f
commit 1f2f113
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/mailcom/parse.py b/mailcom/parse.py
@@ -120,7 +120,6 @@ def make_dir(path: str):
     # html_files = list_of_files(path_input, "html")
     for file in eml_files:
         text = get_text(file)
-        # if detect_html(text):
         text = get_html_text(text)
         print(text)
         # skip this text if email could not be parsed

diff --git a/mailcom/test/test_inout.py b/mailcom/test/test_inout.py
@@ -1,5 +1,6 @@
 from mailcom.inout import list_of_files, get_text, get_html_text
 import pytest
+from pathlib import Path
 
 def test_list_of_files_found(tmp_path):
     p = tmp_path / "test.eml"
@@ -34,4 +35,13 @@ def test_get_text_err():
 
 def test_get_html_text():
     html = """<html><head><title>Test</title></head></html>"""
-    assert get_html_text(html) == 'Test'
+    assert get_html_text(html) == 'Test'
+
+def test_get_html_text_noHtml():
+    noHtml = """Test"""
+    assert get_html_text(noHtml) == 'Test'
+
+def test_get_text_no_file(tmp_path):
+    p = tmp_path / "test.eml"
+    with pytest.raises(OSError):
+        get_text(p)