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(), )