From 4c4c708d0c496632752bb31ea9d500f068c618e8 Mon Sep 17 00:00:00 2001
From: Jared O'Connell <joconnel@redhat.com>
Date: Thu, 12 Oct 2023 16:10:48 -0400
Subject: [PATCH] Remove more ATP v2 parts

---
 arcaflow_plugin_wait/arcaflow_plugin_wait.py |  1 -
 tests/test_arcaflow_plugin_wait.py           | 14 +-------------
 2 files changed, 1 insertion(+), 14 deletions(-)

diff --git a/arcaflow_plugin_wait/arcaflow_plugin_wait.py b/arcaflow_plugin_wait/arcaflow_plugin_wait.py
index f3cf2d9..127e49f 100644
--- a/arcaflow_plugin_wait/arcaflow_plugin_wait.py
+++ b/arcaflow_plugin_wait/arcaflow_plugin_wait.py
@@ -49,7 +49,6 @@ class ErrorOutput:
     },
 )
 def wait(
-    self,
     params: InputParams,
 ) -> typing.Tuple[str, typing.Union[SuccessOutput, ErrorOutput]]:
     """
diff --git a/tests/test_arcaflow_plugin_wait.py b/tests/test_arcaflow_plugin_wait.py
index ee8f3ea..00ce3ae 100644
--- a/tests/test_arcaflow_plugin_wait.py
+++ b/tests/test_arcaflow_plugin_wait.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 import unittest
 
-from arcaflow_plugin_sdk import plugin, predefined_schemas
+from arcaflow_plugin_sdk import plugin
 
 import time
 import arcaflow_plugin_wait
@@ -50,18 +50,6 @@ def test_functional(self):
             "Waited {:0.2f} seconds after being scheduled to wait for {}"
             " seconds.".format(output_data.actual_wait_seconds, WAIT_TIME)
         )
-        # Test cancellation
-        input_params = arcaflow_plugin_wait.InputParams(
-            seconds=SKIPPED_WAIT_TIME
-        )
-        wait_step = arcaflow_plugin_wait.WaitStep()
-        wait_step.cancel_step(wait_step, predefined_schemas.cancelInput())
-        start_time = time.time()
-        output_id, output_data = wait_step.wait(input_params)
-        run_duration = time.time() - start_time
-        # It starts in a cancelled state, so it should be plenty less than
-        # the full wait time.
-        self.assertLess(run_duration, SKIPPED_WAIT_TIME / 2.0)
 
 
 if __name__ == '__main__':