From 650c656e674f0890d4a13d19c8ecba480145d681 Mon Sep 17 00:00:00 2001
From: Tyler Neely <t@jujit.su>
Date: Thu, 4 Aug 2022 12:05:25 +0000
Subject: [PATCH] Add Simulator::RegisterNew helper method

---
 src/io/simulator/simulator.hpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/io/simulator/simulator.hpp b/src/io/simulator/simulator.hpp
index 8e3ad85c0..6bad9239e 100644
--- a/src/io/simulator/simulator.hpp
+++ b/src/io/simulator/simulator.hpp
@@ -23,6 +23,7 @@ namespace memgraph::io::simulator {
 class Simulator {
   std::mt19937 rng_;
   std::shared_ptr<SimulatorHandle> simulator_handle_;
+  uint16_t auto_port_ = 0;
 
  public:
   explicit Simulator(SimulatorConfig config)
@@ -30,6 +31,11 @@ class Simulator {
 
   void ShutDown() { simulator_handle_->ShutDown(); }
 
+  Io<SimulatorTransport> RegisterNew() {
+    Address address = Address::TestAddress(auto_port_++);
+    return Register(address);
+  }
+
   Io<SimulatorTransport> Register(Address address) {
     std::uniform_int_distribution<uint64_t> seed_distrib;
     uint64_t seed = seed_distrib(rng_);