diff --git a/tests/unit_utilities.cpp b/tests/unit_utilities.cpp index e7e9ab4..f1ba9db 100644 --- a/tests/unit_utilities.cpp +++ b/tests/unit_utilities.cpp @@ -90,7 +90,6 @@ TEST_CASE("dag_runner", "[utilities_dag_runner]") { } } - /* SECTION("Recovery from Error") { auto cleanup = []() { // Cleanup @@ -128,9 +127,9 @@ TEST_CASE("dag_runner", "[utilities_dag_runner]") { // Get the DAG Run Attempts auto record = logger.getDAGRun(runID); - REQUIRE(record.taskAttempts["A"].size() == 1); // A ran fine - REQUIRE(record.taskAttempts["B"].size() == 2); // B errored and had to be retried - REQUIRE(record.taskAttempts["C"].size() == 1); // C wasn't run because B errored + REQUIRE(record.taskAttempts["A_0"].size() == 1); // A ran fine + REQUIRE(record.taskAttempts["B_0"].size() == 2); // B errored and had to be retried + REQUIRE(record.taskAttempts["C_0"].size() == 1); // C wasn't run because B errored cleanup(); } @@ -175,10 +174,9 @@ TEST_CASE("dag_runner", "[utilities_dag_runner]") { } // Ensure that children were updated properly - REQUIRE(record.tasks["A"].children == std::unordered_set{"B_0", "B_1", "C"}); + REQUIRE(record.tasks["A_0"].children == std::unordered_set{"B_0", "B_1", "C"}); REQUIRE(record.tasks["B_0"].children == std::unordered_set{"C"}); REQUIRE(record.tasks["B_1"].children == std::unordered_set{"C"}); - REQUIRE(record.tasks["C"].children.empty()); + REQUIRE(record.tasks["C_0"].children.empty()); } - */ }