fix tests

DataDog · Oct 7, 2024 · 277cefe · 277cefe
1 parent a3413c5
commit 277cefe
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tracer/src/Datadog.Trace/AppSec/Security.cs b/tracer/src/Datadog.Trace/AppSec/Security.cs
@@ -207,9 +207,9 @@ private ApplyDetails[] UpdateFromRcm(Dictionary<string, List<RemoteConfiguration
             {
                 // store the last config state, clearing any previous state, without deserializing any payloads yet.
                 var anyChange = _configurationStatus.StoreLastConfigState(configsByProduct, removedConfigs);
-
+                var securityStateChange = Enabled != _configurationStatus.EnableAsm;
                 // normally CanBeToggled should not need a check as asm_features capacity is only sent if AppSec env var is null, but still guards it in case
-                if (_configurationStatus.IncomingUpdateState.SecurityStateChange && _settings.CanBeToggled)
+                if (securityStateChange && _settings.CanBeToggled)
                 {
                     // disable ASM scenario
                     if (Enabled && _configurationStatus.EnableAsm == false)
@@ -239,11 +239,10 @@ private ApplyDetails[] UpdateFromRcm(Dictionary<string, List<RemoteConfiguration
                             WafRuleFileVersion = updateResult.RuleFileVersion;
                         }
 
+                        _configurationStatus.ResetUpdateMarkers();
                         UpdateActiveAddresses();
                     }
                 }
-
-                _configurationStatus.ResetUpdateMarkers();
             }
             catch (Exception e)
             {