summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/or/scheduler.c4
-rw-r--r--src/or/scheduler.h2
2 files changed, 3 insertions, 3 deletions
diff --git a/src/or/scheduler.c b/src/or/scheduler.c
index 544ec83258..bf86810517 100644
--- a/src/or/scheduler.c
+++ b/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;
diff --git a/src/or/scheduler.h b/src/or/scheduler.h
index b99491b221..b0b66ee957 100644
--- a/src/or/scheduler.h
+++ b/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));