mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-10-01 20:04:52 +00:00
15b92735b9
* Fix #1093: Add Grafana Support * Update generated TypeScript types * Grafana test fix * Update --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: Akash Verma <akashverma@Mac.lan>
135 lines
4.9 KiB
Python
135 lines
4.9 KiB
Python
# Copyright 2025 Collate
|
|
# Licensed under the Collate Community License, Version 1.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
# https://github.com/open-metadata/OpenMetadata/blob/main/ingestion/LICENSE
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""
|
|
Simple unit tests for Grafana connector components
|
|
"""
|
|
|
|
from unittest import TestCase
|
|
from unittest.mock import MagicMock
|
|
|
|
from metadata.ingestion.source.dashboard.grafana.client import GrafanaApiClient
|
|
from metadata.ingestion.source.dashboard.grafana.metadata import GrafanaSource
|
|
from metadata.ingestion.source.dashboard.grafana.models import GrafanaDashboardResponse
|
|
|
|
|
|
class TestGrafanaComponents(TestCase):
|
|
"""Test individual Grafana components"""
|
|
|
|
def test_panel_type_mapping(self):
|
|
"""Test Grafana panel type to OpenMetadata chart type mapping"""
|
|
# Create a minimal instance just for testing the method
|
|
source = MagicMock(spec=GrafanaSource)
|
|
|
|
# Add the method we want to test
|
|
source._map_panel_type_to_chart_type = (
|
|
GrafanaSource._map_panel_type_to_chart_type.__get__(source)
|
|
)
|
|
|
|
test_cases = {
|
|
"graph": "Line",
|
|
"timeseries": "Line",
|
|
"table": "Table",
|
|
"stat": "Text",
|
|
"gauge": "Gauge",
|
|
"bargauge": "Bar",
|
|
"bar": "Bar",
|
|
"piechart": "Pie",
|
|
"heatmap": "Heatmap",
|
|
"histogram": "Histogram",
|
|
"geomap": "Map",
|
|
"nodeGraph": "Graph",
|
|
"unknown": "Other",
|
|
}
|
|
|
|
for panel_type, expected_chart_type in test_cases.items():
|
|
result = source._map_panel_type_to_chart_type(panel_type)
|
|
# The method returns an enum value, compare its value
|
|
self.assertEqual(result.value, expected_chart_type)
|
|
|
|
def test_extract_datasource_name(self):
|
|
"""Test datasource name extraction"""
|
|
source = MagicMock(spec=GrafanaSource)
|
|
source._extract_datasource_name = (
|
|
GrafanaSource._extract_datasource_name.__get__(source)
|
|
)
|
|
|
|
# Test with string datasource in target
|
|
target = MagicMock()
|
|
target.datasource = "postgres-uid"
|
|
panel = MagicMock()
|
|
panel.datasource = None
|
|
|
|
result = source._extract_datasource_name(target, panel)
|
|
self.assertEqual(result, "postgres-uid")
|
|
|
|
# Test with dict datasource in target
|
|
target.datasource = {"uid": "postgres-uid", "type": "postgres"}
|
|
result = source._extract_datasource_name(target, panel)
|
|
self.assertEqual(result, "postgres-uid")
|
|
|
|
# Test fallback to panel datasource
|
|
target.datasource = None
|
|
panel.datasource = "panel-datasource"
|
|
result = source._extract_datasource_name(target, panel)
|
|
self.assertEqual(result, "panel-datasource")
|
|
|
|
def test_dashboard_response_parsing(self):
|
|
"""Test parsing of dashboard response"""
|
|
dashboard_data = {
|
|
"dashboard": {
|
|
"uid": "test-uid",
|
|
"title": "Test Dashboard",
|
|
"tags": ["test", "demo"],
|
|
"panels": [
|
|
{
|
|
"id": 1,
|
|
"type": "graph",
|
|
"title": "Test Panel",
|
|
}
|
|
],
|
|
},
|
|
"meta": {
|
|
"type": "db",
|
|
"canSave": True,
|
|
"canEdit": True,
|
|
"canAdmin": True,
|
|
"canStar": True,
|
|
"canDelete": True,
|
|
"slug": "test-dashboard",
|
|
"url": "/d/test-uid/test-dashboard",
|
|
},
|
|
}
|
|
|
|
response = GrafanaDashboardResponse(**dashboard_data)
|
|
self.assertEqual(response.dashboard.uid, "test-uid")
|
|
self.assertEqual(response.dashboard.title, "Test Dashboard")
|
|
self.assertEqual(len(response.dashboard.panels), 1)
|
|
self.assertEqual(response.meta.slug, "test-dashboard")
|
|
|
|
def test_api_client_initialization(self):
|
|
"""Test API client initialization"""
|
|
client = GrafanaApiClient(
|
|
host_port="https://grafana.example.com",
|
|
api_key="test_key",
|
|
verify_ssl=True,
|
|
page_size=50,
|
|
)
|
|
|
|
self.assertEqual(client.host_port, "https://grafana.example.com")
|
|
self.assertEqual(client.page_size, 50)
|
|
self.assertTrue(client.verify_ssl)
|
|
|
|
# Test session headers
|
|
session = client.session
|
|
self.assertEqual(session.headers["Authorization"], "Bearer test_key")
|
|
self.assertEqual(session.headers["Accept"], "application/json")
|