Code coverage Fix

Author: yashnap

src/extension/tests/Test_TelemetryWriter.py

@@ -158,10 +158,10 @@ def test_events_deleted_outside_of_extension_while_extension_is_running(self):
         self.telemetry_writer.write_event("testing telemetry write to file", Constants.TelemetryEventLevel.Error, "Test Task")
         os.listdir = backup_os_listdir
 
-    # ==================== Common Helper Method for Loading Event Files ====================
+    # ==================== Integration test for credential sanitization in telemetry ====================
     def _load_sanitized_event(self, message):
         """
-        Common helper method to write event to telemetry and load the sanitized message.
+        Helper method to write event to telemetry and load the sanitized message.
         The regex sanitization happens automatically in TelemetryWriter.
 
         Args:
@@ -184,9 +184,8 @@ def _load_sanitized_event(self, message):
             f.close()
             return sanitized_message
 
-    # ==================== Test cases for credential sanitization in telemetry messages ====================
-    def test_sanitize_credentials_multiple_repos(self):
-        """Test 2: Failed repo sync with multiple repo URLs containing different credentials"""
+    def test_sanitize_credentials_in_telemetry_event(self):
+        """Integration test: Verify credentials are sanitized in telemetry events"""
         if self.runtime.is_github_runner:
             return
 
@@ -196,26 +195,17 @@ def test_sanitize_credentials_multiple_repos(self):
         expected_message = "Failed repo sync: https://user1@repo1.example.com https://user2@repo2.example.com/path"
         self.assertEqual(sanitized_message, expected_message)
 
-    def test_sanitize_credentials_username_only_no_password(self):
-        """Test 3: Using mirror with username only (no password)"""
-        if self.runtime.is_github_runner:
-            return
+    def test_load_sanitized_event_skips_on_github_runner(self):
+        """Test: Helper method returns None when running on GitHub runner"""
+        # Mock is_github_runner to be True to test the skip path
+        original_is_github_runner = self.runtime.is_github_runner
+        self.runtime.is_github_runner = True
 
-        message = "Using mirror https://testuser@repo.example.com/path"
+        result = self._load_sanitized_event("test message")
+        self.assertIsNone(result)
 
-        sanitized_message = self._load_sanitized_event(message)
-        self.assertIn("testuser@repo.example.com", sanitized_message)
-
-    def test_sanitize_credentials_special_characters_in_password(self):
-        """Test 4: Downloading from repo with special characters in password"""
-        if self.runtime.is_github_runner:
-            return
-
-        message = "Downloading from https://svc-user:AbC_123-.$%!@repo.contoso.com/rpm"
-
-        sanitized_message = self._load_sanitized_event(message)
-        self.assertNotIn("AbC_123-.$%!", sanitized_message)
-        self.assertIn("svc-user@repo.contoso.com", sanitized_message)
+        # Restore
+        self.runtime.is_github_runner = original_is_github_runner
 
 if __name__ == '__main__':
      SUITE = unittest.TestLoader().loadTestsFromTestCase(TestTelemetryWriter)