add entry remove, call it

Azure · Jul 26, 2024 · 1e7a071 · 1e7a071
1 parent 92b26ac
commit 1e7a071
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/tools/test-proxy/Azure.Sdk.Tools.TestProxy/Common/RecordSession.cs b/tools/test-proxy/Azure.Sdk.Tools.TestProxy/Common/RecordSession.cs
@@ -1,4 +1,4 @@
-// Copyright (c) Microsoft Corporation. All rights reserved.
+// Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
 using System;
@@ -84,7 +84,7 @@ public void Record(RecordEntry entry)
 
         public RecordEntry Lookup(RecordEntry requestEntry, RecordMatcher matcher, IEnumerable<RecordedTestSanitizer> sanitizers, bool remove = true)
         {
-            foreach(RecordedTestSanitizer sanitizer in sanitizers)
+            foreach (RecordedTestSanitizer sanitizer in sanitizers)
             {
                 sanitizer.Sanitize(requestEntry);
             }
@@ -103,6 +103,14 @@ public RecordEntry Lookup(RecordEntry requestEntry, RecordMatcher matcher, IEnum
             }
         }
 
+        public void Remove(RecordEntry entry)
+        {
+            lock (Entries)
+            {
+                Entries.Remove(entry);
+            }
+        }
+
         public void Sanitize(RecordedTestSanitizer sanitizer)
         {
             lock (Entries)

diff --git a/tools/test-proxy/Azure.Sdk.Tools.TestProxy/RecordingHandler.cs b/tools/test-proxy/Azure.Sdk.Tools.TestProxy/RecordingHandler.cs
@@ -521,7 +521,7 @@ public async Task HandlePlaybackRequest(string recordingId, HttpRequest incoming
 
             if (remove)
             {
-                session.Session.Lookup(entry, session.CustomMatcher ?? Matcher, sanitizers, remove: true);
+                session.Session.Remove(match);
             }
         }
-Original file line number
+Diff line change
@@ Expand Up @@
                 if (remove)
                 {
-                    session.Session.Lookup(entry, session.CustomMatcher ?? Matcher, sanitizers, remove: true);
+                    session.Session.Remove(match);
                 }
             }
@@ Expand Down @@