More vector test types

Author: lukasz-antoniak

cassandra-analytics-core/src/test/java/org/apache/cassandra/spark/endtoend/DataTypeTests.java

@@ -196,9 +196,12 @@ public void testVectorTuple(CassandraBridge bridge)
 
     private static Gen<CqlField.NativeType> vectorSupportedTypes(CassandraBridge bridge)
     {
-        List<CqlField.NativeType> supportedTypes = List.of(bridge.aInt(), bridge.bigint(), bridge.aFloat(),
-                                                           bridge.aDouble(), bridge.smallint(), bridge.tinyint(),
-                                                           bridge.decimal());
+        // TODO: Vector of durations fail, because we cannot replace DurationSerializer with
+        //  AnalyticsDurationSerializer across all serializers used by VectorType.
+        // TODO: Fix for CASSANDRA-20979 required.
+        List<CqlField.NativeType> supportedTypes = bridge.supportedTypes().stream()
+                                                         .filter(t -> !t.equals(bridge.date()) && !t.equals(bridge.time()) && !t.equals(bridge.duration()))
+                                                         .collect(Collectors.toList());
         return arbitrary().pick(supportedTypes);
     }