public async Task GetIndexList_WithValidInput_ReadsQueryExecutorValidDatabase(string indexName, string databaseName, string tableName)
{
var mockQueryExecutor = new Mock <IQueryExecutor>();
var mockDetails = new Mock <IConnectionDetails>();
mockDetails.SetupGet(d => d.DefaultDatabaseName).Returns(databaseName);
mockQueryExecutor.SetupGet(x => x.DefaultDatabaseName).Returns(mockDetails.Object.DefaultDatabaseName);
var searchString = $"search TableName: '{tableName}' | search DatabaseName: '{databaseName}' |";
using IDataReader stubIndexReader = new DataReaderMock(
new List <Dictionary <string, object> >()
{
new Dictionary <string, object> {
{ "1", "somevalue1" },
},
});
mockQueryExecutor.Setup(exec => exec.ExecuteControlCommandAsync(It.Is <string>(s => s.Contains(searchString, StringComparison.OrdinalIgnoreCase)), It.IsAny <RequestContext>()))
.Returns(Task.FromResult(stubIndexReader));
var kusto = new KustoDataAccess(mockQueryExecutor.Object, It.IsAny <RequestContext>(), new Mock <ILogger <KustoDataAccess> >().Object);
var indexResponse = await kusto.GetIndexListAsync(indexName);
Assert.IsNotNull(indexResponse, $"null response for indexname {indexName}");
var itr = indexResponse.Aggregations.IndexCollection.Buckets.GetEnumerator();
itr.MoveNext();
Assert.NotNull(itr.Current, $"failed to provide valid search term with database name {databaseName} and table name {tableName} from indexname {indexName}. expected: {searchString}");
}
public async Task GetIndexList_ValidIndexAndValidFunction_ReturnBoth()
{
using IDataReader stubIndexReader1 = new DataReaderMock(
new List <Dictionary <string, object> >()
{
new Dictionary <string, object> {
{ "1", "myTable" },
},
});
using IDataReader stubIndexReader2 = new DataReaderMock(
new List <Dictionary <string, object> >()
{
new Dictionary <string, object> {
{ "1", "myFunction" },
},
});
mockQueryExecutor.Setup(exec => exec.ExecuteControlCommandAsync(
It.Is <string>(q => q.StartsWith(".show databases", Ordinal)), It.IsAny <RequestContext>()))
.Returns(Task.FromResult(stubIndexReader1));
mockQueryExecutor.Setup(exec => exec.ExecuteControlCommandAsync(
It.Is <string>(q => q.StartsWith(".show functions", Ordinal)), It.IsAny <RequestContext>()))
.Returns(Task.FromResult(stubIndexReader2));
var kusto = new KustoDataAccess(mockQueryExecutor.Object, It.IsAny <RequestContext>(), new Mock <ILogger <KustoDataAccess> >().Object);
var indexResponse = await kusto.GetIndexListAsync("testIndex");
Assert.IsNotNull(indexResponse);
var itr = indexResponse.Aggregations.IndexCollection.Buckets.GetEnumerator();
Assert.True(itr.MoveNext());
Assert.AreEqual(((TermBucket)itr.Current).Key, "myTable");
Assert.True(itr.MoveNext());
Assert.AreEqual(((TermBucket)itr.Current).Key, "myFunction");
Assert.False(itr.MoveNext());
}
public async Task GetIndexList_WithValidFunction_ReturnIndexList()
{
using IDataReader stubIndexReader = new DataReaderMock(
new List <Dictionary <string, object> >()
{
new Dictionary <string, object> {
{ "1", "somevalue1" },
},
});
mockQueryExecutor.Setup(exec => exec.ExecuteControlCommandAsync(
It.Is <string>(q => q.StartsWith(".show functions", Ordinal)), It.IsAny <RequestContext>()))
.Returns(Task.FromResult(stubIndexReader));
var kusto = new KustoDataAccess(mockQueryExecutor.Object, It.IsAny <RequestContext>(), new Mock <ILogger <KustoDataAccess> >().Object);
var indexResponse = await kusto.GetIndexListAsync("testIndex");
mockQueryExecutor.Verify(exec => exec.ExecuteControlCommandAsync(
".show functions"
+ " | where Parameters == '()'"
+ " | distinct Name"
+ " | search Name: 'testIndex'"
+ " | project strcat(\"\", \":\", Name)", It.IsAny <RequestContext>()));
Assert.IsNotNull(indexResponse);
var itr = indexResponse.Aggregations.IndexCollection.Buckets.GetEnumerator();
itr.MoveNext();
Assert.AreEqual(((TermBucket)itr.Current).Key, "somevalue1");
Assert.False(itr.MoveNext());
}
public void SetUp_DefaultMocks()
{
mockQueryExecutor = new Mock <IQueryExecutor>();
mockDetails = new Mock <IConnectionDetails>();
mockDetails.SetupGet(d => d.DefaultDatabaseName).Returns(string.Empty);
mockQueryExecutor.SetupGet(x => x.DefaultDatabaseName).Returns(mockDetails.Object.DefaultDatabaseName);
using IDataReader emptyReader = new DataReaderMock(new List <Dictionary <string, object> >());
mockQueryExecutor.Setup(exec => exec.ExecuteControlCommandAsync(It.IsNotNull <string>(), It.IsAny <RequestContext>()))
.Returns(Task.FromResult(emptyReader));
using IDataReader emptyReader2 = new DataReaderMock(new List <Dictionary <string, object> >());
mockQueryExecutor.Setup(exec => exec.ExecuteQueryAsync(It.IsNotNull <QueryData>(), It.IsAny <RequestContext>()))
.Returns(Task.FromResult((TimeSpan.FromSeconds(1), emptyReader2)));
}
public async Task GetFieldCaps_WithValidFunction_ReturnFieldCaps()
{
Func <string, string, Dictionary <string, object> > column = (name, type) =>
new Dictionary <string, object> {
{ "ColumnName", name },
{ "ColumnType", type },
};
var testData = new List <Dictionary <string, object> >()
{
column("myint", "System.Int32"),
column("mystring", "System.String"),
};
using var testReader = new DataReaderMock(testData);
mockQueryExecutor.Setup(exec => exec.ExecuteQueryAsync(It.IsAny <QueryData>(), It.IsAny <RequestContext>()))
.Returns(Task.FromResult((TimeSpan.FromSeconds(1), (IDataReader)testReader)));
var kusto = new KustoDataAccess(mockQueryExecutor.Object, It.IsAny <RequestContext>(), new Mock <ILogger <KustoDataAccess> >().Object);
var response = await kusto.GetFieldCapsAsync("testIndexName");
mockQueryExecutor.Verify(exec => exec.ExecuteQueryAsync(
It.Is <QueryData>(d =>
d.IndexName == "testIndexName" &&
d.QueryCommandText == "testIndexName | getschema | project ColumnName, ColumnType=DataType"), It.IsAny <RequestContext>()));
JToken.FromObject(response).Should().BeEquivalentTo(JToken.Parse(@"
{
""fields"": {
""myint"": {
""integer"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""integer""
}
},
""mystring"": {
""keyword"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""keyword""
}
}
}
}
"));
}
public async Task GetFieldCaps_WithValidIndex_ReturnFieldCaps()
{
Func <string, string, Dictionary <string, object> > column = (name, type) =>
new Dictionary <string, object> {
{ "ColumnName", name },
{ "ColumnType", type },
};
var testData = new List <Dictionary <string, object> >()
{
column("mybool", "System.SByte"),
column("myint", "System.Int32"),
column("mylong", "System.Int64"),
column("myreal", "System.Double"),
column("mystring", "System.String"),
column("mydatetime", "System.DateTime"),
column("mydynamic", "System.Object"),
column("myguid", "System.Guid"),
column("mytimespan", "System.TimeSpan"),
column("mydecimal", "System.Data.SqlTypes.SqlDecimal"),
};
using IDataReader testReader = new DataReaderMock(testData);
mockQueryExecutor.Setup(exec => exec.ExecuteControlCommandAsync(It.IsNotNull <string>(), It.IsAny <RequestContext>()))
.Returns(Task.FromResult(testReader));
var kusto = new KustoDataAccess(mockQueryExecutor.Object, It.IsAny <RequestContext>(), new Mock <ILogger <KustoDataAccess> >().Object);
var response = await kusto.GetFieldCapsAsync("testIndexName");
JToken.FromObject(response).Should().BeEquivalentTo(JToken.Parse(@"
{
""fields"": {
""mybool"": {
""boolean"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""boolean""
}
},
""myint"": {
""integer"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""integer""
}
},
""mylong"": {
""long"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""long""
}
},
""myreal"": {
""double"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""double""
}
},
""mystring"": {
""keyword"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""keyword""
}
},
""mydatetime"": {
""date"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""date""
}
},
""mydynamic"": {
""object"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""object""
}
},
""myguid"": {
""string"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""string""
}
},
""mytimespan"": {
""string"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""string""
}
},
""mydecimal"": {
""double"": {
""aggregatable"": true,
""searchable"": true,
""type"": ""double""
}
}
}
}
"));
}
