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); }