mirror of
https://github.com/getredash/redash.git
synced 2025-12-19 17:37:19 -05:00
5d31429ca8
* Update Azure Data Explorer query runner to latest version * Fix black issue * downgrade azure-kusto-data to 4.6.3 * Freeze numpy to 1.24.4 because of 2.0.0 incompatibility * Fix failing test * Reformat test
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
from unittest import TestCase
|
|
from unittest.mock import patch
|
|
|
|
from redash.query_runner.azure_kusto import AzureKusto
|
|
|
|
|
|
class TestAzureKusto(TestCase):
|
|
def setUp(self):
|
|
self.configuration = {
|
|
"cluster": "https://example.kusto.windows.net",
|
|
"database": "sample_db",
|
|
"azure_ad_client_id": "client_id",
|
|
"azure_ad_client_secret": "client_secret",
|
|
"azure_ad_tenant_id": "tenant_id",
|
|
}
|
|
self.kusto = AzureKusto(self.configuration)
|
|
|
|
@patch.object(AzureKusto, "run_query")
|
|
def test_get_schema(self, mock_run_query):
|
|
mock_response = {
|
|
"rows": [
|
|
{
|
|
"DatabaseSchema": '{"Databases":{"sample_db":{"Tables":{"Table1":{"Name":"Table1","OrderedColumns":[{"Name":"Column1","Type":"System.String","CslType":"string"},{"Name":"Column2","Type":"System.DateTime","CslType":"datetime"}]}},"MaterializedViews":{"View1":{"Name":"View1","OrderedColumns":[{"Name":"Column1","Type":"System.String","CslType":"string"},{"Name":"Column2","Type":"System.DateTime","CslType":"datetime"}]}}}}}'
|
|
}
|
|
]
|
|
}
|
|
mock_run_query.return_value = (mock_response, None)
|
|
|
|
expected_schema = [
|
|
{
|
|
"name": "Table1",
|
|
"columns": [{"name": "Column1", "type": "string"}, {"name": "Column2", "type": "datetime"}],
|
|
},
|
|
{
|
|
"name": "View1",
|
|
"columns": [{"name": "Column1", "type": "string"}, {"name": "Column2", "type": "datetime"}],
|
|
},
|
|
]
|
|
|
|
schema = self.kusto.get_schema()
|
|
print(schema)
|
|
self.assertEqual(schema, expected_schema)
|