Skip to content

Commit

Permalink
allow IdentificationTests to override the strategy used for resolving…
Browse files Browse the repository at this point in the history
… the expected form (#555)
  • Loading branch information
drivenflywheel authored Sep 15, 2023
1 parent 082d9ed commit 4fc5d0d
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion src/test/java/emissary/test/core/junit5/IdentificationTest.java
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ public void testIdentificationPlace(String resource) {

try (InputStream doc = new ResourceReader().getResourceAsStream(resource)) {
byte[] data = IOUtils.toByteArray(doc);
String expectedAnswer = resource.replaceAll("^.*/([^/@]+)(@\\d+)?\\.dat$", "$1");
String expectedAnswer = resolveFormForResource(resource);
IBaseDataObject payload = DataObjectFactory.getInstance(data, resource, Form.UNKNOWN);
processPreHook(payload, resource);
place.agentProcessHeavyDuty(payload);
Expand All @@ -71,6 +71,16 @@ public void testIdentificationPlace(String resource) {
}
}

/**
* Resolves the expected form for the test resource.
*
* @param resource complete file path of the test resource
* @return form expected for the resource
*/
protected String resolveFormForResource(String resource) {
return resource.replaceAll("^.*/([^/@]+)(@\\d+)?\\.dat$", "$1");
}

protected void processPreHook(IBaseDataObject payload, String resource) {
// Nothing to do here
}
Expand Down

0 comments on commit 4fc5d0d

Please sign in to comment.