Browse Source

Make scheduler_run() mockable

Andrea Shepard 10 years ago
parent
commit
f8ceb0f028
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

@@ -376,8 +376,8 @@ scheduler_release_channel,(channel_t *chan))
 
 /** Run the scheduling algorithm if necessary */
 
-void
-scheduler_run(void)
+MOCK_IMPL(void,
+scheduler_run, (void))
 {
   int n_cells, n_chans_before, n_chans_after;
   uint64_t q_len_before, q_heur_before, q_len_after, q_heur_after;

+ 1 - 1
src/or/scheduler.h

@@ -18,7 +18,7 @@
 /* Set up and shut down the scheduler from main.c */
 void scheduler_free_all(void);
 void scheduler_init(void);
-void scheduler_run(void);
+MOCK_DECL(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));