Added tests for builder

1 files changed, 145 insertions, 0 deletions

diff --git a/tests/test_builder.py b/tests/test_builder.py
new file mode 100644
index 0000000..7b43d7c
--- /dev/null
+++ b/tests/test_builder.py
@@ -0,0 +1,145 @@
+import re
+
+from edify.builder import ANY, WORD, DIGIT
+from edify.builder import AtLeast
+from edify.builder import AtMost
+from edify.builder import Escaped
+from edify.builder import Exact
+from edify.builder import OneOrMore
+from edify.builder import Optional
+from edify.builder import Range
+from edify.builder import RegexBuilder
+from edify.builder import ZeroOrMore
+
+def catch_exception(subroutine):
+    try:
+        subroutine()
+    except Exception as e:
+        return e.__class__
+
+def test_invalid_section_error():
+    assert catch_exception(lambda: RegexBuilder().add(1)) is ValueError
+
+
+def test_optional():
+    regex = (
+        RegexBuilder()
+        .add(Optional(ANY))
+        .build()
+    )
+    assert re.match(regex, "hello") is not None
+    assert re.match(regex, "") is not None
+
+
+def test_zero_or_more():
+    regex = (
+        RegexBuilder()
+        .add(ZeroOrMore(ANY))
+        .build()
+    )
+    assert re.match(regex, "hello") is not None
+    assert re.match(regex, "") is not None
+
+
+def test_one_or_more():
+    regex = (
+        RegexBuilder()
+        .add(OneOrMore(ANY))
+        .build()
+    )
+    assert re.match(regex, "hello") is not None
+    assert re.match(regex, "") is None
+
+
+def test_exact():
+    regex = (
+        RegexBuilder()
+        .add(Exact(WORD, 2))
+        .build()
+    )
+    assert re.match(regex, "hello") is not None
+    assert re.match(regex, "hello world") is not None
+    assert re.match(regex, "hello world hello") is not None
+
+
+def test_range():
+    regex = (
+        RegexBuilder()
+        .add(Range(DIGIT, 1, 2))
+        .build()
+    )
+    assert re.match(regex, "1") is not None
+    assert re.match(regex, "2") is not None
+
+
+def test_at_least():
+    regex = (
+        RegexBuilder()
+        .add(AtLeast(ANY, 2))
+        .build()
+    )
+    assert re.match(regex, "hello") is not None
+    assert re.match(regex, "hello world") is not None
+    assert re.match(regex, "hello world hello") is not None
+
+
+def test_at_most():
+    regex = (
+        RegexBuilder()
+        .add(AtMost(ANY, 2))
+        .build()
+    )
+    assert re.match(regex, "hello") is not None
+    assert re.match(regex, "hello world") is not None
+    assert re.match(regex, "hello world hello") is not None
+
+
+def test_escaped():
+    regex = (
+        RegexBuilder()
+        .add(Escaped('.'))
+        .build()
+    )
+    assert re.match(regex, ".") is not None
+    assert re.match(regex, "..") is not None
+    assert re.match(regex, "...") is not None
+
+
+def test_escaped_with_quantifier():
+    regex = (
+        RegexBuilder()
+        .add(Escaped('.'))
+        .add(AtLeast(ANY, 2))
+        .build()
+    )
+    assert re.match(regex, "..") is None
+    assert re.match(regex, "...") is not None
+    assert re.match(regex, "....") is not None
+
+def test_escaped_with_quantifier_and_optional():
+    regex = (
+        RegexBuilder()
+        .add(Escaped('.'))
+        .add(Optional(ANY))
+        .add(AtLeast(ANY, 2))
+        .build()
+    )
+    assert re.match(regex, "..") is None
+    assert re.match(regex, "...") is not None
+    assert re.match(regex, "....") is not None
+    assert re.match(regex, ".....") is not None
+
+
+def test_email_using_builder():
+    regex = (
+        RegexBuilder()
+        .add(OneOrMore(ANY))
+        .add(Escaped('@'))
+        .add(OneOrMore(ANY))
+        .add(Escaped('.'))
+        .add(OneOrMore(ANY))
+        .build()
+    )
+    assert re.match(regex, "[email protected]") is not None
+    assert re.match(regex, "hello@example") is None
+