|
34 | 34 | SqlserverServerStateMetrics,
|
35 | 35 | SqlserverTempDBFileSpaceUsageMetrics,
|
36 | 36 | )
|
| 37 | +from datadog_checks.sqlserver.utils import Database |
37 | 38 |
|
38 | 39 | from .common import (
|
39 | 40 | CHECK_NAME,
|
@@ -1491,3 +1492,35 @@ def test_sqlserver_database_metrics_defaults(
|
1491 | 1492 | databases=AUTODISCOVERY_DBS,
|
1492 | 1493 | )
|
1493 | 1494 | assert database_metrics.enabled == enabled
|
| 1495 | + |
| 1496 | + |
| 1497 | +@pytest.mark.unit |
| 1498 | +def test_sqlserver_database_metrics_init( |
| 1499 | + init_config, |
| 1500 | + instance_docker_metrics, |
| 1501 | +): |
| 1502 | + instance_docker_metrics['database_autodiscovery'] = True |
| 1503 | + instance_docker_metrics['database_metrics'] = { |
| 1504 | + 'index_usage_metrics': { |
| 1505 | + 'enabled': True, |
| 1506 | + 'enabled_tempdb': False, |
| 1507 | + }, |
| 1508 | + } |
| 1509 | + |
| 1510 | + sqlserver_check = SQLServer(CHECK_NAME, init_config, [instance_docker_metrics]) |
| 1511 | + sqlserver_check.databases = {Database(db) for db in AUTODISCOVERY_DBS + ['tempdb']} |
| 1512 | + assert sqlserver_check.database_metrics is not None |
| 1513 | + |
| 1514 | + index_usage_metrics = [ |
| 1515 | + metric for metric in sqlserver_check.database_metrics if isinstance(metric, SqlserverIndexUsageMetrics) |
| 1516 | + ] |
| 1517 | + assert len(index_usage_metrics) == 1 |
| 1518 | + assert index_usage_metrics[0].enabled is True |
| 1519 | + assert 'tempdb' not in index_usage_metrics[0].databases |
| 1520 | + |
| 1521 | + database_file_metrics = [ |
| 1522 | + metric for metric in sqlserver_check.database_metrics if isinstance(metric, SqlserverDatabaseFilesMetrics) |
| 1523 | + ] |
| 1524 | + assert len(database_file_metrics) == 1 |
| 1525 | + assert database_file_metrics[0].enabled is True |
| 1526 | + assert 'tempdb' in database_file_metrics[0].databases |
0 commit comments