Browse Source

Make scheduler_release_channel() mockable

Andrea Shepard 10 years ago
parent
commit
85ee070852
2 changed files with 4 additions and 3 deletions
  1. 2 2
      src/or/scheduler.c
  2. 2 1
      src/or/scheduler.h

+ 2 - 2
src/or/scheduler.c

@@ -360,8 +360,8 @@ scheduler_retrigger(void)
 
 /** Notify the scheduler of a channel being closed */
 
-void
-scheduler_release_channel(channel_t *chan)
+MOCK_IMPL(void,
+scheduler_release_channel,(channel_t *chan))
 {
   tor_assert(chan);
   tor_assert(channels_pending);

+ 2 - 1
src/or/scheduler.h

@@ -11,6 +11,7 @@
 
 #include "or.h"
 #include "channel.h"
+#include "testsupport.h"
 
 /* Global-visibility scheduler functions */
 
@@ -25,7 +26,7 @@ void scheduler_channel_has_waiting_cells(channel_t *chan);
 void scheduler_channel_wants_writes(channel_t *chan);
 
 /* Notify the scheduler of a channel being closed */
-void scheduler_release_channel(channel_t *chan);
+MOCK_DECL(void,scheduler_release_channel,(channel_t *chan));
 
 /* Notify scheduler of queue size adjustments */
 void scheduler_adjust_queue_size(channel_t *chan, char dir, uint64_t adj);