Abdullah/stub storage test cases

src/main/java/model/exceptions/LightChainDistributedStorageException.java

@@ -3,4 +3,7 @@
 /**
  * Represents a runtime exception happens on distributed storage layer of LightChain.
  */
-public class LightChainDistributedStorageException extends Exception{}
+public class LightChainDistributedStorageException extends Exception {
+  public LightChainDistributedStorageException(String s) {
+  }
+}

src/main/java/network/NetworkAdapter.java

@@ -1,5 +1,7 @@
 package network;
 
+import java.util.ArrayList;
+
 import model.Entity;
 import model.exceptions.LightChainDistributedStorageException;
 import model.exceptions.LightChainNetworkingException;
@@ -38,4 +40,13 @@ public interface NetworkAdapter {
    * @throws LightChainDistributedStorageException any unhappy path taken on retrieving the Entity.
    */
   Entity get(Identifier identifier, String namespace) throws LightChainDistributedStorageException;
+
+  /**
+   * Retrieves all entities stored on the underlying DHT of nodes that stored on this channel.
+   *
+   * @param namespace the namespace on which this query is resolved.
+   * @return list of all entities stored on this channel from underlying DHT.
+   * @throws LightChainDistributedStorageException any unhappy path taken on retrieving the Entities.
+   */
+  ArrayList<Entity> allEntities(String namespace) throws LightChainDistributedStorageException;
 }

src/test/java/networking/MockConduit.java

@@ -53,7 +53,7 @@ public void unicast(Entity e, Identifier target) throws LightChainNetworkingExce
    */
   @Override
   public void put(Entity e) throws LightChainDistributedStorageException {
-
+    this.networkAdapter.put(e, channel);
   }
 
   /**
@@ -66,12 +66,12 @@ public void put(Entity e) throws LightChainDistributedStorageException {
    */
   @Override
   public Entity get(Identifier identifier) throws LightChainDistributedStorageException {
-    return null;
+    return this.networkAdapter.get(identifier, channel);
   }
 
   @Override
   public ArrayList<Entity> allEntities() throws LightChainDistributedStorageException {
-    return null;
+    return this.networkAdapter.allEntities(channel);
   }
 
   public boolean hasSent(Identifier entityId) {

src/test/java/networking/stub/Hub.java

@@ -1,29 +1,103 @@
 package networking.stub;
 
+import java.util.ArrayList;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.locks.ReentrantReadWriteLock;
 
 import model.Entity;
+import model.exceptions.LightChainDistributedStorageException;
 import model.lightchain.Identifier;
 import network.Network;
 
 /**
  * Models the core communication part of the networking layer that allows stub network instances to talk to each other.
  */
 public class Hub {
+  private final ReentrantReadWriteLock lock;
+  private final String channel1 = "test-network-channel-1";
+  private final String channel2 = "test-network-channel-2";
   private final ConcurrentHashMap<Identifier, Network> networks;
+  private final ConcurrentHashMap<Identifier, Entity> channelMap1;
+  private final ConcurrentHashMap<Identifier, Entity> channelMap2;
 
   /**
    * Create a hub.
    */
   public Hub() {
     this.networks = new ConcurrentHashMap<>();
+    this.channelMap1 = new ConcurrentHashMap<>();
+    this.channelMap2 = new ConcurrentHashMap<>();
+    this.lock = new ReentrantReadWriteLock();
+  }
+
+  /**
+   * Put Entity to channel.
+   *
+   * @param e         entitiy.
+   * @param namespace channel name.
+   * @throws LightChainDistributedStorageException any unhappy path taken on storing the Entity.
+   */
+  public void putEntityToChannel(Entity e, String namespace) throws LightChainDistributedStorageException {
+    try {
+      lock.writeLock().lock();
+      if (namespace.equals(channel1)) {
+        channelMap1.put(e.id(), e);
+      } else if (namespace.equals(channel2)) {
+        channelMap2.put(e.id(), e);
+      } else {
+        throw new LightChainDistributedStorageException("entity could not be put the given channel");
+      }
+    } finally {
+      lock.writeLock().unlock();
+    }
+  }
+
+  /**
+   * Get entity from channel.
+   *
+   * @param identifier of entity.
+   * @param namespace  channel name.
+   * @return entity.
+   * @throws LightChainDistributedStorageException any unhappy path taken on storing the Entity.
+   */
+  public Entity getEntityFromChannel(Identifier identifier, String namespace)
+      throws LightChainDistributedStorageException {
+    try {
+      lock.readLock().lock();
+      if (namespace.equals(channel1)) {
+        return channelMap1.get(identifier);
+      } else if (namespace.equals(channel2)) {
+        return channelMap2.get(identifier);
+      } else {
+        throw new LightChainDistributedStorageException("could not get the entity");
+      }
+    } finally {
+      lock.readLock().unlock();
+    }
+  }
+
+  /**
+   * Retrieves all entities stored on the underlying DHT of nodes that stored on this channel.
+   *
+   * @param namespace the namespace on which this query is resolved.
+   * @return list of all entities stored on this channel from underlying DHT.
+   * @throws LightChainDistributedStorageException any unhappy path taken on retrieving the Entities.
+   */
+  public ArrayList<Entity> getAllEntities(String namespace) throws LightChainDistributedStorageException {
+    if (namespace.equals(channel1)) {
+      return new ArrayList<>(channelMap1.values());
+    } else if (namespace.equals(channel2)) {
+      return new ArrayList<>(channelMap2.values());
+    } else {
+      throw new LightChainDistributedStorageException("could not get the entities");
+    }
   }
 
   /**
    * Registeration of a network to the Hub.
    *
    * @param identifier identifier of network.
-   * @param network to be registered.
+   * @param network    to be registered.
    */
   public void registerNetwork(Identifier identifier, Network network) {
     networks.put(identifier, network);
@@ -32,8 +106,8 @@ public void registerNetwork(Identifier identifier, Network network) {
   /**
    * Transfer entity from to another network on the same channel.
    *
-   * @param entity entity to be transferred.
-   * @param target identifier of target.
+   * @param entity  entity to be transferred.
+   * @param target  identifier of target.
    * @param channel channel on which the entity is delivered to target.
    */
   public void transferEntity(Entity entity, Identifier target, String channel) throws IllegalStateException {

src/test/java/networking/stub/StubNetwork.java

@@ -1,5 +1,6 @@
 package networking.stub;
 
+import java.util.ArrayList;
 import java.util.concurrent.ConcurrentHashMap;
 
 import model.Entity;
@@ -110,7 +111,7 @@ public void unicast(Entity e, Identifier target, String channel) throws LightCha
    */
   @Override
   public void put(Entity e, String namespace) throws LightChainDistributedStorageException {
-
+    this.hub.putEntityToChannel(e, namespace);
   }
 
   /**
@@ -124,6 +125,18 @@ public void put(Entity e, String namespace) throws LightChainDistributedStorageE
    */
   @Override
   public Entity get(Identifier identifier, String namespace) throws LightChainDistributedStorageException {
-    return null;
+    return this.hub.getEntityFromChannel(identifier, namespace);
+  }
+
+  /**
+   * Retrieves all entities stored on the underlying DHT of nodes that stored on this channel.
+   *
+   * @param namespace the namespace on which this query is resolved.
+   * @return list of all entities stored on this channel from underlying DHT.
+   * @throws LightChainDistributedStorageException any unhappy path taken on retrieving the Entities.
+   */
+  @Override
+  public ArrayList<Entity> allEntities(String namespace) throws LightChainDistributedStorageException {
+    return this.hub.getAllEntities(namespace);
   }
 }