content: /*
   * Licensed to the Apache Software Foundation (ASF) under one
   * or more contributor license agreements.  See the NOTICE file
   * distributed with this work for additional information
   * regarding copyright ownership.  The ASF licenses this file
   * to you 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.
   */
  
  package org.apache.ambari.server.audit;
  
  
  import static org.hamcrest.core.IsEqual.equalTo;
  import static org.junit.Assert.assertThat;
  
  import java.util.Arrays;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.apache.ambari.server.audit.event.LoginAuditEvent;
  import org.junit.Test;
  
  import nl.jqno.equalsverifier.EqualsVerifier;
  
  public class LoginAuditEventTest {
  
    @Test
    public void testAuditMessage() throws Exception {
      // Given
      String testUserName = "USER1";
      String testRemoteIp = "127.0.0.1";
      String testProxyUserName = "PROXYUSER1";
  
      Map<String, List<String>> roles = new HashMap<>();
      roles.put("a", Arrays.asList("r1", "r2", "r3"));
  
      LoginAuditEvent evnt = LoginAuditEvent.builder()
        .withTimestamp(System.currentTimeMillis())
        .withRemoteIp(testRemoteIp)
        .withUserName(testUserName)
        .withProxyUserName(null)
        .withRoles(roles)
        .build();
  
      // When
      String actualAuditMessage = evnt.getAuditMessage();
  
      String roleMessage =  "a: r1, r2, r3";
  
      // Then
      String expectedAuditMessage = String.format("User(%s), RemoteIp(%s), Operation(User login), Roles(%s), Status(Success)",
        testUserName, testRemoteIp, roleMessage);
  
      assertThat(actualAuditMessage, equalTo(expectedAuditMessage));
  
      evnt = LoginAuditEvent.builder()
        .withTimestamp(System.currentTimeMillis())
        .withRemoteIp(testRemoteIp)
        .withUserName(testUserName)
        .withProxyUserName(testProxyUserName)
        .withRoles(roles)
        .build();
  
      // When
      actualAuditMessage = evnt.getAuditMessage();
  
      roleMessage = "a: r1, r2, r3";
  
      // Then
      expectedAuditMessage = String.format("User(%s), RemoteIp(%s), ProxyUser(%s), Operation(User login), Roles(%s), Status(Success)",
        testUserName, testRemoteIp, testProxyUserName, roleMessage);
  
      assertThat(actualAuditMessage, equalTo(expectedAuditMessage));
  
    }
  
    @Test
    public void testFailedAuditMessage() throws Exception {
      // Given
      String testUserName = "USER1";
      String testRemoteIp = "127.0.0.1";
      String testProxyUserName = "PROXYUSER1";
      String reason = "Bad credentials";
      Integer consecutiveFailures = 1;
  
      Map<String, List<String>> roles = new HashMap<>();
      roles.put("a", Arrays.asList("r1", "r2", "r3"));
  
      LoginAuditEvent evnt = LoginAuditEvent.builder()
        .withTimestamp(System.currentTimeMillis())
        .withRemoteIp(testRemoteIp)
        .withUserName(testUserName)
        .withProxyUserName(null)
        .withRoles(roles)
        .withReasonOfFailure(reason)
        .withConsecutiveFailures(consecutiveFailures)
        .build();
  
      // When
      String actualAuditMessage = evnt.getAuditMessage();
  
      String roleMessage =  "a: r1, r2, r3" ;
  
      // Then
      String expectedAuditMessage = String.format("User(%s), RemoteIp(%s), Operation(User login), Roles(%s), Status(Failed), Reason(%s), Consecutive failures(%d)",
        testUserName, testRemoteIp, roleMessage, reason, consecutiveFailures);
  
      assertThat(actualAuditMessage, equalTo(expectedAuditMessage));
  
      evnt = LoginAuditEvent.builder()
          .withTimestamp(System.currentTimeMillis())
          .withRemoteIp(testRemoteIp)
          .withUserName(testUserName)
          .withProxyUserName(testProxyUserName)
          .withRoles(roles)
          .withReasonOfFailure(reason)
          .withConsecutiveFailures(consecutiveFailures)
          .build();
  
      // When
      actualAuditMessage = evnt.getAuditMessage();
  
      roleMessage = "a: r1, r2, r3" ;
  
      // Then
      expectedAuditMessage = String.format("User(%s), RemoteIp(%s), ProxyUser(%s), Operation(User login), Roles(%s), Status(Failed), Reason(%s), Consecutive failures(%d)",
          testUserName, testRemoteIp, testProxyUserName, roleMessage, reason, consecutiveFailures);
  
      assertThat(actualAuditMessage, equalTo(expectedAuditMessage));
    }
  
    @Test
    public void testFailedAuditMessageUnknownUser() throws Exception {
      // Given
      String testUserName = "USER1";
      String testRemoteIp = "127.0.0.1";
      String reason = "Bad credentials";
  
      String testProxyUserName = "PROXYUSER1";
  
      Map<String, List<String>> roles = new HashMap<>();
      roles.put("a", Arrays.asList("r1", "r2", "r3"));
  
      LoginAuditEvent evnt = LoginAuditEvent.builder()
        .withTimestamp(System.currentTimeMillis())
        .withRemoteIp(testRemoteIp)
        .withUserName(testUserName)
        .withProxyUserName(null)
        .withRoles(roles)
        .withReasonOfFailure(reason)
        .withConsecutiveFailures(null)
        .build();
  
      // When
      String actualAuditMessage = evnt.getAuditMessage();
  
      String roleMessage = "a: r1, r2, r3" ;
  
      // Then
      String expectedAuditMessage = String.format("User(%s), RemoteIp(%s), Operation(User login), Roles(%s), Status(Failed), Reason(%s), Consecutive failures(UNKNOWN USER)",
        testUserName, testRemoteIp, roleMessage, reason);
  
      assertThat(actualAuditMessage, equalTo(expectedAuditMessage));
  
      evnt = LoginAuditEvent.builder()
          .withTimestamp(System.currentTimeMillis())
          .withRemoteIp(testRemoteIp)
          .withUserName(testUserName)
          .withProxyUserName(testProxyUserName)
          .withRoles(roles)
          .withReasonOfFailure(reason)
          .withConsecutiveFailures(null)
          .build();
  
      // When
      actualAuditMessage = evnt.getAuditMessage();
  
      roleMessage =  "a: r1, r2, r3";
  
      // Then
      expectedAuditMessage = String.format("User(%s), RemoteIp(%s), ProxyUser(%s), Operation(User login), Roles(%s), Status(Failed), Reason(%s), Consecutive failures(UNKNOWN USER)",
          testUserName, testRemoteIp, testProxyUserName, roleMessage, reason);
  
      assertThat(actualAuditMessage, equalTo(expectedAuditMessage));
    }
  
    @Test
    public void testTimestamp() throws Exception {
      // Given
      long testTimestamp = System.currentTimeMillis();
      LoginAuditEvent evnt = LoginAuditEvent.builder()
        .withTimestamp(testTimestamp)
        .build();
  
      // When
      long actualTimestamp = evnt.getTimestamp();
  
      // Then
      assertThat(actualTimestamp, equalTo(testTimestamp));
  
    }
  
    @Test
    public void testEquals() throws Exception {
      EqualsVerifier.forClass(LoginAuditEvent.class)
        .verify();
    }
  }
  