1 files changed, 44 insertions, 0 deletions
diff --git a/src/testing/testing_api_trait_uuid.c b/src/testing/testing_api_trait_uuid.c
index 6fc44973..1221be0a 100644
--- a/src/testing/testing_api_trait_uuid.c
+++ b/src/testing/testing_api_trait_uuid.c
@@ -28,6 +28,7 @@
 
 
 #define TALER_TESTING_TRAIT_UUID "uuid"
+#define TALER_TESTING_TRAIT_CLAIM_TOKEN "claim_token"
 
 
 /**
@@ -71,3 +72,46 @@ TALER_TESTING_make_trait_uuid (unsigned int index,
   };
   return ret;
 }
+
+
+/**
+ * Obtain a claim token from @a cmd.
+ *
+ * @param cmd command to extract the token from.
+ * @param index which amount to pick if @a cmd has multiple
+ *        on offer
+ * @param[out] ct where to write the token.
+ * @return #GNUNET_OK on success.
+ */
+int
+TALER_TESTING_get_trait_claim_token (const struct TALER_TESTING_Command *cmd,
+                                     unsigned int index,
+                                     const struct TALER_ClaimTokenP **ct)
+{
+  return cmd->traits (cmd->cls,
+                      (const void **) ct,
+                      TALER_TESTING_TRAIT_CLAIM_TOKEN,
+                      index);
+}
+
+
+/**
+ * Offer a claim token in a trait.
+ *
+ * @param index which token to offer, in case there are
+ *        multiple available.
+ * @param ct the token to offer.
+ *
+ * @return the trait.
+ */
+struct TALER_TESTING_Trait
+TALER_TESTING_make_trait_claim_token (unsigned int index,
+                                      const struct TALER_ClaimTokenP *ct)
+{
+  struct TALER_TESTING_Trait ret = {
+    .index = index,
+    .trait_name = TALER_TESTING_TRAIT_CLAIM_TOKEN,
+    .ptr = (const void *) ct
+  };
+  return ret;
+}