Skip to content

Commit 363ff9a

Browse files
committed
adding ShouldProcess block
1 parent a462e67 commit 363ff9a

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

src/Security/Security/Cmdlets/AdvancedThreatProtection/DisableAdvancedThreatProtection .cs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,11 @@ public class DisableAdvancedThreatProtection : SecurityCenterCmdletBase
2828

2929
public override void ExecuteCmdlet()
3030
{
31-
var result = SecurityCenterClient.AdvancedThreatProtection.CreateWithHttpMessagesAsync(ResourceId, isEnabled: false).GetAwaiter().GetResult().Body;
32-
WriteObject(result, enumerateCollection: true);
31+
if (ShouldProcess(target: ResourceId, action: $"Disabling advanced threat protection policy for '{ResourceId}'."))
32+
{
33+
var result = SecurityCenterClient.AdvancedThreatProtection.CreateWithHttpMessagesAsync(ResourceId, isEnabled: false).GetAwaiter().GetResult().Body;
34+
WriteObject(result, enumerateCollection: true);
35+
}
3336
}
3437
}
3538
}

src/Security/Security/Cmdlets/AdvancedThreatProtection/EnableAdvancedThreatProtection .cs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,11 @@ public class EnableAdvancedThreatProtection : SecurityCenterCmdletBase
2828

2929
public override void ExecuteCmdlet()
3030
{
31-
var result = SecurityCenterClient.AdvancedThreatProtection.CreateWithHttpMessagesAsync(ResourceId, isEnabled: true).GetAwaiter().GetResult().Body;
32-
WriteObject(result, enumerateCollection: true);
31+
if (ShouldProcess(target: ResourceId, action: $"Enabling advanced threat protection policy for '{ResourceId}'."))
32+
{
33+
var result = SecurityCenterClient.AdvancedThreatProtection.CreateWithHttpMessagesAsync(ResourceId, isEnabled: true).GetAwaiter().GetResult().Body;
34+
WriteObject(result, enumerateCollection: true);
35+
}
3336
}
3437
}
3538
}

0 commit comments

Comments
 (0)