Enable content__contains pattern (#236)

Co-authored-by: Jonas Lundberg <jonas@5monkeys.se>
lundberg · Jul 7, 2023 · 33438de · 33438de
1 parent 5a74249
commit 33438de
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/docs/api.md b/docs/api.md
@@ -301,10 +301,11 @@ respx.get("all://*.example.org/foo/")
 ### Content
 Matches request raw *content*, using [eq](#eq) as default lookup.
 > Key: `content`  
-> Lookups: [eq](#eq)
+> Lookups: [eq](#eq), [contains](#contains)
 ``` python
 respx.post("https://example.org/", content="foobar")
 respx.post("https://example.org/", content=b"foobar")
+respx.post("https://example.org/", content__contains="bar")
 ```
 
 ### Data

diff --git a/respx/patterns.py b/respx/patterns.py
@@ -483,7 +483,7 @@ def parse(self, request: httpx.Request) -> Any:
 
 
 class Content(ContentMixin, Pattern):
-    lookups = (Lookup.EQUAL,)
+    lookups = (Lookup.EQUAL, Lookup.CONTAINS)
     key = "content"
     value: bytes
 
@@ -492,6 +492,9 @@ def clean(self, value: Union[bytes, str]) -> bytes:
             return value.encode()
         return value
 
+    def _contains(self, value: Union[bytes, str]) -> Match:
+        return Match(self.value in value)
+
 
 class JSON(ContentMixin, PathPattern):
     lookups = (Lookup.EQUAL,)

diff --git a/tests/test_patterns.py b/tests/test_patterns.py
@@ -311,6 +311,9 @@ def test_url_pattern_hash():
     [
         (Lookup.EQUAL, b"foobar", True),
         (Lookup.EQUAL, "foobar", True),
+        (Lookup.CONTAINS, b"bar", True),
+        (Lookup.CONTAINS, "bar", True),
+        (Lookup.CONTAINS, "baz", False),
     ],
 )
 def test_content_pattern(lookup, content, expected):