Stand storage coverage improvement

server/src/test/java/org/spine3/server/storage/StandStorageShould.java

@@ -0,0 +1,133 @@
+/*
+ * Copyright 2016, TeamDev Ltd. All rights reserved.
+ *
+ * Redistribution and use in source and/or binary forms, with or without
+ * modification, must retain the above copyright notice and the following
+ * disclaimer.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+package org.spine3.server.storage;
+
+import com.google.common.base.Function;
+import com.google.common.base.Supplier;
+import com.google.common.collect.Collections2;
+import com.google.protobuf.Any;
+import org.junit.Test;
+import org.spine3.base.Identifiers;
+import org.spine3.protobuf.AnyPacker;
+import org.spine3.protobuf.Timestamps;
+import org.spine3.protobuf.TypeUrl;
+import org.spine3.server.stand.AggregateStateId;
+import org.spine3.test.projection.Project;
+import org.spine3.test.projection.ProjectId;
+import org.spine3.test.projection.Task;
+
+import javax.annotation.Nullable;
+import java.util.Collection;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+
+import static org.spine3.test.Verify.assertContains;
+import static org.spine3.test.Verify.assertSize;
+
+/**
+ * @author Dmytro Dashenkov
+ */
+public abstract class StandStorageShould {
+
+    protected abstract StandStorage createStorage();
+
+    protected static final Supplier<AggregateStateId<ProjectId>> DEFAULT_ID_SUPPLIER
+            = new Supplier<AggregateStateId<ProjectId>>() {
+        @SuppressWarnings("unchecked")
+        @Override
+        public AggregateStateId<ProjectId> get() {
+            final ProjectId projectId = ProjectId.newBuilder()
+                                                 .setId(Identifiers.newUuid())
+                                                 .build();
+            return AggregateStateId.of(projectId, TypeUrl.of(Project.class));
+        }
+    };
+
+    @Test
+    public void retrieve_all_records() {
+        final StandStorage storage = createStorage();
+        final List<AggregateStateId> ids = fill(storage, 10, DEFAULT_ID_SUPPLIER);
+
+        final Map<AggregateStateId, EntityStorageRecord> allRecords = storage.readAll();
+        checkIds(ids, allRecords.values());
+    }
+
+    @Test
+    public void retrieve_records_by_ids() {
+        final StandStorage storage = createStorage();
+        // Use a subset of IDs
+        final List<AggregateStateId> ids = fill(storage, 10, DEFAULT_ID_SUPPLIER).subList(0, 5);
+
+        final Collection<EntityStorageRecord> records = (Collection<EntityStorageRecord>) storage.readMultiple(ids);
+        checkIds(ids, records);
+    }
+
+    protected static List<AggregateStateId> fill(StandStorage storage,
+                                                 int count,
+                                                 Supplier<AggregateStateId<ProjectId>> idSupplier) {
+        final List<AggregateStateId> ids = new LinkedList<>();
+
+        for (int i = 0; i < count; i++) {
+            final AggregateStateId genericId = idSupplier.get();
+            final ProjectId id = (ProjectId) genericId.getAggregateId();
+            final Project project = Project.newBuilder()
+                                           .setId(id)
+                                           .setStatus(Project.Status.CREATED)
+                                           .setName(String.format("test-project-%s", i))
+                                           .addTask(Task.getDefaultInstance())
+                                           .build();
+            final EntityStorageRecord record = EntityStorageRecord.newBuilder()
+                                                                  .setState(AnyPacker.pack(project))
+                                                                  .setWhenModified(Timestamps.getCurrentTime())
+                                                                  .setVersion(1)
+                                                                  .build();
+            storage.write(genericId, record);
+            ids.add(genericId);
+        }
+
+        return ids;
+    }
+
+    protected void checkIds(List<AggregateStateId> ids, Collection<EntityStorageRecord> records) {
+        assertSize(ids.size(), records);
+
+        final Collection<ProjectId> projectIds = Collections2.transform(ids, new Function<AggregateStateId, ProjectId>() {
+            @Nullable
+            @Override
+            public ProjectId apply(@Nullable AggregateStateId input) {
+                if (input == null) {
+                    return null;
+                }
+                return (ProjectId) input.getAggregateId();
+            }
+        });
+
+        for (EntityStorageRecord record : records) {
+            final Any packedState = record.getState();
+            final Project state = AnyPacker.unpack(packedState);
+            final ProjectId id = state.getId();
+
+            assertContains(id, projectIds);
+        }
+    }
+
+}

server/src/test/java/org/spine3/server/storage/memory/InMemoryStandStorageShould.java

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2016, TeamDev Ltd. All rights reserved.
+ *
+ * Redistribution and use in source and/or binary forms, with or without
+ * modification, must retain the above copyright notice and the following
+ * disclaimer.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+package org.spine3.server.storage.memory;
+
+import org.spine3.server.storage.StandStorage;
+import org.spine3.server.storage.StandStorageShould;
+
+/**
+ * @author Dmytro Dashenkov
+ */
+public class InMemoryStandStorageShould extends StandStorageShould {
+
+    @Override
+    protected StandStorage createStorage() {
+        return InMemoryStandStorage.newBuilder()
+                                   .setMultitenant(false)
+                                   .build();
+    }
+}