Tests: Change the way tablet support is checked

Fixes: #1048

See the issue description for more information.

scylla/src/utils/test_utils.rs

@@ -89,17 +89,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();
-    result.single_row().is_ok()
+    supports_feature(session, "TABLETS").await
 }
 
 pub(crate) fn setup_tracing() {

scylla/tests/integration/utils.rs

@@ -119,17 +119,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();
-    result.single_row().is_ok()
+    supports_feature(session, "TABLETS").await
 }
 
 // Creates a generic session builder based on conditional compilation configuration