Make cpuworker_queue_work function mockable.

I'll be using this in the unit tests for consdiffmgr.

diff --git a/src/or/cpuworker.c b/src/or/cpuworker.c
index f79cd067cb1508b44e8f46bdca2acd56a0188c48..af79fafaa639b0a6bef98f0f01b4564cf48de320 100644 (file)
--- a/src/or/cpuworker.c
+++ b/src/or/cpuworker.c
@@ -480,10 +480,10 @@ queue_pending_tasks(void)
 }
 
 /** DOCDOC */
-workqueue_entry_t *
-cpuworker_queue_work(workqueue_reply_t (*fn)(void *, void *),
-                     void (*reply_fn)(void *),
-                     void *arg)
+MOCK_IMPL(workqueue_entry_t *,
+cpuworker_queue_work,(workqueue_reply_t (*fn)(void *, void *),
+                      void (*reply_fn)(void *),
+                      void *arg))
 {
   tor_assert(threadpool);
 

diff --git a/src/or/cpuworker.h b/src/or/cpuworker.h
index 9bc5d962dd140e4ec8fb0aab1113c8cf7f36005b..aedf2fae323986088c4f37ab99b33aed1856169b 100644 (file)
--- a/src/or/cpuworker.h
+++ b/src/or/cpuworker.h
@@ -16,10 +16,10 @@ void cpu_init(void);
 void cpuworkers_rotate_keyinfo(void);
 struct workqueue_entry_s;
 enum workqueue_reply_t;
-struct workqueue_entry_s *cpuworker_queue_work(
+MOCK_DECL(struct workqueue_entry_s *, cpuworker_queue_work, (
                     enum workqueue_reply_t (*fn)(void *, void *),
                     void (*reply_fn)(void *),
-                    void *arg);
+                    void *arg));
 
 struct create_cell_t;
 int assign_onionskin_to_cpuworker(or_circuit_t *circ,