From 46350c084766789ea59e83f1917c57c81d653048 Mon Sep 17 00:00:00 2001
From: "Tschaen, Brendan" <ctschaen@att.com>
Date: Tue, 16 Oct 2018 20:22:35 -0400
Subject: Include Cassandra locking

Change-Id: I085acf8336d5f27782ee12768846a5befd3ee60d
Issue-ID: MUSIC-148
Signed-off-by: Tschaen, Brendan <ctschaen@att.com>
---
 .../org/onap/music/unittests/TestVotingApp.java    | 115 +++++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 src/test/java/org/onap/music/unittests/TestVotingApp.java

(limited to 'src/test/java/org/onap/music/unittests/TestVotingApp.java')

diff --git a/src/test/java/org/onap/music/unittests/TestVotingApp.java b/src/test/java/org/onap/music/unittests/TestVotingApp.java
new file mode 100644
index 00000000..19b10fe8
--- /dev/null
+++ b/src/test/java/org/onap/music/unittests/TestVotingApp.java
@@ -0,0 +1,115 @@
+package org.onap.music.unittests;
+
+import java.util.HashMap;
+import java.util.Map;
+ 
+import org.onap.music.datastore.PreparedQueryObject;
+import org.onap.music.exceptions.MusicLockingException;
+import org.onap.music.exceptions.MusicQueryException;
+import org.onap.music.exceptions.MusicServiceException;
+import org.onap.music.main.MusicCore;
+
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.Row;
+ 
+/**
+ *
+ */
+public class TestVotingApp 
+{
+    String keyspaceName;
+    String tableName;
+       
+    public TestVotingApp() throws MusicServiceException {
+        keyspaceName = "VotingAppForMusic"+System.currentTimeMillis();
+        tableName = "votecount";
+    }
+
+    private void initialize() throws MusicServiceException {
+        createVotingKeyspace();
+        System.out.println("Created keyspaces");
+        createVotingTable();
+        System.out.println("Created tables");
+
+        createEntryForCandidate("Popeye");
+        createEntryForCandidate("Judy");
+        createEntryForCandidate("Flash");
+        createEntryForCandidate("Mickey");
+        System.out.println("Created candidates");
+    }
+
+    private void createVotingKeyspace() throws MusicServiceException {
+              Map<String,Object> replicationInfo = new HashMap<String, Object>();
+              replicationInfo.put("'class'", "'SimpleStrategy'");
+              replicationInfo.put("'replication_factor'", 1);
+              
+              PreparedQueryObject queryObject = new PreparedQueryObject();
+              queryObject.appendQueryString(
+                "CREATE KEYSPACE " + keyspaceName + " WITH REPLICATION = " + replicationInfo.toString().replaceAll("=", ":"));
+              
+              try {
+                     MusicCore.nonKeyRelatedPut(queryObject, "eventual");
+              } catch (MusicServiceException e) {
+                   throw(e);
+              }
+       }
+       
+    private void createVotingTable() throws MusicServiceException {
+        PreparedQueryObject queryObject = new PreparedQueryObject();
+        queryObject.appendQueryString(
+                "CREATE TABLE " + keyspaceName + "." + tableName + " (name text PRIMARY KEY, count int);");
+
+        try {
+            MusicCore.createTable(keyspaceName, tableName, queryObject, "eventual");
+        } catch (MusicServiceException e) {
+            throw (e);
+        }
+    }
+ 
+       private void createEntryForCandidate(String candidateName) throws MusicServiceException {
+              PreparedQueryObject queryObject = new PreparedQueryObject();
+              queryObject.appendQueryString(
+                "INSERT INTO " + keyspaceName + "." + tableName + " (name, count) "
+                            + "VALUES ('"+candidateName+"', 0);");
+              
+              MusicCore.nonKeyRelatedPut(queryObject, "eventual");
+       }
+ 
+       
+       private void updateVoteCount(String candidateName, int numVotes) throws MusicLockingException, MusicQueryException, MusicServiceException {
+              PreparedQueryObject queryObject = new PreparedQueryObject();
+              queryObject.appendQueryString(
+                "UPDATE " + keyspaceName + "." + tableName + " SET count="+numVotes + " where name='"+candidateName+"';");
+              MusicCore.atomicPut(keyspaceName, tableName, candidateName, queryObject, null);
+       }
+ 
+       private HashMap<String, Integer> readAllVotes() throws MusicServiceException {
+              PreparedQueryObject queryObject = new PreparedQueryObject();
+              queryObject.appendQueryString("SELECT * FROM " + keyspaceName + "." + tableName);
+              ResultSet rs = MusicCore.get(queryObject);
+              HashMap<String, Integer> voteCount = new HashMap<String, Integer>();
+              for(Row candidate : rs.all()) {
+            	  	voteCount.put(candidate.getString("name"), candidate.getInt("count"));
+              }
+              return voteCount;
+       }
+       
+       public static void main( String[] args ) throws Exception {
+        TestVotingApp tva = new TestVotingApp();
+        tva.initialize();
+
+        tva.updateVoteCount("Popeye",5);
+        tva.updateVoteCount("Judy",9);
+        tva.updateVoteCount("Mickey",8);
+       tva.updateVoteCount("Flash",1);
+       tva.updateVoteCount("Flash",2);
+
+        HashMap<String, Integer> voteCount = tva.readAllVotes();
+        System.out.println(voteCount);
+		assert(voteCount.get("Popeye") == 5);
+		assert(voteCount.get("Judy") == 9);
+		assert(voteCount.get("Mickey") == 8);
+		assert(voteCount.get("Flash") == 2);
+    }
+ 
+}
\ No newline at end of file
-- 
cgit 1.2.3-korg