1 files changed, 84 insertions, 0 deletions
diff --git a/tests/libtap/test.h b/tests/libtap/test.h
new file mode 100644
index 0000000..1c442f5
--- /dev/null
+++ b/tests/libtap/test.h
@@ -0,0 +1,84 @@
+#ifndef LIBMONGO_CLIENT_TEST_H
+#define LIBMONGO_CLIENT_TEST_H 1
+
+#include "tap.h"
+#include "bson.h"
+#include "mongo-wire.h"
+#include "mongo-sync.h"
+#include "libmongo-private.h"
+
+#include <dlfcn.h>
+
+typedef struct
+{
+  gchar *primary_host;
+  gint primary_port;
+
+  gchar *secondary_host;
+  gint secondary_port;
+
+  gchar *db;
+  gchar *coll;
+  gchar *ns;
+
+  gchar *gfs_prefix;
+} func_config_t;
+
+extern func_config_t config;
+
+#define begin_network_tests(n)						\
+  do									\
+    {									\
+      skip(!test_env_setup (), n, "Environment not set up for network tests")
+
+#define end_network_tests()			\
+      endskip;					\
+      test_env_free();				\
+    } while (0)
+
+#define RUN_TEST(n, t) \
+  int                  \
+  main (void)          \
+  {                    \
+    test_main_setup(); \
+    plan (n);          \
+    test_##t ();       \
+    return 0;          \
+  }
+
+gboolean test_env_setup (void);
+void test_env_free (void);
+void test_main_setup (void);
+
+#define RUN_NET_TEST(n, t)						\
+  int									\
+  main (void)								\
+  {									\
+    test_main_setup();                                                  \
+    if (!test_env_setup ())						\
+      printf ("1..0 # skip, Environment not set up for network tests"); \
+    else								\
+      {									\
+        plan (n);							\
+        test_##t ();							\
+      }									\
+    test_env_free ();							\
+    return 0;								\
+  }
+
+bson *test_bson_generate_full (void);
+mongo_packet *test_mongo_wire_generate_reply (gboolean valid,
+                                              gint32 nreturn,
+                                              gboolean with_docs);
+mongo_sync_connection *test_make_fake_sync_conn (gint fd,
+                                                 gboolean slaveok);
+
+#define SAVE_OLD_FUNC(n)				\
+  static void *(*func_##n)();				\
+  if (!func_##n)					\
+    func_##n = (void *(*)())dlsym (RTLD_NEXT, #n);
+
+#define CALL_OLD_FUNC(n, ...)			\
+  func_##n (__VA_ARGS__)
+
+#endif