How to use

getUserToken

method

in

org.killbill.billing.util.audit.AuditLog

  protected AuditLog createAuditLog() {
    final AuditLog auditLog = Mockito.mock(AuditLog.class);
    final DateTime utcNow = clock.getUTCNow();
    Mockito.when(auditLog.getCreatedDate()).thenReturn(utcNow);
    Mockito.when(auditLog.getReasonCode()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getUserName()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getUserToken()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getComment()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getChangeType()).thenReturn(ChangeType.DELETE);

    return auditLog;
  }
}

  private void verifyAuditLogsForTag(final List<AuditLog> auditLogs, final AuditLevel level) {
    if (AuditLevel.NONE.equals(level)) {
      Assert.assertEquals(auditLogs.size(), 0);
      return;
    }

    Assert.assertEquals(auditLogs.size(), 1);
    Assert.assertEquals(auditLogs.get(0).getUserToken(), internalCallContext.getUserToken().toString());
    Assert.assertEquals(auditLogs.get(0).getChangeType(), ChangeType.INSERT);
    Assert.assertEquals(auditLogs.get(0).getComment(), internalCallContext.getComments());
    Assert.assertEquals(auditLogs.get(0).getReasonCode(), internalCallContext.getReasonCode());
    Assert.assertEquals(auditLogs.get(0).getUserName(), internalCallContext.getCreatedBy());
    Assert.assertNotNull(auditLogs.get(0).getCreatedDate());
  }
}

@Test(groups = "fast")
public void testGetters() throws Exception {
  final TableName tableName = TableName.ACCOUNT_EMAIL_HISTORY;
  final long recordId = Long.MAX_VALUE;
  final ChangeType changeType = ChangeType.DELETE;
  final EntityAudit entityAudit = new EntityAudit(tableName, recordId, changeType, null);
  final UUID accountId = UUID.randomUUID();
  final UUID tenantId = UUID.randomUUID();
  final String userName = UUID.randomUUID().toString();
  final CallOrigin callOrigin = CallOrigin.EXTERNAL;
  final UserType userType = UserType.CUSTOMER;
  final UUID userToken = UUID.randomUUID();
  final DefaultCallContext callContext = new DefaultCallContext(accountId, tenantId, userName, callOrigin, userType, userToken, clock);
  final AuditLog auditLog = new DefaultAuditLog(new AuditLogModelDao(entityAudit, callContext), ObjectType.ACCOUNT_EMAIL, UUID.randomUUID());
  Assert.assertEquals(auditLog.getChangeType(), changeType);
  Assert.assertNull(auditLog.getComment());
  Assert.assertNotNull(auditLog.getCreatedDate());
  Assert.assertNull(auditLog.getReasonCode());
  Assert.assertEquals(auditLog.getUserName(), userName);
  Assert.assertEquals(auditLog.getUserToken(), userToken.toString());
}

  protected AuditLog createAuditLog() {
    final AuditLog auditLog = Mockito.mock(AuditLog.class);
    final DateTime utcNow = clock.getUTCNow();
    Mockito.when(auditLog.getCreatedDate()).thenReturn(utcNow);
    Mockito.when(auditLog.getReasonCode()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getUserName()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getUserToken()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getComment()).thenReturn(UUID.randomUUID().toString());
    Mockito.when(auditLog.getChangeType()).thenReturn(ChangeType.DELETE);

    return auditLog;
  }
}

  private void verifyAuditLogsForTag(final List<AuditLog> auditLogs, final AuditLevel level) {
    if (AuditLevel.NONE.equals(level)) {
      Assert.assertEquals(auditLogs.size(), 0);
      return;
    }

    Assert.assertEquals(auditLogs.size(), 1);
    Assert.assertEquals(auditLogs.get(0).getUserToken(), internalCallContext.getUserToken().toString());
    Assert.assertEquals(auditLogs.get(0).getChangeType(), ChangeType.INSERT);
    Assert.assertEquals(auditLogs.get(0).getComment(), internalCallContext.getComments());
    Assert.assertEquals(auditLogs.get(0).getReasonCode(), internalCallContext.getReasonCode());
    Assert.assertEquals(auditLogs.get(0).getUserName(), internalCallContext.getCreatedBy());
    Assert.assertNotNull(auditLogs.get(0).getCreatedDate());
  }
}

@Test(groups = "fast")
public void testGetters() throws Exception {
  final TableName tableName = TableName.ACCOUNT_EMAIL_HISTORY;
  final long recordId = Long.MAX_VALUE;
  final ChangeType changeType = ChangeType.DELETE;
  final EntityAudit entityAudit = new EntityAudit(tableName, recordId, changeType, null);
  final UUID accountId = UUID.randomUUID();
  final UUID tenantId = UUID.randomUUID();
  final String userName = UUID.randomUUID().toString();
  final CallOrigin callOrigin = CallOrigin.EXTERNAL;
  final UserType userType = UserType.CUSTOMER;
  final UUID userToken = UUID.randomUUID();
  final DefaultCallContext callContext = new DefaultCallContext(accountId, tenantId, userName, callOrigin, userType, userToken, clock);
  final AuditLog auditLog = new DefaultAuditLog(new AuditLogModelDao(entityAudit, callContext), ObjectType.ACCOUNT_EMAIL, UUID.randomUUID());
  Assert.assertEquals(auditLog.getChangeType(), changeType);
  Assert.assertNull(auditLog.getComment());
  Assert.assertNotNull(auditLog.getCreatedDate());
  Assert.assertNull(auditLog.getReasonCode());
  Assert.assertEquals(auditLog.getUserName(), userName);
  Assert.assertEquals(auditLog.getUserToken(), userToken.toString());
}

private <T extends EntitySqlDao<M, E>, M extends EntityModelDao<E>, E extends Entity> void checkAuditLog(final ChangeType changeType, @Nullable final CallContext context, final AuditLog auditLog, final UUID entityId, Class<T> sqlDao,
                                                     boolean useHistory, boolean checkContext) {
  Assert.assertEquals(auditLog.getChangeType(), changeType);
  if (checkContext) {
    Assert.assertEquals(auditLog.getUserName(), context.getUserName());
    Assert.assertEquals(auditLog.getComment(), context.getComments());
    //Assert.assertEquals(auditLog.getCreatedDate().comparesTo(callcontext.getCreatedDate()));
    // We can't take userToken oustide of the 'if' because for instance NextBillingDate invoice will not have it.
    Assert.assertEquals(auditLog.getUserToken(), context.getUserToken().toString());
  }
  final M entityModel = extractEntityModelFromEntityWithTargetRecordId(entityId, auditLog.getId(), sqlDao, context, useHistory);
  Assert.assertEquals(entityModel.getId(), entityId);
}

public AuditLogJson(final AuditLog auditLog) {
  this(auditLog.getChangeType().toString(), auditLog.getCreatedDate(), auditLog.getAuditedObjectType(), auditLog.getAuditedEntityId(), auditLog.getUserName(), auditLog.getReasonCode(),
     auditLog.getComment(), auditLog.getUserToken(), null);
}