Add --notification_config support for dotnet templates

microsoft · Feb 15, 2023 · fc11af5 · fc11af5
1 parent 58f756d
commit fc11af5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/cli/onefuzz/templates/libfuzzer.py b/src/cli/onefuzz/templates/libfuzzer.py
@@ -604,6 +604,7 @@ def dotnet(
         ensemble_sync_delay: Optional[int] = None,
         check_fuzzer_help: bool = True,
         expect_crash_on_failure: bool = False,
+        notification_config: Optional[NotificationConfig] = None,
     ) -> Optional[Job]:
         """
         libfuzzer-dotnet task
@@ -670,6 +671,7 @@ def dotnet(
         ]
 
         helper.create_containers()
+        helper.setup_notifications(notification_config)
 
         helper.upload_setup(setup_dir, target_exe)
         if inputs:
@@ -738,6 +740,7 @@ def dotnet_dll(
         colocate_all_tasks: bool = False,
         colocate_secondary_tasks: bool = True,
         expect_crash_on_failure: bool = False,
+        notification_config: Optional[NotificationConfig] = None,
     ) -> Optional[Job]:
         pool = self.onefuzz.pools.get(pool_name)
 
@@ -814,6 +817,7 @@ def dotnet_dll(
         ]
 
         helper.create_containers()
+        helper.setup_notifications(notification_config)
 
         helper.upload_setup(setup_dir, target_dll)