Browse Source

Make scheduler_channel_doesnt_want_writes() mockable

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

+ 2 - 2
src/or/scheduler.c

@@ -275,8 +275,8 @@ scheduler_channel_doesnt_want_writes,(channel_t *chan))
 
 /** Mark a channel as having waiting cells */
 
-void
-scheduler_channel_has_waiting_cells(channel_t *chan)
+MOCK_IMPL(void,
+scheduler_channel_has_waiting_cells,(channel_t *chan))
 {
   int became_pending = 0;
 

+ 1 - 1
src/or/scheduler.h

@@ -22,7 +22,7 @@ void scheduler_run(void);
 
 /* Mark channels as having cells or wanting/not wanting writes */
 MOCK_DECL(void,scheduler_channel_doesnt_want_writes,(channel_t *chan));
-void scheduler_channel_has_waiting_cells(channel_t *chan);
+MOCK_DECL(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 */