tests: add basic write command test with splitting.

Author: Christian Hergert

CMakeLists.txt

@@ -214,6 +214,7 @@ set(test-libmongoc-sources
    ${SOURCE_DIR}/tests/test-mongoc-stream.c
    ${SOURCE_DIR}/tests/test-mongoc-uri.c
    ${SOURCE_DIR}/tests/test-mongoc-write-concern.c
+   ${SOURCE_DIR}/tests/test-write-command.c
    ${SOURCE_DIR}/tests/TestSuite.c
 )
 

tests/Makefile.am

@@ -83,6 +83,7 @@ test_libmongoc_SOURCES = \
 	tests/test-mongoc-uri.c \
 	tests/test-mongoc-write-concern.c \
 	tests/test-libmongoc.h \
+	tests/test-write-commands.c \
 	tests/TestSuite.c \
 	tests/TestSuite.h
 if ENABLE_SSL

tests/test-libmongoc.c

@@ -42,6 +42,7 @@ extern void test_read_prefs_install       (TestSuite *suite);
 extern void test_rpc_install              (TestSuite *suite);
 extern void test_stream_install           (TestSuite *suite);
 extern void test_uri_install              (TestSuite *suite);
+extern void test_write_command_install    (TestSuite *suite);
 extern void test_write_concern_install    (TestSuite *suite);
 #ifdef MONGOC_ENABLE_SSL
 extern void test_x509_install             (TestSuite *suite);
@@ -127,10 +128,11 @@ main (int   argc,
    TestSuite_Init (&suite, "", argc, argv);
 
    test_array_install (&suite);
-   test_bulk_install (&suite);
    test_buffer_install (&suite);
    test_client_install (&suite);
    test_client_pool_install (&suite);
+   test_write_command_install (&suite);
+   test_bulk_install (&suite);
    test_collection_install (&suite);
    test_cursor_install (&suite);
    test_database_install (&suite);

tests/test-write-commands.c

@@ -0,0 +1,110 @@
+#include <bcon.h>
+#include <mongoc.h>
+
+#include "mongoc-collection-private.h"
+#include "mongoc-write-command-private.h"
+
+#include "TestSuite.h"
+
+#include "test-libmongoc.h"
+#include "mongoc-tests.h"
+
+static char *gTestUri;
+
+
+static mongoc_collection_t *
+get_test_collection (mongoc_client_t *client,
+                     const char      *prefix)
+{
+   mongoc_collection_t *ret;
+   char *str;
+
+   str = gen_collection_name (prefix);
+   ret = mongoc_client_get_collection (client, "test", str);
+   bson_free (str);
+
+   return ret;
+}
+
+
+static void
+test_split_insert (void)
+{
+   mongoc_write_command_t command;
+   mongoc_write_result_t result;
+   mongoc_collection_t *collection;
+   mongoc_client_t *client;
+   bson_oid_t oid;
+   bson_t **docs;
+   bson_t reply = BSON_INITIALIZER;
+   bson_error_t error;
+   const char *key;
+   char str [12];
+   int i;
+   int j;
+   bool r;
+
+   client = mongoc_client_new (gTestUri);
+   assert (client);
+
+   collection = get_test_collection (client, "test_split_insert");
+   assert (collection);
+
+   docs = bson_malloc (sizeof(bson_t*) * 3000);
+
+   for (i = 0; i < 3000; i++) {
+      docs [i] = bson_new ();
+      bson_oid_init (&oid, NULL);
+      BSON_APPEND_OID (docs [i], "_id", &oid);
+      for (j = 0; j < 20; j++) {
+         bson_uint32_to_string (j, &key, str, sizeof str);
+         BSON_APPEND_INT64 (docs [i], key, 1234);
+      }
+   }
+
+   _mongoc_write_result_init (&result);
+
+   _mongoc_write_command_init_insert (&command,
+                                      (const bson_t * const *)docs,
+                                      3000, true);
+
+   _mongoc_write_command_execute (&command, client, 0, collection->db,
+                                  collection->collection, NULL, &result);
+
+   r = _mongoc_write_result_complete (&result, &reply, &error);
+
+   assert (r);
+   assert (result.nInserted == 3000);
+
+   _mongoc_write_command_destroy (&command);
+   _mongoc_write_result_destroy (&result);
+
+   r = mongoc_collection_drop (collection, &error);
+   assert (r);
+
+   for (i = 0; i < 3000; i++) {
+      bson_destroy (docs [i]);
+   }
+
+   bson_free (docs);
+
+   mongoc_collection_destroy (collection);
+   mongoc_client_destroy (client);
+}
+
+
+static void
+cleanup_globals (void)
+{
+   bson_free (gTestUri);
+}
+
+void
+test_write_command_install (TestSuite *suite)
+{
+   gTestUri = bson_strdup_printf("mongodb://%s/", MONGOC_TEST_HOST);
+
+   TestSuite_Add (suite, "/WriteCommand/split_insert", test_split_insert);
+
+   atexit (cleanup_globals);
+}