add sentry sample rate config

config.yaml

@@ -61,6 +61,14 @@ options:
     default: false
     type: boolean
 
+  sentry-sample-rate:
+    description: |
+      A value between 0 (0% of errors) and 1 (100% of errors) to indicate the proportion of errors 
+      to be captured by Sentry.
+    
+    default: 1.0
+    type: float
+
   workflows-file-name:
     description: Name of the wheel file resource attached to the charm.
     default: ""

src/charm.py

@@ -283,6 +283,10 @@ def _validate(self, event):  # noqa: C901
             elif self.config["auth-provider"] == "google":
                 self._check_required_config(REQUIRED_OIDC_CONFIG)
 
+        sample_rate = self.config["sentry-sample-rate"]
+        if self.config["sentry-dsn"] and (sample_rate < 0 or sample_rate > 1):
+            raise ValueError("Invalid config: sentry-sample-rate must be between 0 and 1")
+
     def _update(self, event):
         """Update the Temporal worker configuration and replan its execution.
 

src/resources/worker.py

@@ -135,6 +135,7 @@ async def run_worker(unpacked_file_name, module_name):
                 release=os.getenv("TWC_SENTRY_RELEASE").strip() or None,
                 environment=os.getenv("TWC_SENTRY_ENVIRONMENT").strip() or None,
                 redact_params=os.getenv("TWC_SENTRY_REDACT_PARAMS"),
+                sample_rate=os.getenv("TWC_SENTRY_SAMPLE_RATE"),
             )
 
             worker_opt = WorkerOptions(sentry=sentry)

tests/integration/conftest.py

@@ -28,6 +28,7 @@
 @pytest_asyncio.fixture(name="deploy", scope="module")
 async def deploy(ops_test: OpsTest):
     """Verify the app is up and running."""
+    await ops_test.model.set_config({"update-status-hook-interval": "1m"})
     await setup_temporal_ecosystem(ops_test)
 
     charm = await ops_test.build_charm(".")

tests/integration/helpers.py

@@ -135,10 +135,10 @@ async def setup_temporal_ecosystem(ops_test: OpsTest):
         ops_test: PyTest object.
     """
     await asyncio.gather(
-        ops_test.model.deploy(APP_NAME_SERVER, channel="edge"),
+        ops_test.model.deploy(APP_NAME_SERVER, channel="edge", config={"num-history-shards": 1}),
         ops_test.model.deploy(APP_NAME_ADMIN, channel="edge"),
         ops_test.model.deploy(APP_NAME_UI, channel="edge"),
-        ops_test.model.deploy("postgresql-k8s", channel="14", trust=True),
+        ops_test.model.deploy("postgresql-k8s", channel="14/stable", trust=True),
     )
 
     async with ops_test.fast_forward():

tests/unit/test_charm.py

@@ -137,6 +137,7 @@ def test_ready(self, _process_wheel_file, _setup_container):
                         "TWC_SENTRY_DSN": "",
                         "TWC_SENTRY_ENVIRONMENT": "",
                         "TWC_SENTRY_RELEASE": "",
+                        "TWC_SENTRY_SAMPLE_RATE": 1.0,
                         "TWC_SENTRY_REDACT_PARAMS": False,
                         "TWC_SUPPORTED_ACTIVITIES": "all",
                         "TWC_SUPPORTED_WORKFLOWS": "all",
@@ -149,6 +150,14 @@ def test_ready(self, _process_wheel_file, _setup_container):
                     "on-check-failure": {"up": "ignore"},
                 }
             },
+            "checks": {
+                "up": {
+                    "override": "replace",
+                    "level": "alive",
+                    "period": "10s",
+                    "exec": {"command": "python check_status.py"},
+                }
+            },
         }
         got_plan = harness.get_container_pebble_plan("temporal-worker").to_dict()
         self.assertEqual(got_plan, want_plan)