package org.opendaylight.controller.cluster.datastore.messages;
import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import com.google.common.collect.ImmutableSortedSet;
import java.io.Serializable;
+import java.util.Optional;
+import java.util.SortedSet;
import org.apache.commons.lang.SerializationUtils;
import org.junit.Test;
import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
batched.addModification(new WriteModification(writePath, writeData));
batched.addModification(new MergeModification(mergePath, mergeData));
batched.addModification(new DeleteModification(deletePath));
- batched.setReady(true);
+ assertFalse("isReady", batched.isReady());
+ batched.setReady();
+ assertTrue("isReady", batched.isReady());
batched.setTotalMessagesSent(5);
BatchedModifications clone = (BatchedModifications) SerializationUtils.clone(
(Serializable) batched.toSerializable());
assertEquals("getVersion", DataStoreVersions.CURRENT_VERSION, clone.getVersion());
- assertEquals("getTransactionID", tx1, clone.getTransactionID());
- assertEquals("isReady", true, clone.isReady());
+ assertEquals("getTransactionID", tx1, clone.getTransactionId());
+ assertTrue("isReady", clone.isReady());
+ assertFalse("isDoCommitOnReady", clone.isDoCommitOnReady());
+ assertFalse("participatingShardNames present", clone.getParticipatingShardNames().isPresent());
assertEquals("getTotalMessagesSent", 5, clone.getTotalMessagesSent());
assertEquals("getModifications size", 3, clone.getModifications().size());
assertEquals("getVersion", DataStoreVersions.CURRENT_VERSION, delete.getVersion());
assertEquals("getPath", deletePath, delete.getPath());
- // Test with different params.
+ // Test with participating shard names.
+
final TransactionIdentifier tx2 = nextTransactionId();
batched = new BatchedModifications(tx2, (short)10000);
+ final SortedSet<String> shardNames = ImmutableSortedSet.of("one", "two");
+ batched.setReady(Optional.of(shardNames));
+ batched.setDoCommitOnReady(true);
+ assertTrue("isReady", batched.isReady());
clone = (BatchedModifications) SerializationUtils.clone((Serializable) batched.toSerializable());
assertEquals("getVersion", DataStoreVersions.CURRENT_VERSION, clone.getVersion());
- assertEquals("getTransactionID", tx2, clone.getTransactionID());
- assertEquals("isReady", false, clone.isReady());
+ assertEquals("getTransactionID", tx2, clone.getTransactionId());
+ assertTrue("isReady", clone.isReady());
+ assertTrue("isDoCommitOnReady", clone.isDoCommitOnReady());
+ assertTrue("participatingShardNames present", clone.getParticipatingShardNames().isPresent());
+ assertEquals("participatingShardNames", shardNames, clone.getParticipatingShardNames().get());
+ assertEquals("getModifications size", 0, clone.getModifications().size());
+
+ // Test not ready.
+
+ batched = new BatchedModifications(tx2, DataStoreVersions.CURRENT_VERSION);
+
+ clone = (BatchedModifications) SerializationUtils.clone((Serializable) batched.toSerializable());
+ assertEquals("getVersion", DataStoreVersions.CURRENT_VERSION, clone.getVersion());
+ assertEquals("getTransactionID", tx2, clone.getTransactionId());
+ assertFalse("isReady", clone.isReady());
assertEquals("getModifications size", 0, clone.getModifications().size());
+ // Test pre-Flourine
+
+ batched = new BatchedModifications(tx2, DataStoreVersions.BORON_VERSION);
+ batched.addModification(new WriteModification(writePath, writeData));
+ batched.setReady(Optional.of(ImmutableSortedSet.of("one", "two")));
+
+ clone = (BatchedModifications) SerializationUtils.clone((Serializable) batched.toSerializable());
+
+ assertEquals("getVersion", DataStoreVersions.BORON_VERSION, clone.getVersion());
+ assertEquals("getTransactionID", tx2, clone.getTransactionId());
+ assertTrue("isReady", clone.isReady());
+ assertFalse("participatingShardNames present", clone.getParticipatingShardNames().isPresent());
+ assertEquals("getModifications size", 1, clone.getModifications().size());
}
@Test