added response for test dataset if testMode is true

EGA-archive · Nov 22, 2024 · eaabfcf · eaabfcf
1 parent d9bc3fc
commit eaabfcf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/beacon/request/parameters.py b/beacon/request/parameters.py
@@ -216,6 +216,17 @@ def from_request(self, request: Request) -> Self:
                     self.query.request_parameters[k] = html.escape(v)
                 elif k == 'filters':
                     self.query.request_parameters[k] = html.escape(v)
+                elif k == 'testMode':
+                    v = html.escape(v)
+                    if v.lower() == 'true':
+                        v = True
+                    elif v.lower() == 'false':
+                        v = False
+                    else:
+                        err = 'testMode parameter can only be either true or false value'
+                        errcode=400
+                        raise_exception(err, errcode)
+                    self.query.test_mode = v
                 elif k in ["start", "end", "assemblyId", "referenceName", "referenceBases", "alternateBases", "variantType","variantMinLength","variantMaxLength","geneId","genomicAlleleShortForm","aminoacidChange","clinicalRelevance", "mateName"]:
                     try:
                         if ',' in v:# pragma: no cover

diff --git a/beacon/response/builder.py b/beacon/response/builder.py
@@ -35,7 +35,7 @@ async def builder(self, request: Request, datasets, qparams, entry_type, entry_i
         complete_module='beacon.connections.'+source+'.executor'
         import importlib
         module = importlib.import_module(complete_module, package=None)
-        testMode = qparams.query.testMode
+        testMode = qparams.query.test_mode
         if testMode == True:
             datasets = ['test']
         datasets_docs, datasets_count, count, entity_schema, include, datasets = await module.execute_function(self, entry_type, datasets, qparams, entry_id)