Register-TssDistributedEngine - rewrite

thycotic-ps · Oct 14, 2021 · bd8fb30 · bd8fb30
1 parent d7a3f55
commit bd8fb30
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 46 deletions.
diff --git a/src/functions/distributed-engines/Register-TssDistributedEngine.ps1 b/src/functions/distributed-engines/Register-TssDistributedEngine.ps1
@@ -42,53 +42,10 @@ function Register-TssDistributedEngine {
     )
     begin {
         $tssParams = $PSBoundParameters
-        $invokeParams = . $GetInvokeApiParams $TssSession
     }
     process {
         Get-TssInvocation $PSCmdlet.MyInvocation
-        if ($tssParams.ContainsKey('TssSession') -and $TssSession.IsValidSession()) {
-            Compare-TssVersion $TssSession '10.9.000064' $PSCmdlet.MyInvocation
-            $uri = $TssSession.ApiUrl, 'distributed-engine', 'update-engine-status' -join '/'
-            $invokeParams.Uri = $uri
-            $invokeParams.Method = 'POST'
-
-            $registerBody = @{data = @{} }
-            $allEngines = @()
-            foreach ($engine in $Id) {
-                $cEngine = @{}
-                $cEngine.Add('changeType','Activate')
-                $cEngine.Add('engineId',$engine)
-                $cEngine.Add('siteId',$SiteId)
-                $allEngines += $cEngine
-            }
-            $registerBody.data.Add('engines',$allEngines)
-
-            $invokeParams.Body = $registerBody | ConvertTo-Json -Depth 100
-            if ($PSCmdlet.ShouldProcess("description: $", "$($invokeParams.Method) $uri with: `n$($invokeParams.Body)")) {
-                Write-Verbose "$($invokeParams.Method) $uri with: `n$($invokeParams.Body)"
-                try {
-                    $apiResponse = Invoke-TssApi @invokeParams
-                    $restResponse = . $ProcessResponse $apiResponse
-                } catch {
-                    Write-Warning 'Issue warning message'
-                    $err = $_
-                    . $ErrorHandling $err
-                }
-
-                if ($restResponse.results) {
-                    foreach ($record in $restResponse.results) {
-                        [Thycotic.PowerShell.DistributedEngines.EngineActivation]@{
-                            EngineId     = $record.engineId
-                            EngineName   = $record.engineName
-                            Error        = $record.error
-                            IsSuccessful = $record.success
-                        }
-                    }
-
-                }
-            }
-        } else {
-            Write-Warning 'No valid session found'
-        }
+        $tssParams.Add('Status','Activate')
+        Update-TssDistributedEngine @tssParams
     }
 }
diff --git a/tests/distributed-engines/Register-TssDistributedEngine.Tests.ps1 b/tests/distributed-engines/Register-TssDistributedEngine.Tests.ps1
@@ -3,7 +3,7 @@ BeforeDiscovery {
 }
 Describe "$commandName verify parameters" {
     BeforeDiscovery {
-        [object[]]$knownParameters = 'TssSession', 'Id', 'SiteId'
+        [object[]]$knownParameters = 'TssSession', 'EngineId', 'SiteId'
         [object[]]$currentParams = ([Management.Automation.CommandMetaData]$ExecutionContext.SessionState.InvokeCommand.GetCommand($commandName,'Function')).Parameters.Keys
         [object[]]$commandDetails = [System.Management.Automation.CommandInfo]$ExecutionContext.SessionState.InvokeCommand.GetCommand($commandName,'Function')
         $unknownParameters = Compare-Object -ReferenceObject $knownParameters -DifferenceObject $currentParams -PassThru