From 2cea637b2e936265d2001043d2b2d4e62559dc17 Mon Sep 17 00:00:00 2001
From: dvora-h <67596500+dvora-h@users.noreply.github.com>
Date: Thu, 4 Aug 2022 16:46:28 +0300
Subject: Add support for WITHSUFFIXTRIE to FT.CREATE (#2324)

* withsuffixtrie

* Update test_search.py

* fix
---
 tests/test_search.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'tests/test_search.py')

diff --git a/tests/test_search.py b/tests/test_search.py
index ee4fa55..5fe5ab1 100644
--- a/tests/test_search.py
+++ b/tests/test_search.py
@@ -1712,3 +1712,27 @@ def test_expire_while_search(modclient: redis.Redis):
         modclient.ft().search(Query("*")).docs[1]
     time.sleep(1)
     assert 2 == modclient.ft().search(Query("*")).total
+
+
+@pytest.mark.redismod
+@pytest.mark.experimental
+def test_withsuffixtrie(modclient: redis.Redis):
+    # create index
+    assert modclient.ft().create_index((TextField("txt"),))
+    waitForIndex(modclient, getattr(modclient.ft(), "index_name", "idx"))
+    info = modclient.ft().info()
+    assert "WITHSUFFIXTRIE" not in info["attributes"][0]
+    assert modclient.ft().dropindex("idx")
+
+    # create withsuffixtrie index (text fiels)
+    assert modclient.ft().create_index((TextField("t", withsuffixtrie=True)))
+    waitForIndex(modclient, getattr(modclient.ft(), "index_name", "idx"))
+    info = modclient.ft().info()
+    assert "WITHSUFFIXTRIE" in info["attributes"][0]
+    assert modclient.ft().dropindex("idx")
+
+    # create withsuffixtrie index (tag field)
+    assert modclient.ft().create_index((TagField("t", withsuffixtrie=True)))
+    waitForIndex(modclient, getattr(modclient.ft(), "index_name", "idx"))
+    info = modclient.ft().info()
+    assert "WITHSUFFIXTRIE" in info["attributes"][0]
-- 
cgit v1.2.1