1 files changed, 31 insertions, 10 deletions

diff --git a/src/ats-tests/ats-testing.h b/src/ats-tests/ats-testing.h
index 028789d69..a2937d51d 100644
--- a/src/ats-tests/ats-testing.h
+++ b/src/ats-tests/ats-testing.h
@@ -385,15 +385,43 @@ struct GNUNET_ATS_TEST_Topology
   void *done_cb_cls;
 };
 
-struct Experiment;
+enum OperationType
+{
+  START_SEND,
+  STOP_SEND,
+  SET_RATE,
+  SET_PREFERENCE
+};
 
 struct Episode;
 
+struct Experiment;
+
 typedef void (*GNUNET_ATS_TESTING_EpisodeDoneCallback) (
     struct Episode *e);
 
-typedef void (*GNUNET_ATS_TESTING_ExperimentDoneCallback) (
-    struct Experiment *e, int success);
+typedef void (*GNUNET_ATS_TESTING_ExperimentDoneCallback) (struct Experiment *e,
+    struct GNUNET_TIME_Relative duration,int success);
+
+struct Operation
+{
+  struct Operation *next;
+  struct Operation *prev;
+  long long unsigned int src_id;
+  long long unsigned int dest_id;
+  long long unsigned int value;
+  enum OperationType type;
+};
+
+struct Episode
+{
+  int id;
+  struct Episode *next;
+  struct GNUNET_TIME_Relative duration;
+
+  struct Operation *head;
+  struct Operation *tail;
+};
 
 
 struct Experiment
@@ -416,13 +444,6 @@ struct Experiment
   GNUNET_ATS_TESTING_ExperimentDoneCallback e_done_cb;
 };
 
-struct Episode
-{
-  int id;
-  struct Episode *next;
-  struct GNUNET_TIME_Relative duration;
-};
-
 void
 GNUNET_ATS_TEST_experimentation_run (struct Experiment *e,
     GNUNET_ATS_TESTING_EpisodeDoneCallback ep_done_cb,