add command to test reservations

2 files changed, 109 insertions, 3 deletions

diff --git a/src/ats/test_ats_lib.c b/src/ats/test_ats_lib.c
index 74cff1e61..b55881b6e 100644
--- a/src/ats/test_ats_lib.c
+++ b/src/ats/test_ats_lib.c
@@ -404,7 +404,8 @@ make_address (uint32_t pid,
 /**
  * Our dummy sessions.
  */
-struct Session {
+struct Session
+{
   /**
    * Field to avoid `0 == sizeof(struct Session)`.
    */
@@ -522,6 +523,59 @@ info_cb (void *cls,
 
 
 /**
+ * Function called with reservation result.
+ *
+ * @param cls closure with the reservation command (`struct Command`)
+ * @param peer identifies the peer
+ * @param amount set to the amount that was actually reserved or unreserved;
+ *               either the full requested amount or zero (no partial reservations)
+ * @param res_delay if the reservation could not be satisfied (amount was 0), how
+ *        long should the client wait until re-trying?
+ */
+static void
+reservation_cb (void *cls,
+                const struct GNUNET_PeerIdentity *peer,
+                int32_t amount,
+                struct GNUNET_TIME_Relative res_delay)
+{
+  struct Command *cmd = cls;
+  struct GNUNET_PeerIdentity pid;
+
+  make_peer (cmd->details.reserve_bandwidth.pid,
+             &pid);
+  GNUNET_assert (0 == memcmp (peer,
+                              &pid,
+                              sizeof (struct GNUNET_PeerIdentity)));
+  switch (cmd->details.reserve_bandwidth.expected_result)
+  {
+  case GNUNET_OK:
+    if (amount != cmd->details.reserve_bandwidth.amount)
+    {
+      GNUNET_break (0);
+      GNUNET_SCHEDULER_shutdown ();
+      return;
+    }
+    break;
+  case GNUNET_NO:
+    GNUNET_break ( (0 != amount) ||
+                   (0 != res_delay.rel_value_us) );
+    break;
+  case GNUNET_SYSERR:
+    if ( (amount != cmd->details.reserve_bandwidth.amount) ||
+         (0 == res_delay.rel_value_us) )
+    {
+      GNUNET_break (0);
+      GNUNET_SCHEDULER_shutdown ();
+      return;
+    }
+    break;
+  }
+  off++;
+  run_interpreter ();
+}
+
+
+/**
  * Main interpreter loop. Runs the steps of the test.
  *
  * @param cls NULL
@@ -779,6 +833,20 @@ interpreter (void *cls,
                                                    cmd);
         return;
       }
+    case CMD_RESERVE_BANDWIDTH:
+      {
+        struct GNUNET_PeerIdentity pid;
+
+        make_peer (cmd->details.reserve_bandwidth.pid,
+                   &pid);
+        cmd->details.reserve_bandwidth.rc
+          = GNUNET_ATS_reserve_bandwidth (perf_ats,
+                                          &pid,
+                                          cmd->details.reserve_bandwidth.amount,
+                                          &reservation_cb,
+                                          cmd);
+        return;
+      }
     } /* end switch */
   } /* end while(1) */
 }
diff --git a/src/ats/test_ats_lib.h b/src/ats/test_ats_lib.h
index 590aac750..099fa9b45 100644
--- a/src/ats/test_ats_lib.h
+++ b/src/ats/test_ats_lib.h
@@ -111,9 +111,13 @@ enum CommandCode
    * Obtain list of all addresses, testing
    * #GNUNET_ATS_performance_list_addresses().
    */
-  CMD_LIST_ADDRESSES
+  CMD_LIST_ADDRESSES,
 
-  /* TODO: reserve bandwidth */
+  /**
+   * Reserve bandwidth, testing
+   * #GNUNET_ATS_reserve_bandwidth().
+   */
+  CMD_RESERVE_BANDWIDTH
 
 };
 
@@ -399,6 +403,38 @@ struct CommandListAddresses
 
 
 /**
+ * Details for the #CMD_RESERVE_BANDWIDTH command.
+ */
+struct CommandReserveBandwidth
+{
+  /**
+   * For which peer do we reserve bandwidth?
+   */
+  unsigned int pid;
+
+  /**
+   * How much should we try to reserve?
+   */
+  int32_t amount;
+
+  /**
+   * Should we expect this to work or fail?
+   * #GNUNET_YES: must work
+   * #GNUNET_NO: may work or fail
+   * #GNUNET_SYSERR: must fail
+   */
+  int expected_result;
+
+  /**
+   * Location where we store the return value from
+   * #GNUNET_ATS_reserve_bandwidth().
+   */
+  struct GNUNET_ATS_ReservationContext *rc;
+
+};
+
+
+/**
  * A command for the test case interpreter.
  */
 struct Command
@@ -444,6 +480,8 @@ struct Command
 
     struct CommandListAddresses list_addresses;
 
+    struct CommandReserveBandwidth reserve_bandwidth;
+
   } details;
 
 };