feat: Implement state handler calls

Author: TimothyJones

src/main/java/io/contract_testing/contractcase/client/ConfigHandle.java

@@ -2,7 +2,8 @@
 
 import static io.contract_testing.contractcase.client.ConnectorOutgoingMapper.CONTRACT_CASE_TRIGGER_AND_TEST;
 
-import io.contract_testing.contractcase.ContractCaseCoreError;
+import io.contract_testing.contractcase.ContractCaseConfigurationError;
+import io.contract_testing.contractcase.case_boundary.BoundaryStateHandler;
 import io.contract_testing.contractcase.case_boundary.ContractCaseBoundaryConfig;
 import io.contract_testing.contractcase.case_boundary.ITriggerFunction;
 import org.jetbrains.annotations.NotNull;
@@ -20,12 +21,29 @@ public void setBoundaryConfig(ContractCaseBoundaryConfig boundaryConfig) {
     this.boundaryConfig = boundaryConfig;
   }
 
+  BoundaryStateHandler getStateHandler(String stateName) {
+    if (boundaryConfig.getStateHandlers() == null) {
+      throw new ContractCaseConfigurationError(
+          "No state handlers provided, you must provide a state handler that can run '" + stateName
+              + "'");
+    }
+    var stateHandler = boundaryConfig.getStateHandlers().get(stateName);
+
+    if (stateHandler == null) {
+      throw new ContractCaseConfigurationError(
+          "The state handler named '" + stateName
+              + "' was not provided in the configuration");
+    }
+    return stateHandler;
+  }
+
   @NotNull
   ITriggerFunction getTriggerFunction(String handle) {
     ITriggerFunction trigger = getTriggerInternal(handle);
     if (trigger == null) {
-      throw new ContractCaseCoreError(
-          "Unable to trigger the function with handle '" + handle + "', as it is null");
+      throw new ContractCaseConfigurationError(
+          "The trigger function named '" + handle
+              + "' wasn't provided in the configuration, but is required by this test run");
     }
     return trigger;
   }
@@ -38,10 +56,12 @@ private ITriggerFunction getTriggerInternal(String handle) {
 
     var triggerMap = boundaryConfig.getTriggerAndTests();
     if (triggerMap == null) {
-      throw new ContractCaseCoreError(
-          "Unable to trigger the function with handle '" + handle
-              + "', as the entire trigger map is null");
+      throw new ContractCaseConfigurationError(
+          "No trigger functions were provided, unable to run the trigger function '" + handle
+              + "'");
     }
     return triggerMap.get(handle);
   }
+
+
 }

src/main/java/io/contract_testing/contractcase/client/ContractResponseStreamObserver.java

@@ -9,13 +9,14 @@
 import com.google.protobuf.AbstractMessage;
 import com.google.protobuf.GeneratedMessageV3.Builder;
 import io.contract_testing.contractcase.ContractCaseCoreError;
+import io.contract_testing.contractcase.case_boundary.BoundaryResult;
+import io.contract_testing.contractcase.case_boundary.BoundaryStateHandler;
 import io.contract_testing.contractcase.case_boundary.ILogPrinter;
 import io.contract_testing.contractcase.case_boundary.IResultPrinter;
 import io.contract_testing.contractcase.case_boundary.IRunTestCallback;
-import io.contract_testing.contractcase.grpc.ContractCaseStream;
 import io.contract_testing.contractcase.grpc.ContractCaseStream.ContractResponse;
 import io.contract_testing.contractcase.grpc.ContractCaseStream.ResultResponse;
-import io.contract_testing.contractcase.grpc.ContractCaseStream.ResultSuccess;
+import io.contract_testing.contractcase.grpc.ContractCaseStream.StateHandlerHandle.Stage;
 import io.grpc.Status;
 import io.grpc.stub.StreamObserver;
 import org.jetbrains.annotations.NotNull;
@@ -50,13 +51,20 @@ public void onNext(final ContractResponse note) {
     switch (note.getKindCase()) {
       case RUN_STATE_HANDLER -> {
         final var stateHandlerRunRequest = note.getRunStateHandler();
-        // TODO Implement this properly
+        var stateName = stateHandlerRunRequest.getStateHandlerHandle()
+            .getHandle()
+            .getValue();
+        var handle = configHandle.getStateHandler(
+            stateName);
+
         rpcConnector.sendResponse(
             ResultResponse.newBuilder()
-                .setResult(ContractCaseStream.BoundaryResult.newBuilder()
-                    .setSuccess(ResultSuccess.newBuilder().build())
-                    .build()
-                ).build(),
+                .setResult(mapResult(runStateHandler(
+                    stateHandlerRunRequest.getStateHandlerHandle()
+                        .getStage(),
+                    stateName,
+                    handle
+                ))).build(),
             requestId
         );
       }
@@ -135,7 +143,6 @@ public void onNext(final ContractResponse note) {
       }
       case START_TEST_EVENT -> {
         var startTestEvent = note.getStartTestEvent();
-
         rpcConnector.sendResponse(
             mapResultResponse(
                 runTestCallback.runTest(
@@ -185,4 +192,17 @@ public void onError(final Throwable t) {
   @Override
   public void onCompleted() {
   }
+
+  @NotNull
+  private static BoundaryResult runStateHandler(Stage stage,
+      String stateName,
+      BoundaryStateHandler handle) {
+    return switch (stage) {
+      case STAGE_SETUP_UNSPECIFIED -> handle.setup();
+      case STAGE_TEARDOWN -> handle.teardown();
+      case UNRECOGNIZED -> throw new ContractCaseCoreError(
+          "Unrecognised state handler stage while trying to run '" + stateName +
+              "'");
+    };
+  }
 }