core/src/test/java/com/att/cadi/lur/test/JU_LocalLur.java

   1 /*******************************************************************************\r
   2  * ============LICENSE_START====================================================\r
   3  * * org.onap.aaf\r
   4  * * ===========================================================================\r
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * * ===========================================================================\r
   7  * * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * * you may not use this file except in compliance with the License.\r
   9  * * You may obtain a copy of the License at\r
  10  * * \r
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * * \r
  13  *  * Unless required by applicable law or agreed to in writing, software\r
  14  * * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * * See the License for the specific language governing permissions and\r
  17  * * limitations under the License.\r
  18  * * ============LICENSE_END====================================================\r
  19  * *\r
  20  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
  21  * *\r
  22  ******************************************************************************/\r
  23 package com.att.cadi.lur.test;\r
  24 \r
  25 import static junit.framework.Assert.assertEquals;\r
  26 import static junit.framework.Assert.assertFalse;\r
  27 import static junit.framework.Assert.assertTrue;\r
  28 \r
  29 import java.io.ByteArrayOutputStream;\r
  30 import java.io.IOException;\r
  31 import java.security.Principal;\r
  32 import java.util.ArrayList;\r
  33 import java.util.List;\r
  34 import java.util.Set;\r
  35 import java.util.TreeSet;\r
  36 \r
  37 import org.junit.Test;\r
  38 \r
  39 import com.att.cadi.CredVal.Type;\r
  40 import com.att.cadi.Lur;\r
  41 import com.att.cadi.Permission;\r
  42 import com.att.cadi.PropAccess;\r
  43 import com.att.cadi.Symm;\r
  44 import com.att.cadi.config.UsersDump;\r
  45 import com.att.cadi.lur.LocalLur;\r
  46 import com.att.cadi.lur.LocalPermission;\r
  47 \r
  48 public class JU_LocalLur {\r
  49 \r
  50         @Test\r
  51         public void test() throws IOException {\r
  52                 Symm symmetric = Symm.baseCrypt().obtain();\r
  53                 LocalLur up;\r
  54                 ByteArrayOutputStream baos = new ByteArrayOutputStream();\r
  55                 baos.write(Symm.ENC.getBytes());\r
  56                 symmetric.enpass("<pass>", baos);\r
  57                 PropAccess ta = new PropAccess();\r
  58                 Lur ml = up = new LocalLur(ta,"myname:groupA,groupB","admin:myname,yourname;suser:hisname,hername,m1234%"+baos.toString());\r
  59                 \r
  60                 Permission admin = new LocalPermission("admin");\r
  61                 Permission suser = new LocalPermission("suser");\r
  62                 \r
  63                 // Check User fish\r
  64                 assertTrue(ml.fish(new JUPrincipal("myname"),admin));\r
  65                 assertTrue(ml.fish(new JUPrincipal("hisname"),admin));\r
  66                 assertFalse(ml.fish(new JUPrincipal("noname"),admin));\r
  67                 assertTrue(ml.fish(new JUPrincipal("itsname"),suser));\r
  68                 assertTrue(ml.fish(new JUPrincipal("hername"),suser));\r
  69                 assertFalse(ml.fish(new JUPrincipal("myname"),suser));\r
  70                 \r
  71                 \r
  72                 // Check validate password\r
  73                 assertTrue(up.validate("m1234",Type.PASSWORD, "<pass>".getBytes()));\r
  74                 assertFalse(up.validate("m1234",Type.PASSWORD, "badPass".getBytes()));\r
  75                 \r
  76                 // Check fishAll\r
  77                 Set<String> set = new TreeSet<String>();\r
  78                 List<Permission> perms = new ArrayList<Permission>();\r
  79                 ml.fishAll(new JUPrincipal("myname"), perms);\r
  80                 for(Permission p : perms) {\r
  81                         set.add(p.getKey());\r
  82                 }\r
  83                 assertEquals("[admin, groupA, groupB]",set.toString());\r
  84                 UsersDump.write(System.out, up);\r
  85                 System.out.flush();\r
  86                 \r
  87         }\r
  88         \r
  89         // Simplistic Principal for testing purposes\r
  90         private static class JUPrincipal implements Principal {\r
  91                 private String name;\r
  92                 public JUPrincipal(String name) {\r
  93                         this.name = name;\r
  94                 }\r
  95 //              @Override\r
  96                 public String getName() {\r
  97                         return name;\r
  98                 }\r
  99         }\r
 100 \r
 101 }\r