From 6259fc78bebd7bcf263778f251b295a824aa3188 Mon Sep 17 00:00:00 2001
From: selfisekai <laura@selfisekai.rocks>
Date: Wed, 9 Sep 2020 18:37:20 +0200
Subject: [PATCH] unit tests for regex utils

---
 tests/regex_utils.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 tests/regex_utils.py

diff --git a/tests/regex_utils.py b/tests/regex_utils.py
new file mode 100644
index 0000000..0cf632e
--- /dev/null
+++ b/tests/regex_utils.py
@@ -0,0 +1,38 @@
+from librefi.utils import regex_search_string
+
+HTML_STRING = """
+    <form method="POST" action="/?your=mother">
+        <input type="hidden" value="tesco" />
+    </form>
+"""
+
+PATTERN = r'<form [^>]+action="([^"]+)'
+
+EXPECTED_RESULT = "/?your=mother"
+
+
+def test_regex_search_single():
+    result = regex_search_string(PATTERN, HTML_STRING)
+    assert result is not None
+    assert result == EXPECTED_RESULT
+
+
+def test_regex_search_multiple():
+    result = regex_search_string(
+        [r'dupsko', PATTERN, r'byle jakie'], HTML_STRING)
+    assert result is not None
+    assert result == EXPECTED_RESULT
+
+
+def test_regex_search_invalid():
+    result = regex_search_string([r'dupsko', r'czorne'], HTML_STRING)
+    assert result is None
+    assert result != EXPECTED_RESULT
+
+
+def test_regex_search_default():
+    default = "sth"
+    result = regex_search_string(
+        [r'dupsko', r'czorne'], HTML_STRING, default=default)
+    assert result is not None
+    assert result == default