Skip to content

Commit d3c52f1

Browse files
authored
Improve Postgres autodiscovery test performance (DataDog#21459)
* Improve Postgres autodiscovery test performance * Clean * Clean
1 parent f72a071 commit d3c52f1

1 file changed

Lines changed: 0 additions & 29 deletions

File tree

postgres/tests/test_discovery.py

Lines changed: 0 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,6 @@ def test_autodiscovery_simple(integration_check, pg_instance):
8787
Test simple autodiscovery.
8888
"""
8989
pg_instance["database_autodiscovery"] = DISCOVERY_CONFIG
90-
pg_instance['relations'] = ['pg_index']
9190
del pg_instance['dbname']
9291
check = integration_check(pg_instance)
9392
run_one_check(check, pg_instance)
@@ -106,7 +105,6 @@ def test_autodiscovery_global_view_db_specified(integration_check, pg_instance):
106105
"""
107106
pg_instance["database_autodiscovery"] = copy.deepcopy(DISCOVERY_CONFIG)
108107
pg_instance["database_autodiscovery"]["global_view_db"] = "dogs_0"
109-
pg_instance['relations'] = ['pg_index']
110108
del pg_instance['dbname']
111109
check = integration_check(pg_instance)
112110
run_one_check(check, pg_instance)
@@ -125,7 +123,6 @@ def test_autodiscovery_max_databases(integration_check, pg_instance):
125123
"""
126124
pg_instance["database_autodiscovery"] = copy.deepcopy(DISCOVERY_CONFIG)
127125
pg_instance['database_autodiscovery']['max_databases'] = 20
128-
pg_instance['relations'] = ['pg_index']
129126
del pg_instance['dbname']
130127

131128
check = integration_check(pg_instance)
@@ -158,7 +155,6 @@ def test_autodiscovery_refresh(integration_check, pg_instance):
158155

159156
pg_instance["database_autodiscovery"] = copy.deepcopy(DISCOVERY_CONFIG)
160157
pg_instance['database_autodiscovery']['include'].append(database_to_find)
161-
pg_instance['relations'] = ['pg_index']
162158
del pg_instance['dbname']
163159
pg_instance["database_autodiscovery"]['refresh'] = 1
164160
check = integration_check(pg_instance)
@@ -184,21 +180,6 @@ def test_autodiscovery_refresh(integration_check, pg_instance):
184180
)
185181

186182

187-
@pytest.mark.integration
188-
@pytest.mark.usefixtures('dd_environment')
189-
def test_autodiscovery_relations_disabled(integration_check, pg_instance):
190-
"""
191-
If no relation metrics are being collected, autodiscovery should still run.
192-
"""
193-
pg_instance["database_autodiscovery"] = DISCOVERY_CONFIG
194-
pg_instance['relations'] = []
195-
del pg_instance['dbname']
196-
check = integration_check(pg_instance)
197-
run_one_check(check, pg_instance)
198-
199-
assert check.autodiscovery is not None
200-
201-
202183
@pytest.mark.integration
203184
@pytest.mark.usefixtures('dd_environment')
204185
def test_autodiscovery_collect_all_metrics(aggregator, integration_check, pg_instance):
@@ -265,18 +246,12 @@ def test_autodiscovery_exceeds_min_interval(aggregator, integration_check, pg_in
265246
"""
266247
pg_instance["database_autodiscovery"] = copy.deepcopy(DISCOVERY_CONFIG)
267248
pg_instance["database_autodiscovery"]["include"] = ["dogs$", "dogs_noschema$", "dogs_nofunc$"]
268-
pg_instance['relations'] = [
269-
{'relation_regex': '.*'},
270-
]
271249
pg_instance['min_collection_interval'] = 0.001
272250
del pg_instance['dbname']
273251

274252
check = integration_check(pg_instance)
275253
check.check(pg_instance)
276254

277-
aggregator.assert_metric(
278-
'dd.postgres._collect_relations_autodiscovery.time',
279-
)
280255
assert len(check.warnings) == 1
281256
test_structure = re.compile(
282257
"Collecting metrics on autodiscovery metrics took .* ms, which is longer than "
@@ -293,7 +268,6 @@ def test_autodiscovery_dbname_specified(integration_check, pg_instance):
293268
If a dbname is specified in the config, autodiscovery should not run.
294269
"""
295270
pg_instance["database_autodiscovery"] = DISCOVERY_CONFIG
296-
pg_instance['relations'] = ['breed']
297271
pg_instance['dbname'] = "dogs_30"
298272

299273
with pytest.raises(ConfigurationError):
@@ -315,9 +289,6 @@ def test_handle_cannot_connect(aggregator, integration_check, pg_instance):
315289
db_to_disable = "dogs_0"
316290
_set_allow_connection(db_to_disable, False)
317291
pg_instance["database_autodiscovery"] = {"enabled": True, "include": ["dogs_[0-3]"]}
318-
pg_instance['relations'] = [
319-
{'relation_regex': '.*'},
320-
]
321292
del pg_instance['dbname']
322293
check = integration_check(pg_instance)
323294
run_one_check(check, pg_instance)

0 commit comments

Comments
 (0)