AuditTest.java example

Explorer
midpoint-master
/*
 * Copyright (c) 2010-2017 Evolveum
 *
 * 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 com.evolveum.midpoint.repo.sql;

import com.evolveum.midpoint.audit.api.AuditEventRecord;
import com.evolveum.midpoint.audit.api.AuditReferenceValue;
import com.evolveum.midpoint.prism.polystring.PolyString;
import com.evolveum.midpoint.repo.sql.data.audit.RAuditEventRecord;
import com.evolveum.midpoint.repo.sql.util.DtoTranslationException;
import com.evolveum.midpoint.repo.sql.util.SimpleTaskAdapter;
import com.evolveum.midpoint.util.exception.SystemException;
import com.evolveum.midpoint.util.logging.Trace;
import com.evolveum.midpoint.util.logging.TraceManager;
import com.evolveum.midpoint.xml.ns._public.common.common_3.RoleType;
import com.evolveum.midpoint.xml.ns._public.common.common_3.UserType;
import org.hibernate.Query;
import org.hibernate.Session;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ContextConfiguration;
import org.testng.annotations.Test;

import java.util.*;

import static org.testng.AssertJUnit.assertEquals;

/**
 * @author mederly
 */
@ContextConfiguration(locations = {"../../../../../ctx-test.xml"})
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS)
public class AuditTest extends BaseSQLRepoTest {

    private static final Trace LOGGER = TraceManager.getTrace(AuditTest.class);

    @Test
    public void test100AuditSimple() throws Exception {
        LOGGER.info("===[ test100AuditSimple ]===");

        // WHEN
		AuditEventRecord record = new AuditEventRecord();
		record.addPropertyValue("prop1", "val1.1");
		record.addPropertyValue("prop1", "val1.2");
		record.addPropertyValue("prop2", "val2");
		record.addPropertyValue("prop3", null);
		AuditReferenceValue refVal1_1 = new AuditReferenceValue("oid1.1", UserType.COMPLEX_TYPE, new PolyString("user1.1"));
		AuditReferenceValue refVal1_2 = new AuditReferenceValue("oid1.2", RoleType.COMPLEX_TYPE, new PolyString("role1.2"));
		AuditReferenceValue refVal2 = new AuditReferenceValue("oid2", null, new PolyString("object2"));
		AuditReferenceValue refVal3 = new AuditReferenceValue();
		record.addReferenceValue("ref1", refVal1_1);
		record.addReferenceValue("ref1", refVal1_2);
		record.addReferenceValue("ref2", refVal2);
		record.addReferenceValue("ref3", refVal3);
		LOGGER.info("Adding audit record {}", record);
		auditService.audit(record, new SimpleTaskAdapter());

		// THEN
		System.out.println("Record written:\n" + record.debugDump());
		System.out.println("Repo ID: " + record.getRepoId());
		AuditEventRecord loaded = getAuditEventRecord(1, 0);
		System.out.println("Record loaded:\n" + loaded.debugDump());
		System.out.println("Repo ID: " + loaded.getRepoId());
		assertEquals("Wrong # of properties", 3, loaded.getProperties().size());
		assertEquals("Wrong prop1 values", new HashSet<>(Arrays.asList("val1.1", "val1.2")), loaded.getPropertyValues("prop1"));
		assertEquals("Wrong prop2 values", new HashSet<>(Collections.singletonList("val2")), loaded.getPropertyValues("prop2"));
		assertEquals("Wrong prop3 values", new HashSet<>(Collections.singletonList(null)), loaded.getPropertyValues("prop3"));
		assertEquals("Wrong # of references", 3, loaded.getReferences().size());
		assertEquals("Wrong ref1 values", new HashSet<>(Arrays.asList(refVal1_1, refVal1_2)), loaded.getReferenceValues("ref1"));
		assertEquals("Wrong ref2 values", new HashSet<>(Collections.singletonList(refVal2)), loaded.getReferenceValues("ref2"));
		assertEquals("Wrong ref3 values", new HashSet<>(Collections.singletonList(refVal3)), loaded.getReferenceValues("ref3"));

	}

    @Test
    public void test110AuditSecond() throws Exception {
        LOGGER.info("===[ test110AuditSecond ]===");

        // WHEN
		AuditEventRecord record = new AuditEventRecord();
		record.addPropertyValue("prop", "val");
		LOGGER.info("Adding audit record {}", record);
		auditService.audit(record, new SimpleTaskAdapter());

		// THEN
		System.out.println("Record written:\n" + record.debugDump());
		System.out.println("Repo ID: " + record.getRepoId());
		AuditEventRecord loaded = getAuditEventRecord(2, 1);
		System.out.println("Record loaded:\n" + loaded.debugDump());
		System.out.println("Repo ID: " + loaded.getRepoId());
		assertEquals("Wrong # of properties", 1, loaded.getProperties().size());
		assertEquals("Wrong prop values", new HashSet<>(Collections.singletonList("val")), loaded.getPropertyValues("prop"));
		assertEquals("Wrong # of references", 0, loaded.getReferences().size());
	}

	private AuditEventRecord getAuditEventRecord(int expectedCount, int index) {
		Session session = getFactory().openSession();
		try {
			session.beginTransaction();

			Query query = session.createQuery("from " + RAuditEventRecord.class.getSimpleName() + " order by id");
			List<RAuditEventRecord> records = query.list();
			assertEquals(expectedCount, records.size());
			AuditEventRecord eventRecord = RAuditEventRecord.fromRepo(records.get(index), prismContext);
			session.getTransaction().commit();
			return eventRecord;
		} catch (DtoTranslationException e) {
			throw new SystemException(e);
		} finally {
			session.close();
		}

	}

}