BUG-5280: add FrontendMetadata

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / persisted / FrontendShardDataTreeSnapshotMetadataTest.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/persisted/FrontendShardDataTreeSnapshotMetadataTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/persisted/FrontendShardDataTreeSnapshotMetadataTest.java
new file mode 100644 (file)
index 0000000..f99cf6f
--- /dev/null
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/persisted/FrontendShardDataTreeSnapshotMetadataTest.java
@@ -0,0 +1,134 @@
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.controller.cluster.datastore.persisted;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import com.google.common.collect.Range;
+import com.google.common.collect.RangeSet;
+import com.google.common.collect.TreeRangeSet;
+import com.google.common.primitives.UnsignedLong;
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import org.junit.Test;
+import org.opendaylight.controller.cluster.access.concepts.ClientIdentifier;
+import org.opendaylight.controller.cluster.access.concepts.FrontendIdentifier;
+import org.opendaylight.controller.cluster.access.concepts.FrontendType;
+import org.opendaylight.controller.cluster.access.concepts.MemberName;
+
+public class FrontendShardDataTreeSnapshotMetadataTest {
+
+    @Test(expected = NullPointerException.class)
+    public final void testCreateMetadataSnapshotNullInput() {
+        new FrontendShardDataTreeSnapshotMetadata(null);
+    }
+
+    @Test
+    public final void testCreateMetadataSnapshotEmptyInput() throws Exception {
+        final FrontendShardDataTreeSnapshotMetadata emptyOrigSnapshot = createEmptyMetadataSnapshot();
+        final FrontendShardDataTreeSnapshotMetadata emptyCopySnapshot = copy(emptyOrigSnapshot);
+        testMetadataSnapshotEqual(emptyOrigSnapshot, emptyCopySnapshot);
+    }
+
+    @Test
+    public final void testSerializeMetadataSnapshotWithOneClient() throws Exception {
+        final FrontendShardDataTreeSnapshotMetadata origSnapshot = createMetadataSnapshot(1);
+        final FrontendShardDataTreeSnapshotMetadata copySnapshot = copy(origSnapshot);
+        testMetadataSnapshotEqual(origSnapshot, copySnapshot);
+    }
+
+    @Test
+    public final void testSerializeMetadataSnapshotWithMoreClients() throws Exception {
+        final FrontendShardDataTreeSnapshotMetadata origSnapshot = createMetadataSnapshot(5);
+        final FrontendShardDataTreeSnapshotMetadata copySnapshot = copy(origSnapshot);
+        testMetadataSnapshotEqual(origSnapshot, copySnapshot);
+    }
+
+    private static void testMetadataSnapshotEqual(final FrontendShardDataTreeSnapshotMetadata origSnapshot,
+            final FrontendShardDataTreeSnapshotMetadata copySnapshot) {
+
+        final List<FrontendClientMetadata> origClientList = origSnapshot.getClients();
+        final List<FrontendClientMetadata> copyClientList = copySnapshot.getClients();
+
+        assertTrue(origClientList.size() == copyClientList.size());
+
+        final Map<ClientIdentifier, FrontendClientMetadata> origIdent = new HashMap<>();
+        final Map<ClientIdentifier, FrontendClientMetadata> copyIdent = new HashMap<>();
+        origClientList.forEach(client -> origIdent.put(client.getIdentifier(), client));
+        origClientList.forEach(client -> copyIdent.put(client.getIdentifier(), client));
+
+        assertTrue(origIdent.keySet().containsAll(copyIdent.keySet()));
+        assertTrue(copyIdent.keySet().containsAll(origIdent.keySet()));
+
+        origIdent.values().forEach(client -> {
+            final FrontendClientMetadata copyClient = copyIdent.get(client.getIdentifier());
+            testObject(client.getIdentifier(), copyClient.getIdentifier());
+            assertTrue(client.getPurgedHistories().equals(copyClient.getPurgedHistories()));
+            assertTrue(client.getCurrentHistories().equals(copyClient.getCurrentHistories()));
+        });
+    }
+
+    private static FrontendShardDataTreeSnapshotMetadata createEmptyMetadataSnapshot() {
+        return new FrontendShardDataTreeSnapshotMetadata(Collections.<FrontendClientMetadata> emptyList());
+    }
+
+    private static FrontendShardDataTreeSnapshotMetadata createMetadataSnapshot(final int size) {
+        final List<FrontendClientMetadata> clients = new ArrayList<>();
+        for (long i = 0; i < size; i++) {
+            clients.add(createFrontedClientMetadata(i));
+        }
+        return new FrontendShardDataTreeSnapshotMetadata(clients);
+    }
+
+    private static FrontendClientMetadata createFrontedClientMetadata(final long i) {
+        final String index = String.valueOf(i);
+        final String indexName = "test_" + index;
+        final FrontendIdentifier frontendIdentifier = FrontendIdentifier.create(MemberName.forName(indexName),
+                FrontendType.forName(index));
+        final ClientIdentifier clientIdentifier = ClientIdentifier.create(frontendIdentifier, i);
+
+        final RangeSet<UnsignedLong> purgedHistories = TreeRangeSet.create();
+        purgedHistories.add(Range.closed(UnsignedLong.ZERO, UnsignedLong.ONE));
+
+        final Collection<FrontendHistoryMetadata> currentHistories = Collections
+                .singleton(new FrontendHistoryMetadata(i, i, i, true));
+
+        return new FrontendClientMetadata(clientIdentifier, purgedHistories, currentHistories);
+    }
+
+    private static final <T> void testObject(final T object, final T equalObject) {
+        assertEquals(object.hashCode(), equalObject.hashCode());
+        assertTrue(object.equals(object));
+        assertTrue(object.equals(equalObject));
+        assertFalse(object.equals(null));
+        assertFalse(object.equals("dummy"));
+    }
+
+    @SuppressWarnings("unchecked")
+    private static <T> T copy(final T o) throws IOException, ClassNotFoundException {
+        final ByteArrayOutputStream bos = new ByteArrayOutputStream();
+        try (ObjectOutputStream oos = new ObjectOutputStream(bos)) {
+            oos.writeObject(o);
+        }
+
+        try (ObjectInputStream ois = new ObjectInputStream(new ByteArrayInputStream(bos.toByteArray()))) {
+            return (T) ois.readObject();
+        }
+    }
+}