exchange

testing_api_cmd_batch.c (5717B)

---

 /*
   This file is part of TALER
   Copyright (C) 2014-2021 Taler Systems SA
 
   TALER is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as
   published by the Free Software Foundation; either version 3, or
   (at your option) any later version.
 
   TALER is distributed in the hope that it will be useful, but
   WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
 
   You should have received a copy of the GNU General Public
   License along with TALER; see the file COPYING.  If not, see
   <http://www.gnu.org/licenses/>
 */
 /**
  * @file testing/testing_api_cmd_batch.c
  * @brief Implement batch-execution of CMDs.
  * @author Marcello Stanisci
  */
 #include "taler/taler_json_lib.h"
 #include <gnunet/gnunet_curl_lib.h>
 #include "taler/taler_testing_lib.h"
 
 
 /**
  * State for a "batch" CMD.
  */
 struct BatchState
 {
   /**
    * CMDs batch.
    */
   struct TALER_TESTING_Command *batch;
 
   /**
    * My command (the batch command).
    */
   const struct TALER_TESTING_Command *cmd;
 
   /**
    * Internal command pointer.
    */
   unsigned int batch_ip;
 };
 
 
 /**
  * Run the command.
  *
  * @param cls closure.
  * @param cmd the command being executed.
  * @param is the interpreter state.
  */
 static void
 batch_run (void *cls,
            const struct TALER_TESTING_Command *cmd,
            struct TALER_TESTING_Interpreter *is)
 {
   struct BatchState *bs = cls;
   struct TALER_TESTING_Command *bcmd = &bs->batch[bs->batch_ip];
 
   bs->cmd = cmd;
   if (NULL != bcmd->label)
     TALER_LOG_INFO ("Running batched command: %s\n",
                     bcmd->label);
 
   /* hit end command, leap to next top-level command.  */
   if (NULL == bcmd->label)
   {
     TALER_LOG_INFO ("Exiting from batch: %s\n",
                     cmd->label);
     TALER_TESTING_interpreter_next (is);
     return;
   }
   bcmd->start_time
     = bcmd->last_req_time
       = GNUNET_TIME_absolute_get ();
   bcmd->num_tries++;
   TALER_TESTING_update_variables_ (is,
                                    bcmd);
   bcmd->run (bcmd->cls,
              bcmd,
              is);
 }
 
 
 /**
  * Cleanup the state from a "reserve status" CMD, and possibly
  * cancel a pending operation thereof.
  *
  * @param cls closure.
  * @param cmd the command which is being cleaned up.
  */
 static void
 batch_cleanup (void *cls,
                const struct TALER_TESTING_Command *cmd)
 {
   struct BatchState *bs = cls;
 
   (void) cmd;
   for (unsigned int i = 0;
        NULL != bs->batch[i].label;
        i++)
     if (NULL != bs->batch[i].cleanup)
       bs->batch[i].cleanup (bs->batch[i].cls,
                             &bs->batch[i]);
   GNUNET_free (bs->batch);
   GNUNET_free (bs);
 }
 
 
 /**
  * Offer internal data from a "batch" CMD, to other commands.
  *
  * @param cls closure.
  * @param[out] ret result.
  * @param trait name of the trait.
  * @param index index number of the object to offer.
  * @return #GNUNET_OK on success.
  */
 static enum GNUNET_GenericReturnValue
 batch_traits (void *cls,
               const void **ret,
               const char *trait,
               unsigned int index)
 {
   struct BatchState *bs = cls;
   struct TALER_TESTING_Trait traits[] = {
     TALER_TESTING_make_trait_batch_cmds (bs->batch),
     TALER_TESTING_trait_end ()
   };
 
   /* Always return current command.  */
   return TALER_TESTING_get_trait (traits,
                                   ret,
                                   trait,
                                   index);
 }
 
 
 struct TALER_TESTING_Command
 TALER_TESTING_cmd_batch (const char *label,
                          struct TALER_TESTING_Command *batch)
 {
   struct BatchState *bs;
   unsigned int i;
 
   bs = GNUNET_new (struct BatchState);
 
   /* Get number of commands.  */
   for (i = 0; NULL != batch[i].label; i++)
     /* noop */
     ;
 
   bs->batch = GNUNET_new_array (i + 1,
                                 struct TALER_TESTING_Command);
   GNUNET_memcpy (bs->batch,
                  batch,
                  sizeof (struct TALER_TESTING_Command) * i);
   {
     struct TALER_TESTING_Command cmd = {
       .cls = bs,
       .label = label,
       .run = &batch_run,
       .cleanup = &batch_cleanup,
       .traits = &batch_traits
     };
 
     return cmd;
   }
 }
 
 
 bool
 TALER_TESTING_cmd_batch_next (struct TALER_TESTING_Interpreter *is,
                               void *cls)
 {
   struct BatchState *bs = cls;
   struct TALER_TESTING_Command *bcmd = &bs->batch[bs->batch_ip];
 
   if (NULL == bcmd->label)
   {
     /* This batch is done */
     return true;
   }
   if (TALER_TESTING_cmd_is_batch (bcmd))
   {
     if (TALER_TESTING_cmd_batch_next (is,
                                       bcmd->cls))
     {
       /* sub-batch is done */
       bcmd->finish_time = GNUNET_TIME_absolute_get ();
       bs->batch_ip++;
       return false;
     }
   }
   /* Simple command is done */
   bcmd->finish_time = GNUNET_TIME_absolute_get ();
   bs->batch_ip++;
   return false;
 }
 
 
 bool
 TALER_TESTING_cmd_is_batch (const struct TALER_TESTING_Command *cmd)
 {
   return cmd->run == &batch_run;
 }
 
 
 struct TALER_TESTING_Command *
 TALER_TESTING_cmd_batch_get_current (const struct TALER_TESTING_Command *cmd)
 {
   struct BatchState *bs = cmd->cls;
 
   GNUNET_assert (cmd->run == &batch_run);
   return &bs->batch[bs->batch_ip];
 }
 
 
 void
 TALER_TESTING_cmd_batch_set_current (const struct TALER_TESTING_Command *cmd,
                                      unsigned int new_ip)
 {
   struct BatchState *bs = cmd->cls;
 
   /* sanity checks */
   GNUNET_assert (cmd->run == &batch_run);
   for (unsigned int i = 0; i < new_ip; i++)
     GNUNET_assert (NULL != bs->batch[i].label);
   /* actual logic */
   bs->batch_ip = new_ip;
 }