QRunIO/qqq
2
0
Fork 0
mirror of https://github.com/Kingsrook/qqq.git synced 2025-07-19 05:30:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make postRun methods take a subclass of backendInput/Output, that make it clear they don't have the full record list

Browse Source
This commit is contained in:
darin.kelkhoff
2023-03-09 11:36:21 -06:00
parent e53a982d12
commit 3a172b3fb4
7 changed files with 102 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
qqq-backend-core/src/test/java/com/kingsrook/qqq/backend/core/processes/implementations/etl/streamedwithfrontend/StreamedETLWithFrontendProcessTest.java
View File

@ -53,6 +53,7 @@ import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
@ -332,6 +333,39 @@ public class StreamedETLWithFrontendProcessTest extends BaseTest
/*******************************************************************************
**
*******************************************************************************/
@Test
void testPostRun() throws QException
{
QInstance instance = QContext.getQInstance();
////////////////////////////////////////////////////////
// define the process - an ELT from Shapes to Persons //
////////////////////////////////////////////////////////
QProcessMetaData process = StreamedETLWithFrontendProcess.defineProcessMetaData(
TestUtils.TABLE_NAME_SHAPE,
TestUtils.TABLE_NAME_PERSON,
ExtractViaQueryStep.class,
NoopTransformStep.class,
TestLoadPostRunStep.class);
process.setName("test");
process.setTableName(TestUtils.TABLE_NAME_SHAPE);
instance.addProcess(process);
instance.getTable(TestUtils.TABLE_NAME_PERSON).setBackendName(TestUtils.MEMORY_BACKEND_NAME);
TestUtils.insertDefaultShapes(instance);
/////////////////////
// run the process //
/////////////////////
RunProcessOutput runProcessOutput = runProcess(instance, process, Collections.emptyMap(), new Callback(), RunProcessInput.FrontendStepBehavior.SKIP);
assertEquals(47, runProcessOutput.getValues().get("valueFromPostStep"));
}
/*******************************************************************************
**
*******************************************************************************/
@ -540,6 +574,49 @@ public class StreamedETLWithFrontendProcessTest extends BaseTest
/*******************************************************************************
**
*******************************************************************************/
public static class TestLoadPostRunStep extends AbstractLoadStep
{
/*******************************************************************************
**
*******************************************************************************/
@Override
public void run(RunBackendStepInput runBackendStepInput, RunBackendStepOutput runBackendStepOutput) throws QException
{
///////////////////////////////////
// just pass the records through //
///////////////////////////////////
for(QRecord record : runBackendStepInput.getRecords())
{
runBackendStepOutput.addRecord(record);
}
}
/*******************************************************************************
**
*******************************************************************************/
@Override
public void postRun(BackendStepPostRunInput runBackendStepInput, BackendStepPostRunOutput runBackendStepOutput) throws QException
{
assertThatThrownBy(() -> runBackendStepInput.getRecords())
.isInstanceOf(IllegalStateException.class);
assertThatThrownBy(() -> runBackendStepOutput.getRecords())
.isInstanceOf(IllegalStateException.class);
assertThat(runBackendStepInput.getPreviewRecordList()).isNotEmpty();
assertThat(runBackendStepOutput.getPreviewRecordList()).isNotEmpty();
runBackendStepOutput.addValue("valueFromPostStep", 47);
}
}
/*******************************************************************************
**
*******************************************************************************/