X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Forg%2Fonap%2Fmusic%2Funittests%2FTstRestMusicLockAPI.java;fp=src%2Ftest%2Fjava%2Forg%2Fonap%2Fmusic%2Funittests%2FTstRestMusicLockAPI.java;h=0000000000000000000000000000000000000000;hb=11ee6836d6f25a1becdea60a322a72fbffd4b8b6;hp=e9321d2567379b2f5d46732be2bce2767827a733;hpb=a00014e78f18134f998fb46a7dd543e6ea05a3bd;p=music.git diff --git a/src/test/java/org/onap/music/unittests/TstRestMusicLockAPI.java b/src/test/java/org/onap/music/unittests/TstRestMusicLockAPI.java deleted file mode 100644 index e9321d25..00000000 --- a/src/test/java/org/onap/music/unittests/TstRestMusicLockAPI.java +++ /dev/null @@ -1,768 +0,0 @@ -/* - * ============LICENSE_START========================================== - * org.onap.music - * =================================================================== - * Copyright (c) 2017 AT&T Intellectual Property - * =================================================================== - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - * - * ============LICENSE_END============================================= - * ==================================================================== - */ - -package org.onap.music.unittests; - -import static org.junit.Assert.assertEquals; -import static org.junit.Assert.assertTrue; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import java.util.Map; -import java.util.UUID; -import javax.ws.rs.core.MultivaluedMap; -import javax.ws.rs.core.Response; -import javax.ws.rs.core.UriInfo; -import org.junit.After; -import org.junit.AfterClass; -import org.junit.BeforeClass; -import org.junit.Ignore; -import org.junit.Test; -import org.junit.runner.RunWith; -import org.mindrot.jbcrypt.BCrypt; -import org.mockito.Mock; -import org.mockito.Mockito; -import org.mockito.junit.MockitoJUnitRunner; -import org.onap.music.datastore.PreparedQueryObject; -import org.onap.music.datastore.jsonobjects.JsonInsert; -import org.onap.music.datastore.jsonobjects.JsonLeasedLock; -import org.onap.music.datastore.jsonobjects.JsonLock; -import org.onap.music.datastore.jsonobjects.JsonTable; -import org.onap.music.datastore.jsonobjects.JsonUpdate; -import org.onap.music.exceptions.MusicServiceException; -import org.onap.music.lockingservice.cassandra.CassaLockStore; -import org.onap.music.lockingservice.cassandra.LockType; -import org.onap.music.main.MusicCore; -import org.onap.music.main.MusicUtil; -import org.onap.music.rest.RestMusicDataAPI; -import org.onap.music.rest.RestMusicLocksAPI; -import com.datastax.driver.core.DataType; -import com.datastax.driver.core.ResultSet; -import com.datastax.driver.core.Row; -import com.sun.jersey.core.util.Base64; -import com.sun.jersey.core.util.MultivaluedMapImpl; - -@RunWith(MockitoJUnitRunner.class) -public class TstRestMusicLockAPI { - - - @Mock - UriInfo info; - - RestMusicLocksAPI lock = new RestMusicLocksAPI(); - RestMusicDataAPI data = new RestMusicDataAPI(); - static PreparedQueryObject testObject; - - static String appName = "TestApp"; - static String userId = "TestUser"; - static String password = "TestPassword"; - static String authData = userId + ":" + password; - static String wrongAuthData = userId + ":" + "pass"; - static String authorization = new String(Base64.encode(authData.getBytes())); - static String wrongAuthorization = new String(Base64.encode(wrongAuthData.getBytes())); - static boolean isAAF = false; - static UUID uuid = UUID.fromString("abc66ccc-d857-4e90-b1e5-df98a3d40ce6"); - static String keyspaceName = "testcassa"; - static String tableName = "employees"; - static String onboardUUID = null; - static String lockName = "testcassa.employees.testname"; - - @BeforeClass - public static void init() throws Exception { - System.out.println("Testing RestMusicLock class"); - try { - createKeyspace(); - } catch (Exception e) { - e.printStackTrace(); - throw new Exception("Unable to initialize before TestRestMusicData test class. " + e.getMessage()); - } - } - - @After - public void afterEachTest() throws MusicServiceException { - clearAllTablesFromKeyspace(); - } - - @AfterClass - public static void tearDownAfterClass() throws Exception { - testObject = new PreparedQueryObject(); - testObject.appendQueryString("DROP KEYSPACE IF EXISTS " + keyspaceName); - MusicCore.eventualPut(testObject); - } - - @SuppressWarnings("unchecked") - @Test - public void test_createLockReference() throws Exception { - System.out.println("Testing create lockref"); - createAndInsertIntoTable(); - Response response = lock.createLockReference(lockName, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", null, null, appName); - Map respMap = (Map) response.getEntity(); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - - assertEquals(200, response.getStatus()); - assertTrue(respMap.containsKey("lock")); - assertTrue(((Map) respMap.get("lock")).containsKey("lock")); - } - - @Test - public void test_createBadLockReference() throws Exception { - System.out.println("Testing create bad lockref"); - createAndInsertIntoTable(); - Response response = lock.createLockReference("badlock", "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", null, null, appName); - Map respMap = (Map) response.getEntity(); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - - assertEquals(400, response.getStatus()); - } - - @Test - public void test_createReadLock() throws Exception { - System.out.println("Testing create read lockref"); - createAndInsertIntoTable(); - JsonLock jsonLock = createJsonLock(LockType.READ); - Response response = lock.createLockReference(lockName, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", jsonLock, null, appName); - Map respMap = (Map) response.getEntity(); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - - assertEquals(200, response.getStatus()); - assertTrue(respMap.containsKey("lock")); - assertTrue(((Map) respMap.get("lock")).containsKey("lock")); - } - - @Test - public void test_createWriteLock() throws Exception { - System.out.println("Testing create read lockref"); - createAndInsertIntoTable(); - JsonLock jsonLock = createJsonLock(LockType.WRITE); - Response response = lock.createLockReference(lockName, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", jsonLock, null, appName); - Map respMap = (Map) response.getEntity(); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - - assertEquals(200, response.getStatus()); - assertTrue(respMap.containsKey("lock")); - assertTrue(((Map) respMap.get("lock")).containsKey("lock")); - } - - @Test - public void test_accquireLock() throws Exception { - System.out.println("Testing acquire lock"); - createAndInsertIntoTable(); - String lockRef = createLockReference(); - - Response response = - lock.accquireLock(lockRef, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - } - - @Test - public void test_acquireReadLock() throws Exception { - System.out.println("Testing acquire read lock"); - createAndInsertIntoTable(); - String lockRef = createLockReference(LockType.READ); - String lockRef2 = createLockReference(LockType.READ); - - Response response = - lock.accquireLock(lockRef, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - response = - lock.accquireLock(lockRef2, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - } - - @Test - public void test_acquireReadLockaFail() throws Exception { - System.out.println("Testing acquire read lock"); - createAndInsertIntoTable(); - String lockRef = createLockReference(LockType.WRITE); - String lockRef2 = createLockReference(LockType.READ); - - Response response = - lock.accquireLock(lockRef, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - response = - lock.accquireLock(lockRef2, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_writeWReadLock() throws Exception { - System.out.println("Testing writing with a read lock"); - createAndInsertIntoTable(); - String lockRef = createLockReference(LockType.READ); - - JsonUpdate jsonUpdate = new JsonUpdate(); - Map consistencyInfo = new HashMap<>(); - Map values = new HashMap<>(); - values.put("emp_salary", 2500); - consistencyInfo.put("type", "critical"); - consistencyInfo.put("lockId", lockRef); - jsonUpdate.setConsistencyInfo(consistencyInfo); - jsonUpdate.setKeyspaceName(keyspaceName); - jsonUpdate.setTableName(tableName); - jsonUpdate.setValues(values); - MultivaluedMap row = new MultivaluedMapImpl(); - row.add("emp_name", "testname"); - Mockito.when(info.getQueryParameters()).thenReturn(row); - - Response response = data.updateTable("1", "1", "1", "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName, - authorization, jsonUpdate, keyspaceName, tableName, info); - - assertEquals(400, response.getStatus()); - } - - @Test - public void test_writeWWriteLock() throws Exception { - System.out.println("Testing writing with a read lock"); - createAndInsertIntoTable(); - String lockRef = createLockReference(LockType.WRITE); - - JsonUpdate jsonUpdate = new JsonUpdate(); - Map consistencyInfo = new HashMap<>(); - Map values = new HashMap<>(); - values.put("emp_salary", 2500); - consistencyInfo.put("type", "critical"); - consistencyInfo.put("lockId", lockRef); - jsonUpdate.setConsistencyInfo(consistencyInfo); - jsonUpdate.setKeyspaceName(keyspaceName); - jsonUpdate.setTableName(tableName); - jsonUpdate.setValues(values); - MultivaluedMap row = new MultivaluedMapImpl(); - row.add("emp_name", "testname"); - Mockito.when(info.getQueryParameters()).thenReturn(row); - - Response response = data.updateTable("1", "1", "1", "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName, - authorization, jsonUpdate, keyspaceName, tableName, info); - - assertEquals(200, response.getStatus()); - } - - @Test - public void test_accquireLockWLease() throws Exception { - System.out.println("Testing acquire lock with lease"); - createAndInsertIntoTable(); - String lockRef = createLockReference(); - - JsonLeasedLock jsonLock = new JsonLeasedLock(); - jsonLock.setLeasePeriod(10000); // 10 second lease period? - Response response = lock.accquireLockWithLease(jsonLock, lockRef, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - } - - @Test - public void test_accquireBadLockWLease() throws Exception { - System.out.println("Testing acquire bad lock ref with lease"); - createAndInsertIntoTable(); - String lockRef = createLockReference(); - - JsonLeasedLock jsonLock = new JsonLeasedLock(); - jsonLock.setLeasePeriod(10000); // 10 second lease period? - Response response = lock.accquireLockWithLease(jsonLock, "badlock", "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_accquireBadLock() throws Exception { - System.out.println("Testing acquire lock that is not lock-holder"); - createAndInsertIntoTable(); - // This is required to create an initial loc reference. - String lockRef1 = createLockReference(); - // This will create the next lock reference, whcih will not be avalale yet. - String lockRef2 = createLockReference(); - - Response response = lock.accquireLock(lockRef2, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_accquireBadLockRef() throws Exception { - System.out.println("Testing acquire bad lock ref"); - createAndInsertIntoTable(); - // This is required to create an initial loc reference. - String lockRef1 = createLockReference(); - - Response response = lock.accquireLock("badlockref", "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_currentLockHolder() throws Exception { - System.out.println("Testing get current lock holder"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = - lock.enquireLock(lockName, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - Map respMap = (Map) response.getEntity(); - assertEquals(lockRef, ((Map) respMap.get("lock")).get("lock-holder")); - } - - @Test - public void test_nocurrentLockHolder() throws Exception { - System.out.println("Testing get current lock holder w/ bad lockref"); - createAndInsertIntoTable(); - - Response response = - lock.enquireLock(lockName, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_badcurrentLockHolder() throws Exception { - System.out.println("Testing get current lock holder w/ bad lockref"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = - lock.enquireLock("badlock", "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_holders() throws Exception { - System.out.println("Testing holders api"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = - lock.currentLockHolder(lockName, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - Map respMap = (Map) response.getEntity(); - assertEquals(lockRef, ((Map) respMap.get("lock")).get("lock-holder").get(0)); - } - - @Test - public void test_holdersbadRef() throws Exception { - System.out.println("Testing holders api w/ bad lockref"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = - lock.currentLockHolder("badname", "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_unLock() throws Exception { - System.out.println("Testing unlock"); - createAndInsertIntoTable(); - String lockRef = createLockReference(); - - Response response = - lock.unLock(lockRef, "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - } - - @Test - public void test_unLockBadRef() throws Exception { - System.out.println("Testing unlock w/ bad lock ref"); - createAndInsertIntoTable(); - String lockRef = createLockReference(); - - Response response = - lock.unLock("badref", "1", "1", authorization, "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @Test - public void test_getLockState() throws Exception { - System.out.println("Testing get lock state"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = lock.currentLockState(lockName, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - Map respMap = (Map) response.getEntity(); - assertEquals(lockRef, ((Map) respMap.get("lock")).get("lock-holder")); - } - - @Test - public void test_getLockStateBadRef() throws Exception { - System.out.println("Testing get lock state w/ bad ref"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = lock.currentLockState("badname", "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(400, response.getStatus()); - } - - @SuppressWarnings("unchecked") - @Test - public void test_deadlock() throws Exception { - System.out.println("Testing deadlock"); - createAndInsertIntoTable(); - insertAnotherIntoTable(); - - // Process 1 creates and acquires a lock on row 1 - JsonLock jsonLock = createJsonLock(LockType.WRITE); - Response responseCreate1 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLock, "process1", appName); - Map respMapCreate1 = (Map) responseCreate1.getEntity(); - String lockRefCreate1 = ((Map) respMapCreate1.get("lock")).get("lock"); - - Response responseAcquire1 = - lock.accquireLock(lockRefCreate1, "1", "1", authorization, "abc66001-d857-4e90-b1e5-df98a3d40ce6", appName); - - // Process 2 creates and acquires a lock on row 2 - Response responseCreate2 = lock.createLockReference(lockName + "2", "1", "1", authorization, - "abcde002-d857-4e90-b1e5-df98a3d40ce6", jsonLock, "process2", appName); - Map respMapCreate2 = (Map) responseCreate2.getEntity(); - String lockRefCreate2 = ((Map) respMapCreate2.get("lock")).get("lock"); - - Response responseAcquire2 = - lock.accquireLock(lockRefCreate2, "1", "1", authorization, "abc66002-d857-4e90-b1e5-df98a3d40ce6", appName); - - // Process 2 creates a lock on row 1 - Response responseCreate3 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde003-d857-4e90-b1e5-df98a3d40ce6", jsonLock, "process2", appName); - - // Process 1 creates a lock on row 2, causing deadlock - Response responseCreate4 = lock.createLockReference(lockName + "2", "1", "1", authorization, - "abcde004-d857-4e90-b1e5-df98a3d40ce6", jsonLock, "process1", appName); - Map respMapCreate4 = (Map) responseCreate4.getEntity(); - - System.out.println("Status: " + responseCreate4.getStatus() + ". Entity " + responseCreate4.getEntity()); - assertEquals(400, responseCreate4.getStatus()); - assertTrue(respMapCreate4.containsKey("error")); - assertTrue( ((String)respMapCreate4.get("error")).toLowerCase().indexOf("deadlock") > -1 ); - } - - - @SuppressWarnings("unchecked") - @Test - public void test_lockPromotion() throws Exception { - System.out.println("Testing lock promotion"); - createAndInsertIntoTable(); - insertAnotherIntoTable(); - - // creates a lock 1 - JsonLock jsonLock = createJsonLock(LockType.READ); - Response responseCreate1 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLock, "process1", appName); - Map respMapCreate1 = (Map) responseCreate1.getEntity(); - String lockRefCreate1 = ((Map) respMapCreate1.get("lock")).get("lock"); - - Response respMapPromote = lock.promoteLock(lockRefCreate1, "1", "1", authorization); - System.out.println("Status: " + respMapPromote.getStatus() + ". Entity " + respMapPromote.getEntity()); - - assertEquals(200, respMapPromote.getStatus()); - } - - @SuppressWarnings("unchecked") - @Test - public void test_lockPromotionReadWrite() throws Exception { - System.out.println("Testing lock promotion with read and writes"); - createAndInsertIntoTable(); - insertAnotherIntoTable(); - - // creates a lock 1 - JsonLock jsonLockRead = createJsonLock(LockType.READ); - Response responseCreate1 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockRead, "process1", appName); - Map respMapCreate1 = (Map) responseCreate1.getEntity(); - String lockRefCreate1 = ((Map) respMapCreate1.get("lock")).get("lock"); - - JsonLock jsonLockWrite = createJsonLock(LockType.WRITE); - Response responseCreate2 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockWrite, "process1", appName); - Map respMapCreate2 = (Map) responseCreate2.getEntity(); - String lockRefCreate2 = ((Map) respMapCreate2.get("lock")).get("lock"); - - Response respMapPromote = lock.promoteLock(lockRefCreate1, "1", "1", authorization); - System.out.println("Status: " + respMapPromote.getStatus() + ". Entity " + respMapPromote.getEntity()); - - assertEquals(200, respMapPromote.getStatus()); - } - - @SuppressWarnings("unchecked") - @Test - public void test_lockPromotionWriteRead() throws Exception { - System.out.println("Testing lock promotion with reads not at top of queue"); - createAndInsertIntoTable(); - insertAnotherIntoTable(); - - // creates a lock 1 - JsonLock jsonLockWrite = createJsonLock(LockType.WRITE); - Response responseCreate2 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockWrite, "process1", appName); - Map respMapCreate2 = (Map) responseCreate2.getEntity(); - String lockRefCreate2 = ((Map) respMapCreate2.get("lock")).get("lock"); - - // creates a lock 2 - JsonLock jsonLockRead = createJsonLock(LockType.READ); - Response responseCreate1 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockRead, "process1", appName); - Map respMapCreate1 = (Map) responseCreate1.getEntity(); - String lockRefCreate1 = ((Map) respMapCreate1.get("lock")).get("lock"); - - Response respMapPromote = lock.promoteLock(lockRefCreate1, "1", "1", authorization); - System.out.println("Status: " + respMapPromote.getStatus() + ". Entity " + respMapPromote.getEntity()); - - assertEquals(200, respMapPromote.getStatus()); - } - - @SuppressWarnings("unchecked") - @Test - public void test_lockPromotion2Reads() throws Exception { - System.out.println("Testing lock promotion w/ 2 ReadLocks"); - createAndInsertIntoTable(); - insertAnotherIntoTable(); - - // creates a lock 1 - JsonLock jsonLockRead = createJsonLock(LockType.READ); - Response responseCreate1 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockRead, "process1", appName); - Map respMapCreate1 = (Map) responseCreate1.getEntity(); - String lockRefCreate1 = ((Map) respMapCreate1.get("lock")).get("lock"); - - Response responseCreate2 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockRead, "process1", appName); - Map respMapCreate2 = (Map) responseCreate1.getEntity(); - String lockRefCreate2 = ((Map) respMapCreate1.get("lock")).get("lock"); - - Response respMapPromote = lock.promoteLock(lockRefCreate1, "1", "1", authorization); - System.out.println("Status: " + respMapPromote.getStatus() + ". Entity " + respMapPromote.getEntity()); - - assertEquals(400, respMapPromote.getStatus()); - } - - @SuppressWarnings("unchecked") - @Test - public void test_2lockPromotions() throws Exception { - System.out.println("Testing 2 lock promotions"); - createAndInsertIntoTable(); - insertAnotherIntoTable(); - - // creates a lock 1 - JsonLock jsonLockRead = createJsonLock(LockType.READ); - Response responseCreate1 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockRead, "process1", appName); - Map respMapCreate1 = (Map) responseCreate1.getEntity(); - String lockRefCreate1 = ((Map) respMapCreate1.get("lock")).get("lock"); - - Response responseCreate2 = lock.createLockReference(lockName, "1", "1", authorization, - "abcde001-d857-4e90-b1e5-df98a3d40ce6", jsonLockRead, "process1", appName); - Map respMapCreate2 = (Map) responseCreate2.getEntity(); - String lockRefCreate2 = ((Map) respMapCreate2.get("lock")).get("lock"); - - Response respMapPromote = lock.promoteLock(lockRefCreate1, "1", "1", authorization); - System.out.println("Status: " + respMapPromote.getStatus() + ". Entity " + respMapPromote.getEntity()); - - assertEquals(400, respMapPromote.getStatus()); - - Response respMap2Promote = lock.promoteLock(lockRefCreate2, "1", "1", authorization); - System.out.println("Status: " + respMap2Promote.getStatus() + ". Entity " + respMap2Promote.getEntity()); - - assertEquals(400, respMapPromote.getStatus()); - } - - - - // Ignoring since this is now a duplicate of delete lock ref. - @Test - @Ignore - public void test_deleteLock() throws Exception { - System.out.println("Testing get lock state"); - createAndInsertIntoTable(); - - String lockRef = createLockReference(); - - Response response = lock.deleteLock(lockName, "1", "1", - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", authorization, appName); - System.out.println("Status: " + response.getStatus() + ". Entity " + response.getEntity()); - assertEquals(200, response.getStatus()); - } - - /** - * Create table and lock reference - * - * @return the lock ref created - * @throws Exception - */ - @SuppressWarnings("unchecked") - private String createLockReference() throws Exception { - Response response = lock.createLockReference(lockName, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", null, null, appName); - Map respMap = (Map) response.getEntity(); - return ((Map) respMap.get("lock")).get("lock"); - } - - /** - * Create table and lock reference - * - * @return the lock ref created - * @throws Exception - */ - @SuppressWarnings("unchecked") - private String createLockReference(LockType lockType) throws Exception { - JsonLock jsonLock = createJsonLock(lockType); - Response response = lock.createLockReference(lockName, "1", "1", authorization, - "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", jsonLock, null, appName); - Map respMap = (Map) response.getEntity(); - return ((Map) respMap.get("lock")).get("lock"); - } - - private static void createKeyspace() throws Exception { - // shouldn't really be doing this here, but create keyspace is currently turned off - PreparedQueryObject query = new PreparedQueryObject(); - query.appendQueryString(CassandraCQL.createKeySpace); - MusicCore.eventualPut(query); - - boolean isAAF = false; - String hashedpwd = BCrypt.hashpw(password, BCrypt.gensalt()); - query = new PreparedQueryObject(); - query.appendQueryString("INSERT into admin.keyspace_master (uuid, keyspace_name, application_name, is_api, " - + "password, username, is_aaf) values (?,?,?,?,?,?,?)"); - query.addValue(MusicUtil.convertToActualDataType(DataType.uuid(), uuid)); - query.addValue(MusicUtil.convertToActualDataType(DataType.text(), keyspaceName)); - query.addValue(MusicUtil.convertToActualDataType(DataType.text(), appName)); - query.addValue(MusicUtil.convertToActualDataType(DataType.cboolean(), "True")); - query.addValue(MusicUtil.convertToActualDataType(DataType.text(), hashedpwd)); - query.addValue(MusicUtil.convertToActualDataType(DataType.text(), userId)); - query.addValue(MusicUtil.convertToActualDataType(DataType.cboolean(), isAAF)); - //CachingUtil.updateMusicCache(keyspaceName, appName); - //CachingUtil.updateMusicValidateCache(appName, userId, hashedpwd); - MusicCore.eventualPut(query); - } - - private void clearAllTablesFromKeyspace() throws MusicServiceException { - ArrayList tableNames = new ArrayList<>(); - PreparedQueryObject query = new PreparedQueryObject(); - query.appendQueryString( - "SELECT table_name FROM system_schema.tables WHERE keyspace_name = '" + keyspaceName + "';"); - ResultSet rs = MusicCore.get(query); - for (Row row : rs) { - tableNames.add(row.getString("table_name")); - } - for (String table : tableNames) { - query = new PreparedQueryObject(); - query.appendQueryString("DROP TABLE " + keyspaceName + "." + table); - MusicCore.eventualPut(query); - } - } - - /** - * Create a table {@link tableName} in {@link keyspaceName} - * - * @throws Exception - */ - private void createTable() throws Exception { - JsonTable jsonTable = new JsonTable(); - Map consistencyInfo = new HashMap<>(); - Map fields = new HashMap<>(); - fields.put("uuid", "text"); - fields.put("emp_name", "text"); - fields.put("emp_salary", "varint"); - fields.put("PRIMARY KEY", "(emp_name)"); - consistencyInfo.put("type", "eventual"); - jsonTable.setConsistencyInfo(consistencyInfo); - jsonTable.setKeyspaceName(keyspaceName); - jsonTable.setPrimaryKey("emp_name"); - jsonTable.setTableName(tableName); - jsonTable.setFields(fields); - // Mockito.doNothing().when(http).addHeader(xLatestVersion, MusicUtil.getVersion()); - Response response = data.createTable("1", "1", "1", "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName, - authorization, jsonTable, keyspaceName, tableName); - } - - /** - * Create table {@link createTable} and insert into said table - * - * @throws Exception - */ - private void createAndInsertIntoTable() throws Exception { - createTable(); - - JsonInsert jsonInsert = new JsonInsert(); - Map consistencyInfo = new HashMap<>(); - Map values = new HashMap<>(); - values.put("uuid", "cfd66ccc-d857-4e90-b1e5-df98a3d40cd6"); - values.put("emp_name", "testname"); - values.put("emp_salary", 500); - consistencyInfo.put("type", "eventual"); - jsonInsert.setConsistencyInfo(consistencyInfo); - jsonInsert.setKeyspaceName(keyspaceName); - jsonInsert.setTableName(tableName); - jsonInsert.setValues(values); - Response response = data.insertIntoTable("1", "1", "1", "abc66ccc-d857-4e90-b1e5-df98a3d40ce6", appName, - authorization, jsonInsert, keyspaceName, tableName); - } - - private void insertAnotherIntoTable() throws Exception { - createTable(); - - JsonInsert jsonInsert = new JsonInsert(); - Map consistencyInfo = new HashMap<>(); - Map values = new HashMap<>(); - values.put("uuid", "cccccccc-d857-4e90-b1e5-df98a3d40cd6"); - values.put("emp_name", "testname2"); - values.put("emp_salary", 700); - consistencyInfo.put("type", "eventual"); - jsonInsert.setConsistencyInfo(consistencyInfo); - jsonInsert.setKeyspaceName(keyspaceName); - jsonInsert.setTableName(tableName); - jsonInsert.setValues(values); - Response response = data.insertIntoTable("1", "1", "1", "abcdef00-d857-4e90-b1e5-df98a3d40ce6", appName, - authorization, jsonInsert, keyspaceName, tableName); - } - - private JsonLock createJsonLock(LockType lockType) { - JsonLock jsonLock = new JsonLock(); - jsonLock.setLockType(lockType); - return jsonLock; - } - -} \ No newline at end of file