2 * ============LICENSE_START====================================================
4 * ===========================================================================
5 * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
6 * ===========================================================================
7 * Modification Copyright © 2020 IBM.
8 * ===========================================================================
10 * Licensed under the Apache License, Version 2.0 (the "License");
11 * you may not use this file except in compliance with the License.
12 * You may obtain a copy of the License at
14 * http://www.apache.org/licenses/LICENSE-2.0
16 * Unless required by applicable law or agreed to in writing, software
17 * distributed under the License is distributed on an "AS IS" BASIS,
18 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19 * See the License for the specific language governing permissions and
20 * limitations under the License.
21 * ============LICENSE_END====================================================
25 package org.onap.aaf.auth.batch.helpers;
27 import static org.junit.Assert.assertEquals;
28 import static org.mockito.MockitoAnnotations.initMocks;
30 import java.io.IOException;
31 import java.nio.ByteBuffer;
32 import java.util.Date;
33 import java.util.List;
34 import java.util.UUID;
36 import org.junit.Before;
37 import org.junit.Test;
38 import org.mockito.Mock;
39 import org.mockito.Mockito;
40 import org.onap.aaf.auth.dao.cass.ApprovalDAO;
41 import org.onap.aaf.auth.dao.cass.FutureDAO;
42 import org.onap.aaf.auth.dao.cass.NsDAO;
43 import org.onap.aaf.auth.dao.cass.RoleDAO;
44 import org.onap.aaf.auth.dao.cass.UserRoleDAO;
45 import org.onap.aaf.auth.env.AuthzTrans;
46 import org.onap.aaf.auth.layer.Result;
47 import org.onap.aaf.cadi.PropAccess;
48 import org.onap.aaf.misc.env.APIException;
49 import org.onap.aaf.misc.env.TimeTaken;
51 import com.datastax.driver.core.Session;
53 public class JU_BatchDataViewTest {
64 BatchDataView batchDataViewObj;
67 public void setUp() throws APIException, IOException {
69 batchDataViewObj = new BatchDataView(trans, session, true);
70 Mockito.doReturn(Mockito.mock(TimeTaken.class)).when(trans).start(
71 Mockito.anyString(), Mockito.anyInt(), Mockito.anyString());
75 public void testNs() {
76 Result<NsDAO.Data> retVal = batchDataViewObj.ns(trans, "test");
77 assertEquals(9,retVal.status);
79 NS n = new NS("test1", "test2", "test3", 1, 2);
80 NS.data.put("test", n);
81 retVal = batchDataViewObj.ns(trans, "test");
82 assertEquals(0,retVal.status);
86 public void testRoleByName() {
87 Result<RoleDAO.Data> retVal = batchDataViewObj.roleByName(trans,
89 assertEquals(9,retVal.status);
91 Role n = new Role("test1");
92 n.rdd = new RoleDAO.Data();
93 Role.byName.put("test", n);
94 retVal = batchDataViewObj.roleByName(trans, "test");
95 assertEquals(0,retVal.status);
98 Role.byName.put("test", n);
99 retVal = batchDataViewObj.roleByName(trans, "test");
100 assertEquals(9,retVal.status);
103 public void testUrsByRole() {
104 Result<List<UserRoleDAO.Data>> retVal = batchDataViewObj
105 .ursByRole(trans, "test");
106 assertEquals(9,retVal.status);
108 Role n = new Role("test1");
109 n.rdd = new RoleDAO.Data();
110 UserRole ur = new UserRole("user", "role", "ns", "rname", new Date());
111 (new UserRole.DataLoadVisitor()).visit(ur);
112 retVal = batchDataViewObj.ursByRole(trans, "role");
113 assertEquals(retVal.status,0);
117 public void testUrsByUser() {
118 Result<List<UserRoleDAO.Data>> retVal = batchDataViewObj
119 .ursByUser(trans, "test");
120 assertEquals(retVal.status,9);
122 Role n = new Role("test1");
123 n.rdd = new RoleDAO.Data();
124 UserRole ur = new UserRole("user", "role", "ns", "rname", new Date());
125 (new UserRole.DataLoadVisitor()).visit(ur);
126 retVal = batchDataViewObj.ursByUser(trans, "user");
127 assertEquals(retVal.status,0);
131 public void testDeleteFuture() {
132 FutureDAO.Data dataObj = new FutureDAO.Data();
133 dataObj.id = new UUID(1000L, 1000L);
134 Result<FutureDAO.Data> retVal = batchDataViewObj.delete(trans, dataObj);
135 assertEquals(retVal.status,0);
139 public void testDeleteApproval() {
140 ApprovalDAO.Data dataObj = new ApprovalDAO.Data();
141 dataObj.id = new UUID(1000L, 1000L);
142 Result<ApprovalDAO.Data> retVal = batchDataViewObj.delete(trans,
144 assertEquals(retVal.status, 0);
149 public void testInsertApproval() {
150 ApprovalDAO.Data dataObj = new ApprovalDAO.Data();
151 dataObj.id = new UUID(1000L, 1000L);
152 dataObj.memo = "memo";
153 dataObj.ticket = new UUID(1000L, 1000L);
154 Result<ApprovalDAO.Data> retVal = batchDataViewObj.insert(trans,
156 assertEquals(retVal.status, 0);
160 public void testInsertFuture() {
161 FutureDAO.Data dataObj = new FutureDAO.Data();
162 dataObj.id = new UUID(1000L, 1000L);
163 dataObj.memo = "memo";
164 dataObj.construct = ByteBuffer.allocate(1000);
165 Result<FutureDAO.Data> retVal = batchDataViewObj.insert(trans, dataObj);
166 assertEquals(retVal.status, 0);
168 dataObj.target_key = "memo";
169 retVal = batchDataViewObj.insert(trans, dataObj);
170 assertEquals(retVal.status, 0);
173 public void testFlush() {
174 batchDataViewObj.flush();