Small regexp improvement + test on CI

Author: federicotdn

.github/workflows/run-code-blocks.yml

@@ -30,6 +30,8 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.13'
+      - name: Check script readiness
+        run: python -m unittest scripts/md-k6.py
       - name: Run Updated Code Blocks
         env:
           ALL_CHANGED_FILES: ${{ steps.changed-files.outputs.all_changed_files }}

docs/sources/k6/next/javascript-api/jslib/http-instrumentation-pyroscope/_index.md

@@ -34,7 +34,7 @@ This module, by default, adds three key-value pairs:
 
 ## Example
 
-This example demonstrates how to use the this library to instrument every HTTP request made in a script with the `baggage` header.
+This example demonstrates how to use the this library to instrument every HTTP request made in a script with the `baggage` header TEST.
 
 {{< code >}}
 

scripts/md-k6.py

@@ -4,6 +4,7 @@
 # Usage:
 #   python3 md-k6.py <file>
 
+import io
 import os
 import re
 import json
@@ -221,10 +222,11 @@ def read_front_matter_env(f: TextIO) -> dict[str, str]:
             continue
 
         parts = line.strip().split(":")
-        key, value = parts[0].strip(), parts[1].strip()
+        key = parts[0].strip()
+        value = ":".join(parts[1:]).strip()
 
         # Look for YAML keys with ALL_CAPS_NAMES
-        if re.match(r"[A-Z0-9_]", key) and value:
+        if re.match(r"^[A-Z0-9_]+$", key) and value:
             result[key] = value
 
     return result
@@ -344,6 +346,26 @@ def testExtractScriptIndexEnv(self):
         self.assertEqual(len(scripts), 1)
         self.assertEqual(scripts[0].text, "testing testing")
 
+    def testReadIndexEnvFile(self):
+        text = io.StringIO(
+            """
+---
+title: Testing
+FOO_BAR: 1234
+   FOO_BAR2:1234
+FOO_BAR3: some:value
+not_added: foo
+not_added2:
+NOT_ADDED3:
+---
+"""
+        )
+        env = read_front_matter_env(text)
+
+        self.assertEqual(
+            env, {"FOO_BAR": "1234", "FOO_BAR2": "1234", "FOO_BAR3": "some:value"}
+        )
+
     def testExtractScriptOptions(self):
         text = """
 <!-- md-k6:nofail,env.A=B,env.C=X- -->