| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.gossip.protocol.json; |
| |
| import com.codahale.metrics.MetricRegistry; |
| import com.fasterxml.jackson.databind.ObjectMapper; |
| import org.apache.gossip.GossipSettings; |
| import org.apache.gossip.Member; |
| import org.apache.gossip.crdt.OrSet; |
| import org.apache.gossip.manager.GossipManager; |
| import org.apache.gossip.manager.GossipManagerBuilder; |
| import org.apache.gossip.protocol.ProtocolManager; |
| import org.junit.Assert; |
| import org.junit.Test; |
| |
| import java.io.IOException; |
| import java.net.URI; |
| import java.net.URISyntaxException; |
| import java.net.UnknownHostException; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.UUID; |
| |
| public class JacksonTest { |
| |
| private static GossipSettings simpleSettings(GossipSettings settings) { |
| settings.setPersistRingState(false); |
| settings.setPersistDataState(false); |
| settings.setTransportManagerClass("org.apache.gossip.transport.UnitTestTransportManager"); |
| settings.setProtocolManagerClass("org.apache.gossip.protocol.json.JacksonProtocolManager"); |
| return settings; |
| } |
| |
| private static GossipSettings withSigning(GossipSettings settings) { |
| settings.setSignMessages(true); |
| return settings; |
| } |
| |
| // formerly of SignedMessageTest. |
| @Test(expected = IllegalArgumentException.class) |
| public void ifSignMustHaveKeys() |
| throws URISyntaxException, UnknownHostException, InterruptedException { |
| String cluster = UUID.randomUUID().toString(); |
| GossipSettings settings = withSigning(simpleSettings(new GossipSettings())); |
| List<Member> startupMembers = new ArrayList<>(); |
| URI uri = new URI("udp://" + "127.0.0.1" + ":" + (30000 + 1)); |
| GossipManager gossipService = GossipManagerBuilder.newBuilder() |
| .cluster(cluster) |
| .uri(uri) |
| .id(1 + "") |
| .gossipMembers(startupMembers) |
| .gossipSettings(settings) |
| .build(); |
| gossipService.init(); |
| } |
| |
| @Test |
| public void jacksonSerialTest() throws InterruptedException, URISyntaxException, IOException { |
| ObjectMapper objectMapper = JacksonProtocolManager.buildObjectMapper(simpleSettings(new GossipSettings())); |
| |
| OrSet<Integer> i = new OrSet<Integer>(new OrSet.Builder<Integer>().add(1).remove(1)); |
| String s = objectMapper.writeValueAsString(i); |
| @SuppressWarnings("unchecked") |
| OrSet<Integer> back = objectMapper.readValue(s, OrSet.class); |
| Assert.assertEquals(back, i); |
| } |
| |
| @Test |
| public void testMessageEqualityAssumptions() { |
| long timeA = System.nanoTime(); |
| long timeB = System.nanoTime(); |
| Assert.assertNotEquals(timeA, timeB); |
| |
| TestMessage messageA0 = new TestMessage(Long.toHexString(timeA)); |
| TestMessage messageA1 = new TestMessage(Long.toHexString(timeA)); |
| TestMessage messageB = new TestMessage(Long.toHexString(timeB)); |
| |
| Assert.assertEquals(messageA0, messageA1); |
| Assert.assertFalse(messageA0 == messageA1); |
| Assert.assertNotEquals(messageA0, messageB); |
| Assert.assertNotEquals(messageA1, messageB); |
| } |
| |
| // ideally, we would test the serializability of every message type, but we just want to make sure this works in |
| // basic cases. |
| @Test |
| public void testMessageSerializationRoundTrip() throws Exception { |
| ProtocolManager mgr = new JacksonProtocolManager(simpleSettings(new GossipSettings()), "foo", new MetricRegistry()); |
| for (int i = 0; i < 100; i++) { |
| TestMessage a = new TestMessage(Long.toHexString(System.nanoTime())); |
| byte[] bytes = mgr.write(a); |
| TestMessage b = (TestMessage) mgr.read(bytes); |
| Assert.assertFalse(a == b); |
| Assert.assertEquals(a, b); |
| Assert.assertEquals(a.getMapOfThings(), b.getMapOfThings()); // concerned about that one, so explicit check. |
| } |
| } |
| } |