Tests: Change the way tablet support is checked

Fixes: #1048

See the issue description for more information.

scylla/src/utils/test_utils.rs

@@ -92,19 +92,7 @@ pub(crate) fn create_new_session_builder() -> GenericSessionBuilder<impl Session
 }
 
 pub(crate) async fn scylla_supports_tablets(session: &Session) -> bool {
-    let result = session
-        .query_unpaged(
-            "select column_name from system_schema.columns where
-                keyspace_name = 'system_schema'
-                and table_name = 'scylla_keyspaces'
-                and column_name = 'initial_tablets'",
-            &[],
-        )
-        .await
-        .unwrap()
-        .into_rows_result();
-
-    result.is_ok_and(|rows_result| rows_result.single_row::<Row>().is_ok())
+    supports_feature(session, "TABLETS").await
 }
 
 pub(crate) fn setup_tracing() {

scylla/tests/integration/utils.rs

@@ -123,19 +123,7 @@ pub(crate) async fn supports_feature(session: &Session, feature: &str) -> bool {
 
 #[allow(unused)]
 pub(crate) async fn scylla_supports_tablets(session: &Session) -> bool {
-    let result = session
-        .query_unpaged(
-            "select column_name from system_schema.columns where
-                keyspace_name = 'system_schema'
-                and table_name = 'scylla_keyspaces'
-                and column_name = 'initial_tablets'",
-            &[],
-        )
-        .await
-        .unwrap()
-        .into_rows_result();
-
-    result.is_ok_and(|rows_result| rows_result.single_row::<Row>().is_ok())
+    supports_feature(session, "TABLETS").await
 }
 
 // Creates a generic session builder based on conditional compilation configuration