def test_add_queries_with_table_name(self):
query_a = Query('mock_table').where(col.numField > 4)
query_b = Query().take(5)
query = query_a + query_b
self.assertEqual(
"mock_table | where numField > 4 | take 5",
query.render(),
)
self.assertEqual(
"mock_table",
query.get_table_name(),
)
# make sure the originals didn't change
self.assertEqual(
"mock_table | where numField > 4",
query_a.render(),
)
self.assertEqual(
"mock_table",
query_a.get_table_name(),
)
self.assertEqual(
" | take 5",
query_b.render(),
)
self.assertEqual(
None,
query_b.get_table_name(),
)
def test_add_queries(self):
query_a = Query(t).where(t.numField > 4)
query_b = Query(t).take(5)
query_c = Query(t).where(t.numField2 > 1).sort_by(
t.stringField, Order.ASC, Nulls.LAST)
query = query_a + query_b + query_c
self.assertEqual(
"mock_table | where numField > 4 | take 5 | where numField2 > 1 | sort by stringField asc nulls last",
query.render(),
)
# make sure the originals didn't change
self.assertEqual(
"mock_table | where numField > 4",
query_a.render(),
)
self.assertEqual(
"mock_table | take 5",
query_b.render(),
)
self.assertEqual(
"mock_table | where numField2 > 1 | sort by stringField asc nulls last",
query_c.render(),
)
def test_add_queries_with_table(self):
table = PyKustoClient(
MockKustoClient(columns_response=mock_columns_response([(
'numField', _KustoType.INT)])))['test_db']['mock_table']
query_a = Query(table).where(table.numField > 4)
query_b = Query(t).take(5).take(2).sort_by(t.stringField, Order.ASC,
Nulls.LAST)
query = query_a + query_b
self.assertEqual(
"mock_table | where numField > 4 | take 5 | take 2 | sort by stringField asc nulls last",
query.render(),
)
# make sure the originals didn't change
self.assertEqual(
"mock_table | where numField > 4",
query_a.render(),
)
self.assertEqual(
"mock_table | take 5 | take 2 | sort by stringField asc nulls last",
query_b.render(),
)
