@@ -21,14 +21,14 @@ function Test-GetComputeNodeById
2121 param ([string ]$accountName , [string ]$poolId )
2222
2323 $context = Get-AzureRMBatchAccountKeys - Name $accountName
24- $computeNodeId = (Get-AzureRMBatchComputeNode_ST - PoolId $poolId - BatchContext $context )[0 ].Id
24+ $computeNodeId = (Get-AzureBatchComputeNode_ST - PoolId $poolId - BatchContext $context )[0 ].Id
2525
26- $computeNode = Get-AzureRMBatchComputeNode_ST - PoolId $poolId - Id $computeNodeId - BatchContext $context
26+ $computeNode = Get-AzureBatchComputeNode_ST - PoolId $poolId - Id $computeNodeId - BatchContext $context
2727
2828 Assert-AreEqual $computeNodeId $computeNode.Id
2929
3030 # Verify positional parameters also work
31- $computeNode = Get-AzureRMBatchComputeNode_ST $poolId $computeNodeId - BatchContext $context
31+ $computeNode = Get-AzureBatchComputeNode_ST $poolId $computeNodeId - BatchContext $context
3232
3333 Assert-AreEqual $computeNodeId $computeNode.Id
3434}
@@ -44,7 +44,7 @@ function Test-ListComputeNodesByFilter
4444 $context = Get-AzureRMBatchAccountKeys - Name $accountName
4545 $filter = " state eq '" + " $state " + " '"
4646
47- $computeNodes = Get-AzureRMBatchComputeNode_ST - PoolId $poolId - Filter $filter - BatchContext $context
47+ $computeNodes = Get-AzureBatchComputeNode_ST - PoolId $poolId - Filter $filter - BatchContext $context
4848
4949 Assert-AreEqual $matches $computeNodes.Length
5050 foreach ($node in $computeNodes )
@@ -53,8 +53,8 @@ function Test-ListComputeNodesByFilter
5353 }
5454
5555 # Verify parent object parameter set also works
56- $pool = Get-AzureRMBatchPool_ST $poolId - BatchContext $context
57- $computeNodes = Get-AzureRMBatchComputeNode_ST - Pool $pool - Filter $filter - BatchContext $context
56+ $pool = Get-AzureBatchPool_ST $poolId - BatchContext $context
57+ $computeNodes = Get-AzureBatchComputeNode_ST - Pool $pool - Filter $filter - BatchContext $context
5858
5959 Assert-AreEqual $matches $computeNodes.Length
6060 foreach ($node in $computeNodes )
@@ -72,13 +72,13 @@ function Test-ListComputeNodesWithMaxCount
7272 param ([string ]$accountName , [string ]$poolId , [string ]$maxCount )
7373
7474 $context = Get-AzureRMBatchAccountKeys - Name $accountName
75- $computeNodes = Get-AzureRMBatchComputeNode_ST - PoolId $poolId - MaxCount $maxCount - BatchContext $context
75+ $computeNodes = Get-AzureBatchComputeNode_ST - PoolId $poolId - MaxCount $maxCount - BatchContext $context
7676
7777 Assert-AreEqual $maxCount $computeNodes.Length
7878
7979 # Verify parent object parameter set also works
80- $pool = Get-AzureRMBatchPool_ST $poolId - BatchContext $context
81- $computeNodes = Get-AzureRMBatchComputeNode_ST - Pool $pool - MaxCount $maxCount - BatchContext $context
80+ $pool = Get-AzureBatchPool_ST $poolId - BatchContext $context
81+ $computeNodes = Get-AzureBatchComputeNode_ST - Pool $pool - MaxCount $maxCount - BatchContext $context
8282
8383 Assert-AreEqual $maxCount $computeNodes.Length
8484}
@@ -92,27 +92,27 @@ function Test-ListAllComputeNodes
9292 param ([string ]$accountName , [string ]$poolId , [string ]$count )
9393
9494 $context = Get-AzureRMBatchAccountKeys - Name $accountName
95- $computeNodes = Get-AzureRMBatchComputeNode_ST - PoolId $poolId - BatchContext $context
95+ $computeNodes = Get-AzureBatchComputeNode_ST - PoolId $poolId - BatchContext $context
9696
9797 Assert-AreEqual $count $computeNodes.Length
9898
9999 # Verify parent object parameter set also works
100- $pool = Get-AzureRMBatchPool_ST $poolId - BatchContext $context
101- $computeNodes = Get-AzureRMBatchComputeNode_ST - Pool $pool - BatchContext $context
100+ $pool = Get-AzureBatchPool_ST $poolId - BatchContext $context
101+ $computeNodes = Get-AzureBatchComputeNode_ST - Pool $pool - BatchContext $context
102102
103103 Assert-AreEqual $count $computeNodes.Length
104104}
105105
106106<#
107107. SYNOPSIS
108- Tests piping Get-AzureRMBatchPool into Get-AzureRMBatchComputeNode
108+ Tests piping Get-AzureBatchPool into Get-AzureBatchComputeNode
109109#>
110110function Test-ListComputeNodePipeline
111111{
112112 param ([string ]$accountName , [string ]$poolId , [string ]$count )
113113
114114 $context = Get-AzureRMBatchAccountKeys - Name $accountName
115- $computeNodes = Get-AzureRMBatchPool_ST - Id $poolId - BatchContext $context | Get-AzureRMBatchComputeNode_ST - BatchContext $context
115+ $computeNodes = Get-AzureBatchPool_ST - Id $poolId - BatchContext $context | Get-AzureBatchComputeNode_ST - BatchContext $context
116116
117117 Assert-AreEqual $count $computeNodes.Count
118118}
@@ -131,14 +131,14 @@ function Test-RebootComputeNode
131131
132132 if ($usePipeline -eq ' 1' )
133133 {
134- Get-AzureRMBatchComputeNode_ST $poolId $computeNodeId - BatchContext $context | Restart-AzureRMBatchComputeNode_ST - RebootOption $rebootOption - BatchContext $context
134+ Get-AzureBatchComputeNode_ST $poolId $computeNodeId - BatchContext $context | Restart-AzureBatchComputeNode_ST - RebootOption $rebootOption - BatchContext $context
135135 }
136136 else
137137 {
138- Restart-AzureRMBatchComputeNode_ST $poolId $computeNodeId - RebootOption $rebootOption - BatchContext $context
138+ Restart-AzureBatchComputeNode_ST $poolId $computeNodeId - RebootOption $rebootOption - BatchContext $context
139139 }
140140
141- $computeNode = Get-AzureRMBatchComputeNode_ST - PoolId $poolId - Filter " id eq '$computeNodeId '" - BatchContext $context
141+ $computeNode = Get-AzureBatchComputeNode_ST - PoolId $poolId - Filter " id eq '$computeNodeId '" - BatchContext $context
142142
143143 Assert-AreEqual ' Rebooting' $computeNode.State
144144}
@@ -157,14 +157,14 @@ function Test-ReimageComputeNode
157157
158158 if ($usePipeline -eq ' 1' )
159159 {
160- Get-AzureRMBatchComputeNode_ST $poolId $computeNodeId - BatchContext $context | Reset-AzureRMBatchComputeNode_ST - ReimageOption $reimageOption - BatchContext $context
160+ Get-AzureBatchComputeNode_ST $poolId $computeNodeId - BatchContext $context | Reset-AzureBatchComputeNode_ST - ReimageOption $reimageOption - BatchContext $context
161161 }
162162 else
163163 {
164- Reset-AzureRMBatchComputeNode_ST $poolId $computeNodeId - ReimageOption $reimageOption - BatchContext $context
164+ Reset-AzureBatchComputeNode_ST $poolId $computeNodeId - ReimageOption $reimageOption - BatchContext $context
165165 }
166166
167- $computeNode = Get-AzureRMBatchComputeNode_ST - PoolId $poolId - Filter " id eq '$computeNodeId '" - BatchContext $context
167+ $computeNode = Get-AzureBatchComputeNode_ST - PoolId $poolId - Filter " id eq '$computeNodeId '" - BatchContext $context
168168
169169 Assert-AreEqual ' Reimaging' $computeNode.State
170170}
0 commit comments