1 files changed, 47 insertions, 0 deletions
diff --git a/tests/unit/mongo/sync/sync_connect_from_cache_enforce_primary.c b/tests/unit/mongo/sync/sync_connect_from_cache_enforce_primary.c
new file mode 100644
index 0000000..5c48ae9
--- /dev/null
+++ b/tests/unit/mongo/sync/sync_connect_from_cache_enforce_primary.c
@@ -0,0 +1,47 @@
+#include "test.h"
+#include "mongo.h"
+
+#define NETWORK_TESTS_NUM 5
+
+void
+test_mongo_sync_connect_from_cache_enforce_primary (void)
+{
+  mongo_sync_conn_recovery_cache *cache;
+  mongo_sync_connection *c;
+
+  begin_network_tests (NETWORK_TESTS_NUM);
+
+  cache = mongo_sync_conn_recovery_cache_new ();
+
+  skip (!config.secondary_host,
+        NETWORK_TESTS_NUM,
+        "Secondary server not configured");
+
+  ok (mongo_sync_conn_recovery_cache_seed_add (cache, config.secondary_host,
+                                               config.secondary_port) == TRUE,
+      "mongo_sync_conn_recovery_seed_add() works");
+
+  ok ((c = mongo_sync_connect_recovery_cache (cache, TRUE)) != NULL,
+      "mongo_sync_connect_recovery_cache() works");
+
+  ok (mongo_sync_cmd_is_master(c) == FALSE,
+      "Secondary server should not be The Master.");
+
+  mongo_sync_disconnect (c);
+
+  ok ((c = mongo_sync_connect_recovery_cache (cache, FALSE)) != NULL,
+      "mongo_sync_connect_recovery_cache() works");
+
+  ok (mongo_sync_cmd_is_master (c) == TRUE,\
+      "Retrieved connection should be The Master when it is forced to be.");
+
+  mongo_sync_disconnect (c);
+
+  endskip;
+
+  mongo_sync_conn_recovery_cache_free (cache);
+
+  end_network_tests ();
+}
+
+RUN_TEST (NETWORK_TESTS_NUM, mongo_sync_connect_from_cache_enforce_primary);