Implement runner checkpoint option

gofractally · Nov 9, 2021 · b9620f3 · b9620f3
1 parent 0cf518f
commit b9620f3
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/contracts/eden/tests/include/nodeos-runner.hpp b/contracts/eden/tests/include/nodeos-runner.hpp
@@ -2,13 +2,39 @@
 
 #include <tester-base.hpp>
 
+#define CATCH_CONFIG_RUNNER
+
+std::string runner_checkpoint;
+
+int main(int argc, char* argv[])
+{
+   Catch::Session session;
+   auto cli =
+       session.cli() | Catch::clara::Opt(runner_checkpoint, "checkpoint")["-c"]["--checkpoint"](
+                           "Stops execution at the given checkpoint label.");
+   session.cli(cli);
+   auto ret = session.applyCommandLine(argc, argv);
+   if (ret)
+      return ret;
+   return session.run();
+}
+
 struct nodeos_runner
 {
    eden_tester tester;
    std::string runner_name;
 
    nodeos_runner(std::string runner_name) : runner_name(runner_name) {}
 
+   void checkpoint(const std::string& checkpoint)
+   {
+      if (runner_checkpoint == checkpoint)
+      {
+         start_nodeos();
+         exit(0);
+      }
+   }
+
    void start_nodeos()
    {
       // tolerance blocks

diff --git a/contracts/eden/tests/run-full-election.cpp b/contracts/eden/tests/run-full-election.cpp
@@ -1,5 +1,3 @@
-#define CATCH_CONFIG_MAIN
-
 #include <nodeos-runner.hpp>
 
 TEST_CASE("Setup Eden chain with full election")
@@ -8,7 +6,9 @@ TEST_CASE("Setup Eden chain with full election")
 
    r.tester.genesis();
    r.tester.run_election(true, 10000, true);
+   r.checkpoint("small_election");
    r.tester.induct_n(100);
+   r.checkpoint("inductions");
    r.tester.run_election(true, 10000, true);
 
    r.start_nodeos();

diff --git a/contracts/eden/tests/run-genesis.cpp b/contracts/eden/tests/run-genesis.cpp
@@ -1,5 +1,3 @@
-#define CATCH_CONFIG_MAIN
-
 #include <nodeos-runner.hpp>
 
 TEST_CASE("Setup Eden chain with basic completed genesis")