From a2de0a1034b3d4c1433c548526a1401e51540bc1 Mon Sep 17 00:00:00 2001
From: Andrea Shepard <andrea@torproject.org>
Date: Thu, 23 Jan 2014 06:42:14 -0800
Subject: [PATCH] Make buf_datalen() mockable

---
 src/or/buffers.c | 4 ++--
 src/or/buffers.h | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/or/buffers.c b/src/or/buffers.c
index 033f86288e..85eff36b4c 100644
--- a/src/or/buffers.c
+++ b/src/or/buffers.c
@@ -559,8 +559,8 @@ buf_clear(buf_t *buf)
 }
 
 /** Return the number of bytes stored in <b>buf</b> */
-size_t
-buf_datalen(const buf_t *buf)
+MOCK_IMPL(size_t,
+buf_datalen, (const buf_t *buf))
 {
   return buf->datalen;
 }
diff --git a/src/or/buffers.h b/src/or/buffers.h
index c90e14750e..6b91a2d6c1 100644
--- a/src/or/buffers.h
+++ b/src/or/buffers.h
@@ -24,7 +24,7 @@ void buf_shrink(buf_t *buf);
 size_t buf_shrink_freelists(int free_all);
 void buf_dump_freelist_sizes(int severity);
 
-size_t buf_datalen(const buf_t *buf);
+MOCK_DECL(size_t, buf_datalen, (const buf_t *buf));
 size_t buf_allocation(const buf_t *buf);
 size_t buf_slack(const buf_t *buf);