authz-cass/src/test/java/com/att/dao/aaf/test/JU_NsDAO.java

   1 /*******************************************************************************\r
   2  * ============LICENSE_START====================================================\r
   3  * * org.onap.aai\r
   4  * * ===========================================================================\r
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * * Copyright © 2017 Amdocs\r
   7  * * ===========================================================================\r
   8  * * Licensed under the Apache License, Version 2.0 (the "License");\r
   9  * * you may not use this file except in compliance with the License.\r
  10  * * You may obtain a copy of the License at\r
  11  * * \r
  12  *  *      http://www.apache.org/licenses/LICENSE-2.0\r
  13  * * \r
  14  *  * Unless required by applicable law or agreed to in writing, software\r
  15  * * distributed under the License is distributed on an "AS IS" BASIS,\r
  16  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  17  * * See the License for the specific language governing permissions and\r
  18  * * limitations under the License.\r
  19  * * ============LICENSE_END====================================================\r
  20  * *\r
  21  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
  22  * *\r
  23  ******************************************************************************/\r
  24 package com.att.dao.aaf.test;\r
  25 \r
  26 import static org.junit.Assert.assertEquals;\r
  27 import static org.junit.Assert.assertFalse;\r
  28 import static org.junit.Assert.assertTrue;\r
  29 \r
  30 import java.io.IOException;\r
  31 import java.nio.ByteBuffer;\r
  32 import java.util.HashMap;\r
  33 import java.util.List;\r
  34 import java.util.Map;\r
  35 import java.util.Map.Entry;\r
  36 import java.util.Set;\r
  37 \r
  38 import org.junit.Test;\r
  39 \r
  40 import com.att.authz.layer.Result;\r
  41 import com.att.dao.aaf.cass.NsDAO;\r
  42 import com.att.dao.aaf.cass.NsDAO.Data;\r
  43 import com.att.dao.aaf.cass.NsType;\r
  44 import com.att.inno.env.APIException;\r
  45 \r
  46 \r
  47 public class JU_NsDAO extends AbsJUCass {\r
  48         private static final String CRM = "ju_crm";\r
  49         private static final String SWM = "ju_swm";\r
  50 \r
  51         @Test\r
  52         public void test() throws APIException, IOException  {\r
  53                 NsDAO nsd = new NsDAO(trans, cluster, AUTHZ);\r
  54                 try {\r
  55                         final String nsparent = "com.test";\r
  56                         final String ns1 = nsparent +".ju_ns";\r
  57                         final String ns2 = nsparent + ".ju_ns2";\r
  58                         \r
  59                         Map<String,String> oAttribs = new HashMap<String,String>();\r
  60                         oAttribs.put(SWM, "swm_data");\r
  61                         oAttribs.put(CRM, "crm_data");\r
  62                         Data data = new NsDAO.Data();\r
  63                         data.name = ns1;\r
  64                         data.type = NsType.APP.type;\r
  65                         data.attrib(true).putAll(oAttribs);\r
  66                         \r
  67 \r
  68                         Result<List<Data>> rdrr;\r
  69 \r
  70                         // CREATE\r
  71                         Result<Data> rdc = nsd.create(trans, data);\r
  72                         assertTrue(rdc.isOK());\r
  73                         \r
  74                         try {\r
  75 //                      Bytification\r
  76                         ByteBuffer bb = data.bytify();\r
  77                         Data bdata = new NsDAO.Data();\r
  78                         bdata.reconstitute(bb);\r
  79                         compare(data, bdata);\r
  80 \r
  81                                 // Test READ by Object\r
  82                                 rdrr = nsd.read(trans, data);\r
  83                                 assertTrue(rdrr.isOKhasData());\r
  84                                 assertEquals(rdrr.value.size(),1);\r
  85                                 Data d = rdrr.value.get(0);\r
  86                                 assertEquals(d.name,data.name);\r
  87                                 assertEquals(d.type,data.type);\r
  88                                 attribsEqual(d.attrib(false),data.attrib(false));\r
  89                                 attribsEqual(oAttribs,data.attrib(false));\r
  90                                 \r
  91                                 // Test Read by Key\r
  92                                 rdrr = nsd.read(trans, data.name);\r
  93                                 assertTrue(rdrr.isOKhasData());\r
  94                                 assertEquals(rdrr.value.size(),1);\r
  95                                 d = rdrr.value.get(0);\r
  96                                 assertEquals(d.name,data.name);\r
  97                                 assertEquals(d.type,data.type);\r
  98                                 attribsEqual(d.attrib(false),data.attrib(false));\r
  99                                 attribsEqual(oAttribs,data.attrib(false));\r
 100                                 \r
 101                                 // Read NS by Type\r
 102                                 Result<Set<String>> rtypes = nsd.readNsByAttrib(trans, SWM);\r
 103                                 Set<String> types;\r
 104                                 if(rtypes.notOK()) {\r
 105                                         throw new IOException(rtypes.errorString());\r
 106                                 } else {\r
 107                                         types = rtypes.value;\r
 108                                 }\r
 109                                 assertEquals(1,types.size());\r
 110                                 assertEquals(true,types.contains(ns1));\r
 111                                 \r
 112                                 // Add second NS to test list of data returned\r
 113                                 Data data2 = new NsDAO.Data();\r
 114                                 data2.name = ns2;\r
 115                                 data2.type = 3; // app\r
 116                                 Result<Data> rdc2 = nsd.create(trans, data2);\r
 117                                 assertTrue(rdc2.isOK());\r
 118                                 \r
 119                                         // Interrupt - test PARENT\r
 120                                         Result<List<Data>> rdchildren = nsd.getChildren(trans, "com.test");\r
 121                                         assertTrue(rdchildren.isOKhasData());\r
 122                                         boolean child1 = false;\r
 123                                         boolean child2 = false;\r
 124                                         for(Data dchild : rdchildren.value) {\r
 125                                                 if(ns1.equals(dchild.name))child1=true;\r
 126                                                 if(ns2.equals(dchild.name))child2=true;\r
 127                                         }\r
 128                                         assertTrue(child1);\r
 129                                         assertTrue(child2);\r
 130 \r
 131                                 // FINISH DATA 2 by deleting\r
 132                                 Result<Void> rddr = nsd.delete(trans, data2, true);\r
 133                                 assertTrue(rddr.isOK());\r
 134 \r
 135                                 // ADD DESCRIPTION\r
 136                                 String description = "This is my test Namespace";\r
 137                                 assertFalse(description.equalsIgnoreCase(data.description));\r
 138                                 \r
 139                                 Result<Void> addDesc = nsd.addDescription(trans, data.name, description);\r
 140                                 assertTrue(addDesc.isOK());\r
 141                                 rdrr = nsd.read(trans, data);\r
 142                                 assertTrue(rdrr.isOKhasData());\r
 143                                 assertEquals(rdrr.value.size(),1);\r
 144                                 assertEquals(rdrr.value.get(0).description,description);\r
 145                                 \r
 146                                 // UPDATE\r
 147                                 String newDescription = "zz1234 Owns This Namespace Now";\r
 148                                 oAttribs.put("mso", "mso_data");\r
 149                                 data.attrib(true).put("mso", "mso_data");\r
 150                                 data.description = newDescription;\r
 151                                 Result<Void> update = nsd.update(trans, data);\r
 152                                 assertTrue(update.isOK());\r
 153                                 rdrr = nsd.read(trans, data);\r
 154                                 assertTrue(rdrr.isOKhasData());\r
 155                                 assertEquals(rdrr.value.size(),1);\r
 156                                 assertEquals(rdrr.value.get(0).description,newDescription);\r
 157                                 attribsEqual(oAttribs, rdrr.value.get(0).attrib);\r
 158                                 \r
 159                                 \r
 160                         } catch (IOException e) {\r
 161                                 e.printStackTrace();\r
 162                         } finally {\r
 163                                 // DELETE\r
 164                                 Result<Void> rddr = nsd.delete(trans, data, true);\r
 165                                 assertTrue(rddr.isOK());\r
 166                                 rdrr = nsd.read(trans, data);\r
 167                                 assertTrue(rdrr.isOK() && rdrr.isEmpty());\r
 168                                 assertEquals(rdrr.value.size(),0);\r
 169                         }\r
 170                 } finally {\r
 171                         nsd.close(trans);\r
 172                 }\r
 173         }\r
 174 \r
 175         private void compare(NsDAO.Data d, NsDAO.Data data) {\r
 176                 assertEquals(d.name,data.name);\r
 177                 assertEquals(d.type,data.type);\r
 178                 attribsEqual(d.attrib(false),data.attrib(false));\r
 179                 attribsEqual(d.attrib(false),data.attrib(false));\r
 180         }\r
 181         \r
 182         private void attribsEqual(Map<String,String> aa, Map<String,String> ba) {\r
 183                 assertEquals(aa.size(),ba.size());\r
 184                 for(Entry<String, String> es : aa.entrySet()) {\r
 185                         assertEquals(es.getValue(),ba.get(es.getKey()));\r
 186                 }\r
 187         }\r
 188 }\r