From 08a95103bdb9d1192186ff08f6192e31f65fafef Mon Sep 17 00:00:00 2001 From: Gian Merlino Date: Wed, 21 Aug 2013 14:45:24 -0700 Subject: [PATCH] More NumberedShardSpec serde tests --- .../druid/shard/NumberedShardSpecTest.java | 22 +++++++++++++++---- 1 file changed, 18 insertions(+), 4 deletions(-) diff --git a/client/src/test/java/com/metamx/druid/shard/NumberedShardSpecTest.java b/client/src/test/java/com/metamx/druid/shard/NumberedShardSpecTest.java index 06707df6dc1..7f3680de8a0 100644 --- a/client/src/test/java/com/metamx/druid/shard/NumberedShardSpecTest.java +++ b/client/src/test/java/com/metamx/druid/shard/NumberedShardSpecTest.java @@ -33,12 +33,26 @@ import java.util.List; public class NumberedShardSpecTest { @Test - public void testSerde() throws Exception + public void testSerdeRoundTrip() throws Exception { final ObjectMapper jsonMapper = new DefaultObjectMapper(); - final ShardSpec spec = new NumberedShardSpec(1, 2); - final ShardSpec spec2 = jsonMapper.readValue(jsonMapper.writeValueAsBytes(spec), ShardSpec.class); - Assert.assertEquals(1, spec2.getPartitionNum()); + final ShardSpec spec = jsonMapper.readValue( + jsonMapper.writeValueAsBytes(new NumberedShardSpec(1, 2)), + ShardSpec.class + ); + Assert.assertEquals(1, spec.getPartitionNum()); + Assert.assertEquals(2, ((NumberedShardSpec) spec).getPartitions()); + } + + @Test + public void testSerdeBackwardsCompat() throws Exception + { + final ObjectMapper jsonMapper = new DefaultObjectMapper(); + final ShardSpec spec = jsonMapper.readValue( + "{\"type\": \"numbered\", \"partition\": 1, \"partitionNum\": 2}", + ShardSpec.class + ); + Assert.assertEquals(1, spec.getPartitionNum()); Assert.assertEquals(2, ((NumberedShardSpec) spec).getPartitions()); }