fix: fetching keploy-test-id from request header

src/keploy/djangoCov.py

@@ -8,13 +8,13 @@ def __init__(self, get_response):
 
     def __call__(self, request):
         id = request.headers.get('KEPLOY-TEST-ID')
+        testSet = request.headers.get('KEPLOY-TEST-SET-ID')
         if id == None:
             return self.get_response(request)
         cov = coverage.Coverage(cover_pylib=False)
         cov.start()
         response = self.get_response(request)
         cov.stop()
         result = cov.get_data()
-        write_dedup(result, id)
+        write_dedup(result, id, testSet)
         return response
-

src/keploy/fastApiCov.py

@@ -7,14 +7,18 @@
 class FastApiCoverageMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request: Request, call_next):
         id = request.headers.get('KEPLOY-TEST-ID')
+        testSet = request.headers.get('KEPLOY-TEST-SET-ID')
         if id is None:
             response = await call_next(request)
             return response
-
+        global_coverage = coverage.Coverage.current()
+        if global_coverage is not None:
+            response = await call_next(request)
+            return response
         cov = coverage.Coverage(cover_pylib=False)
         cov.start()
         response = await call_next(request)
         cov.stop()
         result = cov.get_data()
-        write_dedup(result, id)
+        write_dedup(result, id, testSet)
         return response

src/keploy/flaskCov.py

@@ -1,19 +1,21 @@
 import coverage
-
+from werkzeug.wrappers import Request
 from .utils import write_dedup
 
 class FlaskCoverageMiddleware:
     def __init__(self, app):
         self.app = app
 
     def __call__(self, environ, start_response):
-        id = environ.get('KEPLOY_TEST_ID')
+        request = Request(environ)
+        id = request.headers.get("Keploy-Test-Id")
+        testSet = request.headers.get("Keploy-Test-Set-Id")
         if id == None:
             return self.app(environ, start_response)
         cov = coverage.Coverage(cover_pylib=False)
         cov.start()
         response = self.app(environ, start_response)
         cov.stop()
         result = cov.get_data()
-        write_dedup(result, id)
+        write_dedup(result, id, testSet)
         return response

src/keploy/utils.py

@@ -1,6 +1,6 @@
 import yaml
 
-def write_dedup(result, id):
+def write_dedup(result, id, testSet=""):
     filePath = 'dedupData.yaml'
     existingData = []
     try:
@@ -11,7 +11,7 @@ def write_dedup(result, id):
             pass
 
     yaml_data = {
-        'id': id,
+        'id': testSet + "/" + id,
         'executedLinesByFile': {}
     }
     for file in result.measured_files():
@@ -21,4 +21,3 @@ def write_dedup(result, id):
     existingData.append(yaml_data)
     with open(filePath, 'w') as file:
         yaml.dump(existingData, file, sort_keys=False)
-