Skip to content

Commit

Permalink
Extend simulation search/count tests
Browse files Browse the repository at this point in the history 
Related to MID-8540.
  • Loading branch information
@mederly
mederly committed Mar 1, 2023
1 parent 2df9803 commit 9601d2f
Showing 1 changed file with 109 additions and 88 deletions.
197 changes: 109 additions & 88 deletions ...po-sqale/src/test/java/com/evolveum/midpoint/repo/sqale/func/SimulationsBaselineTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,9 @@
import static org.testng.Assert.assertTrue;

import java.util.Collection;
import java.util.function.Consumer;

import com.evolveum.midpoint.prism.PrismConstants;
import com.evolveum.midpoint.schema.DeltaConvertor;

import com.evolveum.prism.xml.ns._public.types_3.ObjectDeltaType;
Expand Down Expand Up @@ -172,7 +174,6 @@ public void test100CreateSimulation() throws ObjectAlreadyExistsException, Schem
assertThat(processedObjects).as("processed objects").hasSize(1);

and("can be parsed");
// TODO this should work, shouldn't it?
ObjectType objectBefore = processedObjects.get(0).getBefore();
assertThat(objectBefore).as("'object before' from result").isEqualTo(systemConfiguration);

Expand Down Expand Up @@ -316,96 +317,116 @@ public void test200DeleteProcessedObjects() throws ObjectNotFoundException, Sche
}

/** Searching for POs. */
@Test(enabled = false)
@Test
public void test300SearchForProcessedObjects() throws SchemaException {
OperationResult result = createOperationResult();

when("getting all POs from 1st result");
var allFromFirst = getProcessedObjects(firstResultOid, result);

then("search result is OK");
assertThat(allFromFirst).as("all objects from first result").hasSize(1);
assertThat(allFromFirst.iterator().next().getState())
.as("state")
.isEqualTo(ObjectProcessingStateType.ADDED);

when("getting POs from 2nd result by name");
Collection<SimulationResultProcessedObjectType> byName =
repositoryService.searchContainers(
SimulationResultProcessedObjectType.class,
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_NAME)
.eqPoly(USER2_NAME)
.build(),
null,
result);

then("search result is OK");
assertThat(byName).as("POs by name").hasSize(1);
assertThat(byName.iterator().next().getName().getOrig())
.as("name")
.isEqualTo(USER2_NAME);

when("getting POs from 2nd result by type");
Collection<SimulationResultProcessedObjectType> byType =
repositoryService.searchContainers(
SimulationResultProcessedObjectType.class,
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_TYPE)
.eq(ShadowType.COMPLEX_TYPE)
.build(),
null,
result);

then("search result is OK");
assertThat(byType).as("POs by type").hasSize(1);
assertThat(byType.iterator().next().getType())
.as("type")
.isEqualTo(ShadowType.COMPLEX_TYPE);

when("getting POs from 2nd result by OID");
Collection<SimulationResultProcessedObjectType> byOid =
repositoryService.searchContainers(
SimulationResultProcessedObjectType.class,
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_OID)
.eq(USER1_OID)
.build(),
null,
result);

then("search result is OK");
assertThat(byOid).as("POs by OID").hasSize(1);
assertThat(byOid.iterator().next().getOid())
.as("OID")
.isEqualTo(USER1_OID);

when("getting POs from 2nd result by state");
Collection<SimulationResultProcessedObjectType> byState =
repositoryService.searchContainers(
SimulationResultProcessedObjectType.class,
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_STATE)
.eq(ObjectProcessingStateType.ADDED)
.build(),
null,
result);

then("search result is OK");
assertThat(byState).as("objects from 2nd result by state").hasSize(1);
assertThat(byState.iterator().next().getState())
.as("state")
.isEqualTo(ObjectProcessingStateType.ADDED);

// TODO other kinds of search
when("checking 'search all'");
ObjectQuery allFrom1st = PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(firstResultOid)
.build();
checkCountAndSearch("all from 1st", allFrom1st, 1, result,
po -> assertThat(po.getState())
.as("state")
.isEqualTo(ObjectProcessingStateType.ADDED));

when("checking search by name");
ObjectQuery byName =
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_NAME)
.eq(USER2_NAME)
.build();
checkCountAndSearch("by name", byName, 1, result,
po -> assertThat(po.getName().getOrig())
.as("name")
.isEqualTo(USER2_NAME));

when("checking search by OID");
ObjectQuery byOid =
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_OID)
.eq(USER1_OID)
.build();
checkCountAndSearch("by OID", byOid, 1, result,
po -> assertThat(po.getOid())
.as("OID")
.isEqualTo(USER1_OID));

when("checking search by record ID");
var firstId = getProcessedObjects(secondResultOid, result)
.iterator().next().getId();
ObjectQuery byId =
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(PrismConstants.T_ID)
.eq(firstId)
.build();
checkCountAndSearch("by ID", byId, 1, result,
po -> assertThat(po.getId())
.as("ID")
.isEqualTo(firstId));

when("checking search by transaction ID");
ObjectQuery byTxId =
PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
.ownerId(secondResultOid)
.and()
.item(SimulationResultProcessedObjectType.F_TRANSACTION_ID)
.eq("#3")
.build();
checkCountAndSearch("by tx ID", byTxId, 1, result,
po -> assertThat(po.getTransactionId())
.as("tx ID")
.isEqualTo("#3"));

// when("checking search by state");
// ObjectQuery byState =
// PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
// .ownerId(secondResultOid)
// .and()
// .item(SimulationResultProcessedObjectType.F_STATE)
// .eq(ObjectProcessingStateType.ADDED)
// .build();
// checkCountAndSearch("by OID", byState, 1, result,
// po -> assertThat(po.getState())
// .as("state")
// .isEqualTo(ObjectProcessingStateType.ADDED));
//
// when("checking search by type");
// ObjectQuery byType =
// PrismContext.get().queryFor(SimulationResultProcessedObjectType.class)
// .ownerId(secondResultOid)
// .and()
// .item(SimulationResultProcessedObjectType.F_TYPE)
// .eq(ShadowType.COMPLEX_TYPE)
// .build();
// checkCountAndSearch("by type", byType, 1, result,
// po -> assertThat(po.getType())
// .as("type")
// .isEqualTo(ShadowType.COMPLEX_TYPE));
}

@SuppressWarnings("SameParameterValue")
private void checkCountAndSearch(
String label, ObjectQuery query, int expectedCount, OperationResult result,
Consumer<SimulationResultProcessedObjectType> asserter) throws SchemaException {
when("counting POs " + label);
int count = repositoryService.countContainers(SimulationResultProcessedObjectType.class, query, null, result);

then("result is OK");
assertThat(count).as("count " + label).isEqualTo(expectedCount);

when("searching for POs " + label);
var objects = repositoryService.searchContainers(SimulationResultProcessedObjectType.class, query, null, result);

then("results are OK");
assertThat(objects).as("objects " + label).hasSize(expectedCount);
objects.forEach(asserter);
}

private void assertProcessedObjects(String oid, int expectedObjects, OperationResult result)
Expand Down

0 comments on commit 9601d2f

Please sign in to comment.