Add tests

Author: jasonmp85

postgres/datadog_checks/postgres/metadata.py

@@ -291,6 +291,7 @@ def run_job(self):
         self.tags = [t for t in self._tags if not t.startswith("dd.internal")]
         self._tags_no_db = [t for t in self.tags if not t.startswith("db:")]
         self.report_postgres_metadata()
+        self.report_postgres_extensions()
         self._check.db_pool.prune_connections()
 
     @tracked_method(agent_check_getter=agent_check_getter)

postgres/tests/test_metadata.py

@@ -34,6 +34,20 @@ def stop_orphaned_threads():
     DBMAsyncJob.executor = ThreadPoolExecutor()
 
 
+def test_collect_extensions(integration_check, dbm_instance, aggregator):
+    check = integration_check(dbm_instance)
+    check.check(dbm_instance)
+    dbm_metadata = aggregator.get_event_platform_events("dbm-metadata")
+    event = next((e for e in dbm_metadata if e['kind'] == 'pg_extension'), None)
+    assert event is not None
+    assert event['host'] == "stubbed.hostname"
+    assert event['dbms'] == "postgres"
+    assert event['kind'] == "pg_extension"
+    assert len(event["metadata"]) > 0
+    assert set(event["metadata"][0].keys()) == {'id', 'name', 'owner', 'relocatable', 'schema_name', 'version'}
+    assert next((k for k in event['metadata'] if k['name'].startswith('plpgsql')), None) is not None
+
+
 def test_collect_metadata(integration_check, dbm_instance, aggregator):
     dbm_instance["collect_settings"]['ignored_settings_patterns'] = ['max_wal%']
     check = integration_check(dbm_instance)