diff --git a/x-pack/metricbeat/module/sql/query/test_sql_mssql.py b/x-pack/metricbeat/module/sql/query/test_sql_mssql.py index 368a1b7c59a..24b6804413f 100644 --- a/x-pack/metricbeat/module/sql/query/test_sql_mssql.py +++ b/x-pack/metricbeat/module/sql/query/test_sql_mssql.py @@ -11,23 +11,23 @@ class Test(XPackTest): @unittest.skipUnless(metricbeat.INTEGRATION_TESTS, "integration test") def test_query_without_fetch_from_all_databases(self): """ - SQL MSSQL custom query with fetch_from_all_databases="false" + SQL MSSQL custom query with fetch_from_all_databases=false 1 document will be received from the default selected database: 'master'. """ - self.run_query_test(fetch_from_all_databases="false", expected_output_count=1) + self.run_query_test(fetch_from_all_databases=False, expected_output_count=1) @unittest.skipUnless(metricbeat.INTEGRATION_TESTS, "integration test") def test_query_with_fetch_from_all_databases(self): """ - SQL MSSQL custom query with fetch_from_all_databases="true" + SQL MSSQL custom query with fetch_from_all_databases=true 4 documents will be received because 4 databases are present by default in MSSQL and 1 document per database will received. """ - self.run_query_test(fetch_from_all_databases="true", expected_output_count=4) + self.run_query_test(fetch_from_all_databases=True, expected_output_count=4) - def run_query_test(self, fetch_from_all_databases: str, expected_output_count: int) -> list: + def run_query_test(self, fetch_from_all_databases: bool, expected_output_count: int) -> list: self.render_config_template(modules=[{ "name": "sql", "metricsets": ["query"], @@ -35,7 +35,7 @@ def run_query_test(self, fetch_from_all_databases: str, expected_output_count: i "period": "5s", "additional_content": f""" driver: mssql - fetch_from_all_databases: {fetch_from_all_databases} + fetch_from_all_databases: {str(fetch_from_all_databases).lower()} sql_query: SELECT DB_NAME() AS 'database_name'; sql_response_format: table""" }])