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