@@ -29,6 +29,8 @@ public class TimePartitioningTest {
29
29
30
30
private static final Type TYPE_DAY = Type .DAY ;
31
31
private static final Type TYPE_HOUR = Type .HOUR ;
32
+ private static final Type TYPE_MONTH = Type .MONTH ;
33
+ private static final Type TYPE_YEAR = Type .YEAR ;
32
34
private static final long EXPIRATION_MS = 42 ;
33
35
private static final boolean REQUIRE_PARTITION_FILTER = false ;
34
36
private static final String FIELD = "field" ;
@@ -44,11 +46,25 @@ public class TimePartitioningTest {
44
46
.setRequirePartitionFilter (REQUIRE_PARTITION_FILTER )
45
47
.setField (FIELD )
46
48
.build ();
49
+ private static final TimePartitioning TIME_PARTITIONING_MONTH =
50
+ TimePartitioning .newBuilder (TYPE_MONTH )
51
+ .setExpirationMs (EXPIRATION_MS )
52
+ .setRequirePartitionFilter (REQUIRE_PARTITION_FILTER )
53
+ .setField (FIELD )
54
+ .build ();
55
+ private static final TimePartitioning TIME_PARTITIONING_YEAR =
56
+ TimePartitioning .newBuilder (TYPE_YEAR )
57
+ .setExpirationMs (EXPIRATION_MS )
58
+ .setRequirePartitionFilter (REQUIRE_PARTITION_FILTER )
59
+ .setField (FIELD )
60
+ .build ();
47
61
48
62
@ Test
49
63
public void testOf () {
50
64
assertEquals (TYPE_DAY , TIME_PARTITIONING_DAY .getType ());
51
65
assertEquals (TYPE_HOUR , TIME_PARTITIONING_HOUR .getType ());
66
+ assertEquals (TYPE_MONTH , TIME_PARTITIONING_MONTH .getType ());
67
+ assertEquals (TYPE_YEAR , TIME_PARTITIONING_YEAR .getType ());
52
68
assertEquals (EXPIRATION_MS , TIME_PARTITIONING_DAY .getExpirationMs ().longValue ());
53
69
assertEquals (REQUIRE_PARTITION_FILTER , TIME_PARTITIONING_DAY .getRequirePartitionFilter ());
54
70
assertEquals (FIELD , TIME_PARTITIONING_DAY .getField ());
0 commit comments