public void DataSourceCreateFailsWithoutWorkspace() { BasicDelegatingHandler handler = new BasicDelegatingHandler(); using (var undoContext = UndoContext.Current) { undoContext.Start(); var resourceClient = TestHelper.GetResourceClient(handler); var client = TestHelper.GetOperationalInsightsManagementClient(handler); string resourceGroupName = TestUtilities.GenerateName("OIHyak"); var resourceGroup = TestHelper.CreateResourceGroup(resourceGroupName, resourceClient); // Create a data source on a non-existent workspace string dataSourceName = TestUtilities.GenerateName("AzTestDS"); string workspaceName = TestUtilities.GenerateName("AzTest"); var createParameters = new DataSourceCreateOrUpdateParameters { DataSource = new DataSource { Name = dataSourceName, Kind = "AzureAuditLog", Properties = "{'LinkedResourceId':'/subscriptions/bc8edd8f-a09f-499d-978d-6b5ed2f84852/providers/microsoft.insights/eventtypes/management'}" } }; TestHelper.VerifyCloudException(HttpStatusCode.NotFound, () => client.DataSources.CreateOrUpdate(resourceGroupName, workspaceName, createParameters)); } }
public void CanCreateUpdateDeleteDataSource() { BasicDelegatingHandler handler = new BasicDelegatingHandler(); using (var undoContext = UndoContext.Current) { undoContext.Start(); var resourceClient = TestHelper.GetResourceClient(handler); var client = TestHelper.GetOperationalInsightsManagementClient(handler); string resourceGroupName = TestUtilities.GenerateName("OIHyak"); var resourceGroup = TestHelper.CreateResourceGroup(resourceGroupName, resourceClient); // Create a workspace that will house the data source string workspaceName = TestUtilities.GenerateName("AzTest"); var workspaceCreateParameters = new WorkspaceCreateOrUpdateParameters { Workspace = new Workspace { Name = workspaceName, Location = resourceGroup.Location } }; var workspaceCreateResponse = client.Workspaces.CreateOrUpdate(resourceGroupName, workspaceCreateParameters); Assert.True(HttpStatusCode.Created == workspaceCreateResponse.StatusCode || HttpStatusCode.OK == workspaceCreateResponse.StatusCode); // Create a dataSource string dataSourceName = TestUtilities.GenerateName("AzTestDS"); var createParameters = new DataSourceCreateOrUpdateParameters { DataSource = new DataSource { Name = dataSourceName, Kind = "AzureAuditLog", Properties = "{'LinkedResourceId':'/subscriptions/0b88dfdb-55b3-4fb0-b474-5b6dcbe6b2ef/providers/microsoft.insights/eventtypes/management'}" } }; var createResponse = client.DataSources.CreateOrUpdate(resourceGroupName, workspaceName, createParameters); Assert.True(HttpStatusCode.Created == createResponse.StatusCode || HttpStatusCode.OK == createResponse.StatusCode); TestHelper.ValidateDatasource(createParameters.DataSource, createResponse.DataSource); // Get the data source var getResponse = client.DataSources.Get(resourceGroupName, workspaceName, dataSourceName); Assert.Equal(HttpStatusCode.OK, getResponse.StatusCode); TestHelper.ValidateDatasource(createParameters.DataSource, getResponse.DataSource); // Create a second data source for list testing var dataSourceNameTwo = TestUtilities.GenerateName("AzTestDS"); createParameters.DataSource.Name = dataSourceNameTwo; createParameters.DataSource.Properties = "{'LinkedResourceId':'/subscriptions/a6383be3-f0e8-4968-93d5-10f2625f5bb5/providers/microsoft.insights/eventtypes/management'}"; createResponse = client.DataSources.CreateOrUpdate(resourceGroupName, workspaceName, createParameters); Assert.True(HttpStatusCode.Created == createResponse.StatusCode || HttpStatusCode.OK == createResponse.StatusCode); TestHelper.ValidateDatasource(createParameters.DataSource, createResponse.DataSource); // List the data sources in the workspace var listResponse = client.DataSources.ListInWorkspace(resourceGroupName, workspaceName, "AzureAuditLog", string.Empty); Assert.True(HttpStatusCode.OK == listResponse.StatusCode); Assert.Equal(2, listResponse.DataSources.Count); Assert.Null(listResponse.NextLink); Assert.Single(listResponse.DataSources.Where(w => w.Name.Equals(dataSourceName, StringComparison.OrdinalIgnoreCase))); Assert.Single(listResponse.DataSources.Where(w => w.Name.Equals(dataSourceNameTwo, StringComparison.OrdinalIgnoreCase))); // Perform an update on one of the data sources createResponse.DataSource.Properties = "{'LinkedResourceId':'/subscriptions/bc8edd8f-a09f-499d-978d-6b5ed2f84852/providers/microsoft.insights/eventtypes/management'}"; createResponse.DataSource.Name = dataSourceNameTwo; var updateParameters = new DataSourceCreateOrUpdateParameters { DataSource = createResponse.DataSource }; var updateResponse = client.DataSources.CreateOrUpdate(resourceGroupName, workspaceName, updateParameters); Assert.True(HttpStatusCode.OK == updateResponse.StatusCode); TestHelper.ValidateDatasource(updateParameters.DataSource, updateResponse.DataSource); // Delete a data source var deleteResponse = client.DataSources.Delete(resourceGroupName, workspaceName, dataSourceName); Assert.Equal(HttpStatusCode.OK, deleteResponse.StatusCode); // Verify the data source is gone TestHelper.VerifyCloudException(HttpStatusCode.NotFound, () => client.DataSources.Get(resourceGroupName, workspaceName, dataSourceName)); } }
/// <summary> /// Create or update a data source. /// </summary> /// <param name='operations'> /// Reference to the /// Microsoft.Azure.Management.OperationalInsights.IDataSourceOperations. /// </param> /// <param name='resourceGroupName'> /// Required. The resource group name of the data source. /// </param> /// <param name='workspaceName'> /// Required. The name of the parent workspace that will contain the /// data source /// </param> /// <param name='parameters'> /// Required. The parameters required to create or update a data source. /// </param> /// <returns> /// The create or update data source operation response. /// </returns> public static Task <DataSourceCreateOrUpdateResponse> CreateOrUpdateAsync(this IDataSourceOperations operations, string resourceGroupName, string workspaceName, DataSourceCreateOrUpdateParameters parameters) { return(operations.CreateOrUpdateAsync(resourceGroupName, workspaceName, parameters, CancellationToken.None)); }
/// <summary> /// Create or update a data source. /// </summary> /// <param name='resourceGroupName'> /// Required. The resource group name of the data source. /// </param> /// <param name='workspaceName'> /// Required. The name of the parent workspace that will contain the /// data source /// </param> /// <param name='parameters'> /// Required. The parameters required to create or update a data source. /// </param> /// <param name='cancellationToken'> /// Cancellation token. /// </param> /// <returns> /// The create or update data source operation response. /// </returns> public async Task <DataSourceCreateOrUpdateResponse> CreateOrUpdateAsync(string resourceGroupName, string workspaceName, DataSourceCreateOrUpdateParameters parameters, CancellationToken cancellationToken) { // Validate if (resourceGroupName == null) { throw new ArgumentNullException("resourceGroupName"); } if (workspaceName == null) { throw new ArgumentNullException("workspaceName"); } if (parameters == null) { throw new ArgumentNullException("parameters"); } if (parameters.DataSource != null) { if (parameters.DataSource.Name == null) { throw new ArgumentNullException("parameters.DataSource.Name"); } } // 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("workspaceName", workspaceName); tracingParameters.Add("parameters", parameters); TracingAdapter.Enter(invocationId, this, "CreateOrUpdateAsync", 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/Microsoft.OperationalInsights/workspaces/"; url = url + Uri.EscapeDataString(workspaceName); url = url + "/dataSources/"; if (parameters.DataSource != null && parameters.DataSource.Name != null) { url = url + Uri.EscapeDataString(parameters.DataSource.Name); } List <string> queryParameters = new List <string>(); queryParameters.Add("api-version=2015-11-01-preview"); 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.Put; httpRequest.RequestUri = new Uri(url); // Set Headers httpRequest.Headers.Add("x-ms-client-request-id", Guid.NewGuid().ToString()); // Set Credentials cancellationToken.ThrowIfCancellationRequested(); await this.Client.Credentials.ProcessHttpRequestAsync(httpRequest, cancellationToken).ConfigureAwait(false); // Serialize Request string requestContent = null; JToken requestDoc = null; JObject dataSourceCreateOrUpdateParametersValue = new JObject(); requestDoc = dataSourceCreateOrUpdateParametersValue; if (parameters.DataSource != null) { if (parameters.DataSource.Properties != null) { dataSourceCreateOrUpdateParametersValue["properties"] = JObject.Parse(parameters.DataSource.Properties); } if (parameters.DataSource.ETag != null) { dataSourceCreateOrUpdateParametersValue["eTag"] = parameters.DataSource.ETag; } if (parameters.DataSource.Id != null) { dataSourceCreateOrUpdateParametersValue["id"] = parameters.DataSource.Id; } dataSourceCreateOrUpdateParametersValue["name"] = parameters.DataSource.Name; if (parameters.DataSource.Type != null) { dataSourceCreateOrUpdateParametersValue["type"] = parameters.DataSource.Type; } if (parameters.DataSource.Kind != null) { dataSourceCreateOrUpdateParametersValue["kind"] = parameters.DataSource.Kind; } } requestContent = requestDoc.ToString(Newtonsoft.Json.Formatting.Indented); httpRequest.Content = new StringContent(requestContent, Encoding.UTF8); httpRequest.Content.Headers.ContentType = MediaTypeHeaderValue.Parse("application/json"); // 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 && statusCode != HttpStatusCode.Created) { cancellationToken.ThrowIfCancellationRequested(); CloudException ex = CloudException.Create(httpRequest, requestContent, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false)); if (shouldTrace) { TracingAdapter.Error(invocationId, ex); } throw ex; } // Create Result DataSourceCreateOrUpdateResponse result = null; // Deserialize Response if (statusCode == HttpStatusCode.OK || statusCode == HttpStatusCode.Created) { cancellationToken.ThrowIfCancellationRequested(); string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false); result = new DataSourceCreateOrUpdateResponse(); JToken responseDoc = null; if (string.IsNullOrEmpty(responseContent) == false) { responseDoc = JToken.Parse(responseContent); } if (responseDoc != null && responseDoc.Type != JTokenType.Null) { DataSource dataSourceInstance = new DataSource(); result.DataSource = dataSourceInstance; JToken propertiesValue = responseDoc["properties"]; if (propertiesValue != null && propertiesValue.Type != JTokenType.Null) { string propertiesInstance = propertiesValue.ToString(Newtonsoft.Json.Formatting.Indented); dataSourceInstance.Properties = propertiesInstance; } JToken eTagValue = responseDoc["eTag"]; if (eTagValue != null && eTagValue.Type != JTokenType.Null) { string eTagInstance = ((string)eTagValue); dataSourceInstance.ETag = eTagInstance; } JToken idValue = responseDoc["id"]; if (idValue != null && idValue.Type != JTokenType.Null) { string idInstance = ((string)idValue); dataSourceInstance.Id = idInstance; } JToken nameValue = responseDoc["name"]; if (nameValue != null && nameValue.Type != JTokenType.Null) { string nameInstance = ((string)nameValue); dataSourceInstance.Name = nameInstance; } JToken typeValue = responseDoc["type"]; if (typeValue != null && typeValue.Type != JTokenType.Null) { string typeInstance = ((string)typeValue); dataSourceInstance.Type = typeInstance; } JToken kindValue = responseDoc["kind"]; if (kindValue != null && kindValue.Type != JTokenType.Null) { string kindInstance = ((string)kindValue); dataSourceInstance.Kind = kindInstance; } } } 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(); } } }
/// <summary> /// Create or update a data source. /// </summary> /// <param name='operations'> /// Reference to the /// Microsoft.Azure.Management.OperationalInsights.IDataSourceOperations. /// </param> /// <param name='resourceGroupName'> /// Required. The resource group name of the data source. /// </param> /// <param name='workspaceName'> /// Required. The name of the parent workspace that will contain the /// data source /// </param> /// <param name='parameters'> /// Required. The parameters required to create or update a data source. /// </param> /// <returns> /// The create or update data source operation response. /// </returns> public static DataSourceCreateOrUpdateResponse CreateOrUpdate(this IDataSourceOperations operations, string resourceGroupName, string workspaceName, DataSourceCreateOrUpdateParameters parameters) { return(Task.Factory.StartNew((object s) => { return ((IDataSourceOperations)s).CreateOrUpdateAsync(resourceGroupName, workspaceName, parameters); } , operations, CancellationToken.None, TaskCreationOptions.None, TaskScheduler.Default).Unwrap().GetAwaiter().GetResult()); }