update test

src/maggma/api/resource/read_resource.py

@@ -201,8 +201,8 @@ def search(**queries: dict[str, STORE_PARAMS]) -> Union[dict, Response]:
 
             if self.query_to_configure_on_request is not None:
                 # give the key name "request", arbitrary choice, as only the value gets merged into the query
-                queries["request"] = self.header_processor.configure_query_from_request(
-                    request, self.query_to_configure_on_request
+                queries["groups"] = self.header_processor.configure_query_on_request(
+                    request=request, query_operator=self.query_to_configure_on_request
                 )
             # allowed query parameters
             query_params = [
@@ -217,7 +217,6 @@ def search(**queries: dict[str, STORE_PARAMS]) -> Union[dict, Response]:
                         detail="'limit' and 'skip' parameters have been renamed. "
                         "Please update your API client to the newest version.",
                     )
-
                 else:
                     raise HTTPException(
                         status_code=400,

tests/api/test_read_resource.py

@@ -11,7 +11,7 @@
 
 from maggma.api.query_operator import NumericQuery, SparseFieldsQuery, StringQueryOperator
 from maggma.api.resource import ReadOnlyResource
-from maggma.api.resource.core import HintScheme
+from maggma.api.resource.core import HeaderProcessor, HintScheme
 from maggma.stores import AliasingStore, MemoryStore
 
 
@@ -32,6 +32,18 @@ class Owner(BaseModel):
 total_owners = len(owners)
 
 
+# Create a subclass of the header processor to prevent TypeErrors:
+#  Can't instantiate abstract class HeaderProcessor with abstract methods
+class TestHeaderProcessor(HeaderProcessor):
+    def configure_query_on_request(self, request, query_operator):
+        # Implement the method
+        return {"name": "PersonAge9"}
+
+    def process_header(self, response, request):
+        # Implement the method
+        pass
+
+
 @pytest.fixture()
 def owner_store():
     store = MemoryStore("owners", key="name")
@@ -134,6 +146,8 @@ def search_helper(payload, base: str = "/?", debug=True) -> Response:
             NumericQuery(model=Owner),
             SparseFieldsQuery(model=Owner),
         ],
+        header_processor=TestHeaderProcessor(),
+        query_to_configure_on_request=StringQueryOperator(model=Owner),
         disable_validation=True,
     )
     app = FastAPI()
@@ -214,3 +228,16 @@ def test_resource_compound():
     assert len(data) == 1
     assert data[0]["name"] == "PersonAge20Weight200"
     assert "weight" not in data[0]
+
+
+def test_configure_query_on_request():
+    payload = {
+        "name": "PersonAge20Weight200",
+        "_all_fields": False,
+        "_fields": "name,age",
+        "weight_min": 199.3,
+        "weight_max": 201.9,
+        "age": 20,
+    }
+    res, data = search_helper(payload=payload, base="/?", debug=True)
+    assert res.status_code == 200