1 files changed, 6 insertions, 0 deletions
diff --git a/test/mocks.h b/test/mocks.h
index 97a23b7..67a67b5 100644
--- a/test/mocks.h
+++ b/test/mocks.h
@@ -29,6 +29,7 @@
  */
 
 #include <stdbool.h>
+#include "private.h"
 
 void unpatch_all(void);
 
@@ -36,4 +37,9 @@ void patch(void *fn);
 
 bool is_patched(void *fn);
 
+int
+handle_dirty_pages(vfu_ctx_t *vfu_ctx, uint32_t size,
+                   struct iovec **iovecs, size_t *nr_iovecs,
+                   struct vfio_iommu_type1_dirty_bitmap *dirty_bitmap);
+
 /* ex: set tabstop=4 shiftwidth=4 softtabstop=4 expandtab: */