public void GetBatchComputeNodeODataTest()
{
// Setup cmdlet to get a single compute node
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.PoolId = "testPool";
cmdlet.Id = "computeNode1";
cmdlet.Select = "id,state";
string requestSelect = null;
// Fetch the OData clauses off the request. The OData clauses are applied after user provided RequestInterceptors, so a ResponseInterceptor is used.
ComputeNodeGetResponse getResponse = BatchTestHelpers.CreateComputeNodeGetResponse(cmdlet.Id);
RequestInterceptor requestInterceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor <ComputeNodeGetParameters, ComputeNodeGetResponse>(getResponse);
ResponseInterceptor responseInterceptor = new ResponseInterceptor((response, request) =>
{
requestSelect = request.Parameters.DetailLevel.SelectClause;
return(Task.FromResult(response));
});
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
requestInterceptor, responseInterceptor
};
cmdlet.ExecuteCmdlet();
Assert.Equal(cmdlet.Select, requestSelect);
}
public void GetBatchComputeNodeTest()
{
// Setup cmdlet to get a compute node by id
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.PoolId = "pool";
cmdlet.Id = "computeNode1";
cmdlet.Filter = null;
// Build a compute node instead of querying the service on a Get ComputeNode call
ComputeNodeGetResponse response = BatchTestHelpers.CreateComputeNodeGetResponse(cmdlet.Id);
RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor <ComputeNodeGetParameters, ComputeNodeGetResponse>(response);
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSComputeNode> pipeline = new List <PSComputeNode>();
commandRuntimeMock.Setup(r => r.WriteObject(It.IsAny <PSComputeNode>())).Callback <object>(c => pipeline.Add((PSComputeNode)c));
cmdlet.ExecuteCmdlet();
// Verify that the cmdlet wrote the compute node returned from the OM to the pipeline
Assert.Equal(1, pipeline.Count);
Assert.Equal(cmdlet.Id, pipeline[0].Id);
}
