Add Duration sample to getFixedValue, filter Duration from UserTypesTest

tolbertam · tolbertam · commit 0d0119cdeb90 · 2017-02-13T14:22:44.000-06:00
diff --git a/driver-core/src/test/java/com/datastax/driver/core/TestUtils.java b/driver-core/src/test/java/com/datastax/driver/core/TestUtils.java
@@ -347,6 +347,8 @@ public static Object getFixedValue(final DataType type) {
                     return true;
                 case COUNTER:
                     throw new UnsupportedOperationException("Cannot 'getSomeValue' for counters");
+                case DURATION:
+                    return Duration.from("1h20m3s");
                 case DECIMAL:
                     return new BigDecimal("3.1415926535897932384626433832795028841971693993751058209749445923078164062862089986280348253421170679");
                 case DOUBLE:
diff --git a/driver-core/src/test/java/com/datastax/driver/core/UserTypesTest.java b/driver-core/src/test/java/com/datastax/driver/core/UserTypesTest.java
@@ -253,6 +253,8 @@ public void testPrimitiveDatatypes() throws Exception {
                 case DOUBLE:
                     alldatatypes.setDouble(index, ((Double) sampleData).doubleValue());
                     break;
+                case DURATION:
+                    alldatatypes.set(index, Duration.from(sampleData.toString()), Duration.class);
                 case FLOAT:
                     alldatatypes.setFloat(index, ((Float) sampleData).floatValue());
                     break;
@@ -322,8 +324,9 @@ public void testNonPrimitiveDatatypes() throws Exception {
                 "WITH replication = { 'class' : 'SimpleStrategy', 'replication_factor': '1'}");
         session().execute("USE test_nonprimitive_datatypes");
 
-        // counters are not allowed inside collections
+        // counters and durations are not allowed inside collections
         DATA_TYPE_PRIMITIVES.remove(DataType.counter());
+        DATA_TYPE_PRIMITIVES.remove(DataType.duration());
 
         // create UDT
         List<String> alpha_type_list = new ArrayList<String>();
