/*
* Copyright Terracotta, Inc.
*
* Licensed 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.ehcache.clustered.common.internal.messages;
import org.junit.Before;
import org.junit.Test;
import java.util.HashSet;
import java.util.Set;
import java.util.UUID;
import static org.hamcrest.Matchers.containsInAnyOrder;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.notNullValue;
import static org.junit.Assert.assertThat;
public class ReconnectMessageCodecTest {
private ReconnectMessageCodec reconnectMessageCodec;
@Before
public void setUp() {
reconnectMessageCodec = new ReconnectMessageCodec();
}
@Test
public void testClusterTierReconnectCodec() {
ClusterTierReconnectMessage reconnectMessage = new ClusterTierReconnectMessage(UUID.randomUUID());
Set<Long> setToInvalidate = new HashSet<Long>();
setToInvalidate.add(1L);
setToInvalidate.add(11L);
setToInvalidate.add(111L);
reconnectMessage.addInvalidationsInProgress(setToInvalidate);
reconnectMessage.clearInProgress();
ClusterTierReconnectMessage decoded = reconnectMessageCodec.decode(reconnectMessageCodec.encode(reconnectMessage));
assertThat(decoded, notNullValue());
assertThat(decoded.getClientId(), is(reconnectMessage.getClientId()));
assertThat(decoded.getInvalidationsInProgress(), containsInAnyOrder(setToInvalidate.toArray()));
assertThat(decoded.isClearInProgress(), is(true));
}
@Test
public void testClusterTierManagerReconnectCodec() {
UUID uuid = UUID.randomUUID();
ClusterTierManagerReconnectMessage message = new ClusterTierManagerReconnectMessage(uuid);
ClusterTierManagerReconnectMessage decodedMessage = reconnectMessageCodec.decodeReconnectMessage(reconnectMessageCodec.encode(message));
assertThat(decodedMessage.getClientId(), is(uuid));
}
}