1 files changed, 38 insertions, 57 deletions

diff --git a/src/test/test_circuitstats.c b/src/test/test_circuitstats.c
index e15dec5a01..00ca1b544c 100644
--- a/src/test/test_circuitstats.c
+++ b/src/test/test_circuitstats.c
@@ -17,18 +17,13 @@
 #include "core/or/circuituse.h"
 #include "core/or/channel.h"
 
-#include "core/or/cpath_build_state_st.h"
 #include "core/or/crypt_path_st.h"
 #include "core/or/extend_info_st.h"
 #include "core/or/origin_circuit_st.h"
 
-void test_circuitstats_timeout(void *arg);
-void test_circuitstats_hoplen(void *arg);
-origin_circuit_t *subtest_fourhop_circuit(struct timeval, int);
-origin_circuit_t *add_opened_threehop(void);
-origin_circuit_t *build_unopened_fourhop(struct timeval);
-
-int cpath_append_hop(crypt_path_t **head_ptr, extend_info_t *choice);
+static origin_circuit_t *add_opened_threehop(void);
+static origin_circuit_t *build_unopened_fourhop(struct timeval);
+static origin_circuit_t *subtest_fourhop_circuit(struct timeval, int);
 
 static int marked_for_close;
 /* Mock function because we are not trying to test the close circuit that does
@@ -45,85 +40,71 @@ mock_circuit_mark_for_close(circuit_t *circ, int reason, int line,
   return;
 }
 
-origin_circuit_t *
+static origin_circuit_t *
 add_opened_threehop(void)
 {
-  origin_circuit_t *or_circ = origin_circuit_new();
+  struct timeval circ_start_time;
+  memset(&circ_start_time, 0, sizeof(circ_start_time));
   extend_info_t fakehop;
   memset(&fakehop, 0, sizeof(fakehop));
-
-  TO_CIRCUIT(or_circ)->purpose = CIRCUIT_PURPOSE_C_GENERAL;
-
-  or_circ->build_state = tor_malloc_zero(sizeof(cpath_build_state_t));
-  or_circ->build_state->desired_path_len = DEFAULT_ROUTE_LEN;
-
-  cpath_append_hop(&or_circ->cpath, &fakehop);
-  cpath_append_hop(&or_circ->cpath, &fakehop);
-  cpath_append_hop(&or_circ->cpath, &fakehop);
-
-  or_circ->has_opened = 1;
-  TO_CIRCUIT(or_circ)->state = CIRCUIT_STATE_OPEN;
-  TO_CIRCUIT(or_circ)->purpose = CIRCUIT_PURPOSE_C_GENERAL;
-
-  return or_circ;
+  extend_info_t *fakehop_list[DEFAULT_ROUTE_LEN] = {&fakehop,
+                                                    &fakehop,
+                                                    &fakehop};
+
+  return new_test_origin_circuit(true,
+                                 circ_start_time,
+                                 DEFAULT_ROUTE_LEN,
+                                 fakehop_list);
 }
 
-origin_circuit_t *
+static origin_circuit_t *
 build_unopened_fourhop(struct timeval circ_start_time)
 {
-  origin_circuit_t *or_circ = origin_circuit_new();
-  extend_info_t *fakehop = tor_malloc_zero(sizeof(extend_info_t));
-  memset(fakehop, 0, sizeof(extend_info_t));
-
-  TO_CIRCUIT(or_circ)->purpose = CIRCUIT_PURPOSE_C_GENERAL;
-  TO_CIRCUIT(or_circ)->timestamp_began = circ_start_time;
-  TO_CIRCUIT(or_circ)->timestamp_created = circ_start_time;
-
-  or_circ->build_state = tor_malloc_zero(sizeof(cpath_build_state_t));
-  or_circ->build_state->desired_path_len = 4;
-
-  cpath_append_hop(&or_circ->cpath, fakehop);
-  cpath_append_hop(&or_circ->cpath, fakehop);
-  cpath_append_hop(&or_circ->cpath, fakehop);
-  cpath_append_hop(&or_circ->cpath, fakehop);
-
-  tor_free(fakehop);
-
-  return or_circ;
+  extend_info_t fakehop;
+  memset(&fakehop, 0, sizeof(fakehop));
+  extend_info_t *fakehop_list[4] = {&fakehop,
+                                    &fakehop,
+                                    &fakehop,
+                                    &fakehop};
+
+  return new_test_origin_circuit(false,
+                                 circ_start_time,
+                                 4,
+                                 fakehop_list);
 }
 
-origin_circuit_t *
+static origin_circuit_t *
 subtest_fourhop_circuit(struct timeval circ_start_time, int should_timeout)
 {
-  origin_circuit_t *or_circ = build_unopened_fourhop(circ_start_time);
+  origin_circuit_t *origin_circ = build_unopened_fourhop(circ_start_time);
 
   // Now make them open one at a time and call
   // circuit_build_times_handle_completed_hop();
-  or_circ->cpath->state = CPATH_STATE_OPEN;
-  circuit_build_times_handle_completed_hop(or_circ);
+  origin_circ->cpath->state = CPATH_STATE_OPEN;
+  circuit_build_times_handle_completed_hop(origin_circ);
   tt_int_op(get_circuit_build_times()->total_build_times, OP_EQ, 0);
 
-  or_circ->cpath->next->state = CPATH_STATE_OPEN;
-  circuit_build_times_handle_completed_hop(or_circ);
+  origin_circ->cpath->next->state = CPATH_STATE_OPEN;
+  circuit_build_times_handle_completed_hop(origin_circ);
   tt_int_op(get_circuit_build_times()->total_build_times, OP_EQ, 0);
 
   // Third hop: We should count it now.
-  or_circ->cpath->next->next->state = CPATH_STATE_OPEN;
-  circuit_build_times_handle_completed_hop(or_circ);
+  origin_circ->cpath->next->next->state = CPATH_STATE_OPEN;
+  circuit_build_times_handle_completed_hop(origin_circ);
   tt_int_op(get_circuit_build_times()->total_build_times, OP_EQ,
             !should_timeout); // 1 if counted, 0 otherwise
 
   // Fourth hop: Don't double count
-  or_circ->cpath->next->next->next->state = CPATH_STATE_OPEN;
-  circuit_build_times_handle_completed_hop(or_circ);
+  origin_circ->cpath->next->next->next->state = CPATH_STATE_OPEN;
+  circuit_build_times_handle_completed_hop(origin_circ);
   tt_int_op(get_circuit_build_times()->total_build_times, OP_EQ,
             !should_timeout);
 
  done:
-  return or_circ;
+  return origin_circ;
 }
 
-void
+static void
 test_circuitstats_hoplen(void *arg)
 {
   /* Plan: