Make get_total_system_memory mockable.

This patch makes get_total_system_memory mockable, which allows us to
alter the return value of the function in tests.

See: https://bugs.torproject.org/24782

diff --git a/src/common/compat.c b/src/common/compat.c
index 4cb346dfa59cfd0e9aca83f910f3d5fc215882a0..7d9add50b29c8388543d611c28ba8e5e970a681b 100644 (file)
--- a/src/common/compat.c
+++ b/src/common/compat.c
@@ -3409,8 +3409,8 @@ get_total_system_memory_impl(void)
  * Try to find out how much physical memory the system has. On success,
  * return 0 and set *<b>mem_out</b> to that value. On failure, return -1.
  */
-int
-get_total_system_memory(size_t *mem_out)
+MOCK_IMPL(int,
+get_total_system_memory, (size_t *mem_out))
 {
   static size_t mem_cached=0;
   uint64_t m = get_total_system_memory_impl();

diff --git a/src/common/compat.h b/src/common/compat.h
index 93301feda01136a5077962df87d2e1650d6c4369..3088e6835505efb3f9d1267afd137d8c41a8f114 100644 (file)
--- a/src/common/compat.h
+++ b/src/common/compat.h
@@ -699,7 +699,7 @@ char *make_path_absolute(char *fname);
 
 char **get_environment(void);
 
-int get_total_system_memory(size_t *mem_out);
+MOCK_DECL(int, get_total_system_memory, (size_t *mem_out));
 
 int compute_num_cpus(void);