public void ListCertificatesMaxCountTest()
{
// Verify default max count
Assert.Equal(Microsoft.Azure.Commands.Batch.Utils.Constants.DefaultMaxCount, cmdlet.MaxCount);
// Setup cmdlet to list pools without filters and a max count
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.ThumbprintAlgorithm = null;
cmdlet.Thumbprint = null;
cmdlet.Filter = null;
int maxCount = 2;
cmdlet.MaxCount = maxCount;
string[] thumbprintsOfConstructedCerts = new[] { "12345", "67890", "ABCDE" };
// Build some Certificates instead of querying the service on a List Certificates call
CertificateListResponse response = BatchTestHelpers.CreateCertificateListResponse(thumbprintsOfConstructedCerts);
RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor <CertificateListParameters, CertificateListResponse>(response);
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSCertificate> pipeline = new List <PSCertificate>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSCertificate>()))
.Callback <object>(c => pipeline.Add((PSCertificate)c));
cmdlet.ExecuteCmdlet();
// Verify that the max count was respected
Assert.Equal(maxCount, pipeline.Count);
// Verify setting max count <= 0 doesn't return nothing
cmdlet.MaxCount = -5;
pipeline.Clear();
cmdlet.ExecuteCmdlet();
Assert.Equal(thumbprintsOfConstructedCerts.Length, pipeline.Count);
}
public void ListBatchCertificatesWithoutFiltersTest()
{
// Setup cmdlet to list certs without filters
BatchAccountContext context = BatchTestHelpers.CreateBatchContextWithKeys();
cmdlet.BatchContext = context;
cmdlet.ThumbprintAlgorithm = null;
cmdlet.Thumbprint = null;
cmdlet.Filter = null;
string[] thumbprintsOfConstructedCerts = new[] { "12345", "67890", "ABCDE" };
// Build some Certificates instead of querying the service on a List Certificates call
CertificateListResponse response = BatchTestHelpers.CreateCertificateListResponse(thumbprintsOfConstructedCerts);
RequestInterceptor interceptor = BatchTestHelpers.CreateFakeServiceResponseInterceptor <CertificateListParameters, CertificateListResponse>(response);
cmdlet.AdditionalBehaviors = new List <BatchClientBehavior>()
{
interceptor
};
// Setup the cmdlet to write pipeline output to a list that can be examined later
List <PSCertificate> pipeline = new List <PSCertificate>();
commandRuntimeMock.Setup(r =>
r.WriteObject(It.IsAny <PSCertificate>()))
.Callback <object>(c => pipeline.Add((PSCertificate)c));
cmdlet.ExecuteCmdlet();
// Verify that the cmdlet wrote the constructed certs to the pipeline
Assert.Equal(3, pipeline.Count);
int poolCount = 0;
foreach (PSCertificate c in pipeline)
{
Assert.True(thumbprintsOfConstructedCerts.Contains(c.Thumbprint));
poolCount++;
}
Assert.Equal(thumbprintsOfConstructedCerts.Length, poolCount);
}
/// <summary>
/// List all certificates of the Api Management service instance.
/// </summary>
/// <param name='nextLink'>
/// Required. NextLink from the previous successful call to List
/// operation.
/// </param>
/// <param name='cancellationToken'>
/// Cancellation token.
/// </param>
/// <returns>
/// List Certificates operation response details.
/// </returns>
public async Task <CertificateListResponse> ListNextAsync(string nextLink, CancellationToken cancellationToken)
{
// Validate
if (nextLink == null)
{
throw new ArgumentNullException("nextLink");
}
// Tracing
bool shouldTrace = TracingAdapter.IsEnabled;
string invocationId = null;
if (shouldTrace)
{
invocationId = TracingAdapter.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("nextLink", nextLink);
TracingAdapter.Enter(invocationId, this, "ListNextAsync", tracingParameters);
}
// Construct URL
string url = "";
url = url + nextLink;
url = url.Replace(" ", "%20");
// Create HTTP transport objects
HttpRequestMessage httpRequest = null;
try
{
httpRequest = new HttpRequestMessage();
httpRequest.Method = HttpMethod.Get;
httpRequest.RequestUri = new Uri(url);
// Set Headers
// Set Credentials
cancellationToken.ThrowIfCancellationRequested();
await this.Client.Credentials.ProcessHttpRequestAsync(httpRequest, cancellationToken).ConfigureAwait(false);
// Send Request
HttpResponseMessage httpResponse = null;
try
{
if (shouldTrace)
{
TracingAdapter.SendRequest(invocationId, httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
httpResponse = await this.Client.HttpClient.SendAsync(httpRequest, cancellationToken).ConfigureAwait(false);
if (shouldTrace)
{
TracingAdapter.ReceiveResponse(invocationId, httpResponse);
}
HttpStatusCode statusCode = httpResponse.StatusCode;
if (statusCode != HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
CloudException ex = CloudException.Create(httpRequest, null, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false));
if (shouldTrace)
{
TracingAdapter.Error(invocationId, ex);
}
throw ex;
}
// Create Result
CertificateListResponse result = null;
// Deserialize Response
if (statusCode == HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
result = new CertificateListResponse();
JToken responseDoc = null;
if (string.IsNullOrEmpty(responseContent) == false)
{
responseDoc = JToken.Parse(responseContent);
}
if (responseDoc != null && responseDoc.Type != JTokenType.Null)
{
CertificatePaged resultInstance = new CertificatePaged();
result.Result = resultInstance;
JToken valueArray = responseDoc["value"];
if (valueArray != null && valueArray.Type != JTokenType.Null)
{
foreach (JToken valueValue in ((JArray)valueArray))
{
CertificateContract certificateContractInstance = new CertificateContract();
resultInstance.Values.Add(certificateContractInstance);
JToken idValue = valueValue["id"];
if (idValue != null && idValue.Type != JTokenType.Null)
{
string idInstance = ((string)idValue);
certificateContractInstance.IdPath = idInstance;
}
JToken subjectValue = valueValue["subject"];
if (subjectValue != null && subjectValue.Type != JTokenType.Null)
{
string subjectInstance = ((string)subjectValue);
certificateContractInstance.Subject = subjectInstance;
}
JToken thumbprintValue = valueValue["thumbprint"];
if (thumbprintValue != null && thumbprintValue.Type != JTokenType.Null)
{
string thumbprintInstance = ((string)thumbprintValue);
certificateContractInstance.Thumbprint = thumbprintInstance;
}
JToken expirationDateValue = valueValue["expirationDate"];
if (expirationDateValue != null && expirationDateValue.Type != JTokenType.Null)
{
DateTime expirationDateInstance = ((DateTime)expirationDateValue);
certificateContractInstance.ExpirationDate = expirationDateInstance;
}
}
}
JToken countValue = responseDoc["count"];
if (countValue != null && countValue.Type != JTokenType.Null)
{
long countInstance = ((long)countValue);
resultInstance.TotalCount = countInstance;
}
JToken nextLinkValue = responseDoc["nextLink"];
if (nextLinkValue != null && nextLinkValue.Type != JTokenType.Null)
{
string nextLinkInstance = ((string)nextLinkValue);
resultInstance.NextLink = nextLinkInstance;
}
}
}
result.StatusCode = statusCode;
if (httpResponse.Headers.Contains("x-ms-request-id"))
{
result.RequestId = httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
if (shouldTrace)
{
TracingAdapter.Exit(invocationId, result);
}
return(result);
}
finally
{
if (httpResponse != null)
{
httpResponse.Dispose();
}
}
}
finally
{
if (httpRequest != null)
{
httpRequest.Dispose();
}
}
}
/// <param name='resourceGroupName'>
/// Required. The name of the resource group.
/// </param>
/// <param name='serviceName'>
/// Required. The name of the Api Management service.
/// </param>
/// <param name='query'>
/// Optional.
/// </param>
/// <param name='cancellationToken'>
/// Cancellation token.
/// </param>
/// <returns>
/// List Certificates operation response details.
/// </returns>
public async Task <CertificateListResponse> ListAsync(string resourceGroupName, string serviceName, QueryParameters query, CancellationToken cancellationToken)
{
// Validate
if (resourceGroupName == null)
{
throw new ArgumentNullException("resourceGroupName");
}
if (serviceName == null)
{
throw new ArgumentNullException("serviceName");
}
// Tracing
bool shouldTrace = TracingAdapter.IsEnabled;
string invocationId = null;
if (shouldTrace)
{
invocationId = TracingAdapter.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("resourceGroupName", resourceGroupName);
tracingParameters.Add("serviceName", serviceName);
tracingParameters.Add("query", query);
TracingAdapter.Enter(invocationId, this, "ListAsync", tracingParameters);
}
// Construct URL
string url = "";
url = url + "/subscriptions/";
if (this.Client.Credentials.SubscriptionId != null)
{
url = url + Uri.EscapeDataString(this.Client.Credentials.SubscriptionId);
}
url = url + "/resourceGroups/";
url = url + Uri.EscapeDataString(resourceGroupName);
url = url + "/providers/";
url = url + "Microsoft.ApiManagement";
url = url + "/service/";
url = url + Uri.EscapeDataString(serviceName);
url = url + "/certificates";
List <string> queryParameters = new List <string>();
queryParameters.Add("api-version=2016-07-07");
List <string> odataFilter = new List <string>();
if (query != null && query.Filter != null)
{
odataFilter.Add(Uri.EscapeDataString(query.Filter));
}
if (odataFilter.Count > 0)
{
queryParameters.Add("$filter=" + string.Join(null, odataFilter));
}
if (query != null && query.Top != null)
{
queryParameters.Add("$top=" + Uri.EscapeDataString(query.Top.Value.ToString()));
}
if (query != null && query.Skip != null)
{
queryParameters.Add("$skip=" + Uri.EscapeDataString(query.Skip.Value.ToString()));
}
if (queryParameters.Count > 0)
{
url = url + "?" + string.Join("&", queryParameters);
}
string baseUrl = this.Client.BaseUri.AbsoluteUri;
// Trim '/' character from the end of baseUrl and beginning of url.
if (baseUrl[baseUrl.Length - 1] == '/')
{
baseUrl = baseUrl.Substring(0, baseUrl.Length - 1);
}
if (url[0] == '/')
{
url = url.Substring(1);
}
url = baseUrl + "/" + url;
url = url.Replace(" ", "%20");
// Create HTTP transport objects
HttpRequestMessage httpRequest = null;
try
{
httpRequest = new HttpRequestMessage();
httpRequest.Method = HttpMethod.Get;
httpRequest.RequestUri = new Uri(url);
// Set Headers
// Set Credentials
cancellationToken.ThrowIfCancellationRequested();
await this.Client.Credentials.ProcessHttpRequestAsync(httpRequest, cancellationToken).ConfigureAwait(false);
// Send Request
HttpResponseMessage httpResponse = null;
try
{
if (shouldTrace)
{
TracingAdapter.SendRequest(invocationId, httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
httpResponse = await this.Client.HttpClient.SendAsync(httpRequest, cancellationToken).ConfigureAwait(false);
if (shouldTrace)
{
TracingAdapter.ReceiveResponse(invocationId, httpResponse);
}
HttpStatusCode statusCode = httpResponse.StatusCode;
if (statusCode != HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
CloudException ex = CloudException.Create(httpRequest, null, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false));
if (shouldTrace)
{
TracingAdapter.Error(invocationId, ex);
}
throw ex;
}
// Create Result
CertificateListResponse result = null;
// Deserialize Response
if (statusCode == HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
result = new CertificateListResponse();
JToken responseDoc = null;
if (string.IsNullOrEmpty(responseContent) == false)
{
responseDoc = JToken.Parse(responseContent);
}
if (responseDoc != null && responseDoc.Type != JTokenType.Null)
{
CertificatePaged resultInstance = new CertificatePaged();
result.Result = resultInstance;
JToken valueArray = responseDoc["value"];
if (valueArray != null && valueArray.Type != JTokenType.Null)
{
foreach (JToken valueValue in ((JArray)valueArray))
{
CertificateContract certificateContractInstance = new CertificateContract();
resultInstance.Values.Add(certificateContractInstance);
JToken idValue = valueValue["id"];
if (idValue != null && idValue.Type != JTokenType.Null)
{
string idInstance = ((string)idValue);
certificateContractInstance.IdPath = idInstance;
}
JToken subjectValue = valueValue["subject"];
if (subjectValue != null && subjectValue.Type != JTokenType.Null)
{
string subjectInstance = ((string)subjectValue);
certificateContractInstance.Subject = subjectInstance;
}
JToken thumbprintValue = valueValue["thumbprint"];
if (thumbprintValue != null && thumbprintValue.Type != JTokenType.Null)
{
string thumbprintInstance = ((string)thumbprintValue);
certificateContractInstance.Thumbprint = thumbprintInstance;
}
JToken expirationDateValue = valueValue["expirationDate"];
if (expirationDateValue != null && expirationDateValue.Type != JTokenType.Null)
{
DateTime expirationDateInstance = ((DateTime)expirationDateValue);
certificateContractInstance.ExpirationDate = expirationDateInstance;
}
}
}
JToken countValue = responseDoc["count"];
if (countValue != null && countValue.Type != JTokenType.Null)
{
long countInstance = ((long)countValue);
resultInstance.TotalCount = countInstance;
}
JToken nextLinkValue = responseDoc["nextLink"];
if (nextLinkValue != null && nextLinkValue.Type != JTokenType.Null)
{
string nextLinkInstance = ((string)nextLinkValue);
resultInstance.NextLink = nextLinkInstance;
}
}
}
result.StatusCode = statusCode;
if (httpResponse.Headers.Contains("x-ms-request-id"))
{
result.RequestId = httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
if (shouldTrace)
{
TracingAdapter.Exit(invocationId, result);
}
return(result);
}
finally
{
if (httpResponse != null)
{
httpResponse.Dispose();
}
}
}
finally
{
if (httpRequest != null)
{
httpRequest.Dispose();
}
}
}
public static async Task <ISet <AutomationAsset> > GetAll(String localWorkspacePath, AutomationManagementClient automationApi, string resourceGroupName, string automationAccountName, string encryptionCertThumbprint, ICollection <ConnectionType> connectionTypes)
{
CancellationTokenSource cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
VariableListResponse cloudVariables = await automationApi.Variables.ListAsync(resourceGroupName, automationAccountName, cts.Token);
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
CredentialListResponse cloudCredentials = await automationApi.PsCredentials.ListAsync(resourceGroupName, automationAccountName, cts.Token);
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
ConnectionListResponse cloudConnections = await automationApi.Connections.ListAsync(resourceGroupName, automationAccountName, cts.Token);
CertificateListResponse cloudCertificates = await automationApi.Certificates.ListAsync(resourceGroupName, automationAccountName, cts.Token);
// need to get connections one at a time to get each connection's values. Values currently come back as empty in list call
var connectionAssetsWithValues = new HashSet <Connection>();
foreach (var connection in cloudConnections.Connection)
{
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
var connectionResponse = await automationApi.Connections.GetAsync(resourceGroupName, automationAccountName, connection.Name, cts.Token);
connectionAssetsWithValues.Add(connectionResponse.Connection);
}
LocalAssets localAssets = LocalAssetsStore.Get(localWorkspacePath, encryptionCertThumbprint, connectionTypes);
var automationAssets = new SortedSet <AutomationAsset>();
// Compare cloud variables to local
foreach (var cloudAsset in cloudVariables.Variables)
{
var localAsset = localAssets.Variables.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationVariable(localAsset, cloudAsset) :
new AutomationVariable(cloudAsset);
automationAssets.Add(automationAsset);
}
// Add remaining locally created variables
foreach (var localAsset in localAssets.Variables)
{
var automationAsset = new AutomationVariable(localAsset);
automationAssets.Add(automationAsset);
}
// Compare cloud credentials to local
foreach (var cloudAsset in cloudCredentials.Credentials)
{
var localAsset = localAssets.PSCredentials.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationCredential(localAsset, cloudAsset) :
new AutomationCredential(cloudAsset);
automationAssets.Add(automationAsset);
}
// Add remaining locally created credentials
foreach (var localAsset in localAssets.PSCredentials)
{
var automationAsset = new AutomationCredential(localAsset);
automationAssets.Add(automationAsset);
}
// Compare cloud connections to local
foreach (var cloudAsset in connectionAssetsWithValues)
{
ConnectionTypeGetResponse connectionType = await automationApi.ConnectionTypes.GetAsync(resourceGroupName, automationAccountName, cloudAsset.Properties.ConnectionType.Name);
var localAsset = localAssets.Connections.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationConnection(localAsset, cloudAsset) :
new AutomationConnection(cloudAsset, connectionType.ConnectionType);
automationAssets.Add(automationAsset);
}
// Add remaining locally created connections
foreach (var localAsset in localAssets.Connections)
{
var automationAsset = new AutomationConnection(localAsset);
automationAssets.Add(automationAsset);
}
// Compare cloud certificates to local
foreach (var cloudAsset in cloudCertificates.Certificates)
{
var localAsset = localAssets.Certificate.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationCertificate(localAsset, cloudAsset) :
new AutomationCertificate(cloudAsset);
automationAssets.Add(automationAsset);
}
// Add remaining locally created certificates
foreach (var localAsset in localAssets.Certificate)
{
var automationAsset = new AutomationCertificate(localAsset);
automationAssets.Add(automationAsset);
}
return(automationAssets);
}
