AuditingClientTest.java

/*
 * Copyright 2012 JBoss Inc
 *
 * Licensed 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.artificer.test.client;

import org.jboss.downloads.artificer._2013.auditing.AuditEntry;
import org.junit.Assert;
import org.junit.Test;
import org.oasis_open.docs.s_ramp.ns.s_ramp_v1.BaseArtifactType;
import org.artificer.client.ArtificerAtomApiClient;
import org.artificer.client.audit.AuditEntrySummary;
import org.artificer.client.audit.AuditResultSet;

import javax.xml.datatype.DatatypeFactory;
import javax.xml.datatype.XMLGregorianCalendar;
import java.util.Calendar;
import java.util.GregorianCalendar;

/**
 * Unit test for the
 *
 * @author eric.wittmann@redhat.com
 */
public class AuditingClientTest extends AbstractClientTest {

    @Test
    public void testAuditing() throws Exception {
        ArtificerAtomApiClient client = client();
        
        // Get all audit entries by user (later used for a delta)
        AuditResultSet resultSet = client.getAuditTrailForUser(getUsername()); 
        Assert.assertNotNull(resultSet);
        long currentAudit = resultSet.getTotalResults();
        
        DatatypeFactory dtFactory = DatatypeFactory.newInstance();
        BaseArtifactType doc = addXmlDoc();
        // Add a second artifact.
        addXmlDoc();
        String artifactUuid = doc.getUuid();

        // Checking auditing - should be 1 event (artifact:add)
        resultSet = client.getAuditTrailForArtifact(artifactUuid);
        Assert.assertNotNull(resultSet);
        Assert.assertEquals(1, resultSet.getTotalResults());
        AuditEntrySummary summary = resultSet.get(0);
        Assert.assertNotNull(summary);
        Assert.assertEquals(getUsername(), summary.getWho()); 
        Assert.assertEquals("artifact:add", summary.getType());

        // Add a custom entry
        AuditEntry auditEntry = new AuditEntry();
        XMLGregorianCalendar now = dtFactory.newXMLGregorianCalendar((GregorianCalendar)Calendar.getInstance());
        auditEntry.setType("junit:test1");
        auditEntry.setWhen(now);
        auditEntry.setWho(getUsername()); 
        AuditEntry newEntry = client.addAuditEntry(artifactUuid, auditEntry );
        Assert.assertNotNull(newEntry);
        Assert.assertEquals(getUsername(), newEntry.getWho()); 
        Assert.assertEquals("junit:test1", newEntry.getType());

        // Get the audit trail again
        resultSet = client.getAuditTrailForArtifact(artifactUuid);
        Assert.assertNotNull(resultSet);
        Assert.assertEquals(2, resultSet.getTotalResults());
        summary = resultSet.get(0);
        Assert.assertNotNull(summary);
        Assert.assertEquals(getUsername(), summary.getWho()); 
        Assert.assertEquals("junit:test1", summary.getType());
        summary = resultSet.get(1);
        Assert.assertNotNull(summary);
        Assert.assertEquals(getUsername(), summary.getWho()); 
        Assert.assertEquals("artifact:add", summary.getType());

        // Get the full audit entry
        auditEntry = client.getAuditEntry(artifactUuid, summary.getUuid());
        Assert.assertNotNull(newEntry);
        Assert.assertEquals(getUsername(), auditEntry.getWho()); 
        Assert.assertEquals("artifact:add", auditEntry.getType());

        // Get all audit entries by user
        resultSet = client.getAuditTrailForUser(getUsername()); 
        Assert.assertNotNull(resultSet);
        Assert.assertEquals(3, resultSet.getTotalResults() - currentAudit);
    }
}