cadi/client/src/test/java/org/onap/aaf/cadi/client/test/JU_AbsAuthentication.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.cadi.client.test;
  23
  24 import static org.junit.Assert.*;
  25 import static org.hamcrest.CoreMatchers.*;
  26 import org.junit.*;
  27 import org.onap.aaf.cadi.CadiException;
  28 import org.onap.aaf.cadi.client.AbsAuthentication;
  29 import org.onap.aaf.cadi.config.SecurityInfoC;
  30
  31 import java.io.ByteArrayOutputStream;
  32 import java.io.IOException;
  33 import java.io.PrintStream;
  34 import java.net.HttpURLConnection;
  35
  36 public class JU_AbsAuthentication {
  37
  38         private final static String ID = "id";
  39         private final static String PASSWORD = "password";
  40         private final static String WARNING = "Your service has 1000 consecutive bad service " +
  41                                                                                         "logins to AAF. AAF Access will be disabled after 10000\n";
  42
  43         private static ByteArrayOutputStream errStream;
  44
  45         @Before
  46         public void setup() {
  47                 errStream = new ByteArrayOutputStream();
  48                 System.setErr(new PrintStream(errStream));
  49         }
  50
  51         @After
  52         public void tearDown() {
  53                 System.setErr(System.err);
  54         }
  55
  56         @Test
  57         public void test() throws IOException, InterruptedException {
  58                 AuthStub stub = new AuthStub(null, null, null);
  59                 assertThat(stub.getID(), is(nullValue()));
  60                 assertThat(stub.headValue(), is(""));
  61                 assertThat(stub.count(), is(0));
  62
  63                 stub.setUser(ID);
  64                 assertThat(stub.getID(), is(ID));
  65
  66                 stub = new AuthStub(null, ID, PASSWORD.getBytes());
  67                 assertThat(stub.getID(), is(ID));
  68                 assertThat(stub.headValue(), is(PASSWORD));
  69                 assertThat(stub.count(), is(0));
  70
  71                 assertThat(stub.setLastResponse(200), is(0));
  72                 assertThat(stub.isDenied(), is(false));
  73
  74                 for (int i = 1; i <= 10; i++) {
  75                         assertThat(stub.setLastResponse(401), is(i));
  76                         assertThat(stub.isDenied(), is(false));
  77                 }
  78                 assertThat(stub.setLastResponse(401), is(11));
  79                 assertThat(stub.isDenied(), is(true));
  80
  81                 stub.setCount(999);
  82                 assertThat(stub.setLastResponse(401), is(1000));
  83                 assertThat(errStream.toString(), is(WARNING));
  84
  85                 // coverage...
  86                 stub.setLastMiss(1);
  87                 assertThat(stub.isDenied(), is(false));
  88         }
  89
  90         private class AuthStub extends AbsAuthentication<HttpURLConnection> {
  91
  92                 public AuthStub(SecurityInfoC<HttpURLConnection> securityInfo, String user, byte[] headValue)
  93                                 throws IOException { super(securityInfo, user, headValue); }
  94
  95                 @Override public void setSecurity(HttpURLConnection client) throws CadiException { }
  96                 @Override public void setUser(String id) { super.setUser(id); }
  97                 @Override public String headValue() throws IOException { return super.headValue(); }
  98
  99                 public void setLastMiss(long lastMiss) { this.lastMiss = lastMiss; }
 100                 public void setCount(int count) { this.count = count; }
 101         }
 102
 103 }