1 files changed, 66 insertions, 0 deletions
diff --git a/tests/unit/bson/bson_append_js_code.c b/tests/unit/bson/bson_append_js_code.c
new file mode 100644
index 0000000..4b250b0
--- /dev/null
+++ b/tests/unit/bson/bson_append_js_code.c
@@ -0,0 +1,66 @@
+#include "tap.h"
+#include "test.h"
+#include "bson.h"
+
+#include <string.h>
+
+void
+test_bson_js_code (void)
+{
+  bson *b;
+
+  /* Test #1: A single JS element, with default size. */
+  b = bson_new ();
+  ok (bson_append_javascript (b, "hello",
+                              "function () { print (\"hello world!\"); }", -1),
+      "bson_append_javascript() works");
+  bson_finish (b);
+
+  cmp_ok (bson_size (b), "==", 56, "BSON javascript element size check");
+  ok (memcmp (bson_data (b),
+              "\070\000\000\000\015\150\145\154\154\157\000\050\000\000\000"
+              "\146\165\156\143\164\151\157\156\040\050\051\040\173\040\160"
+              "\162\151\156\164\040\050\042\150\145\154\154\157\040\167\157"
+              "\162\154\144\041\042\051\073\040\175\000\000",
+              bson_size (b)) == 0,
+      "BSON javascript element contents check");
+  bson_free (b);
+
+  /* Test #2: A single javascript element, with explicit length. */
+  b = bson_new ();
+  ok (bson_append_javascript (b, "hello",
+                              "print (\"hello world!\"); garbage is gone.",
+                          strlen ("print (\"hello world!\");")),
+      "bson_append_javascript() with explicit length works");
+  bson_finish (b);
+
+  cmp_ok (bson_size (b), "==", 40, "BSON javascript element size check, #2");
+  ok (memcmp (bson_data (b),
+              "\050\000\000\000\015\150\145\154\154\157\000\030\000\000\000"
+              "\160\162\151\156\164\040\050\042\150\145\154\154\157\040\167"
+              "\157\162\154\144\041\042\051\073\000\000",
+              bson_size (b)) == 0,
+      "BSON javascript element contents check, #2");
+  bson_free (b);
+
+  /* Test #3: Negative test, passing an invalid arguments. */
+  b = bson_new ();
+  ok (bson_append_javascript (b, "hello", "print();", -42) == FALSE,
+      "bson_append_javascript() with an invalid length should fail");
+  ok (bson_append_javascript (b, NULL, "print();", -1) == FALSE,
+      "bson_append_javascript() should fail without a key name");
+  ok (bson_append_javascript (b, "hello", NULL, -1) == FALSE,
+      "bson_append_javascript() should fail without javascript code");
+  ok (bson_append_javascript (NULL, "hello", "print();", -1) == FALSE,
+      "bson_append_javascript() should fail without a BSON object");
+  bson_finish (b);
+  cmp_ok (bson_size (b), "==", 5,
+          "BSON object should be empty");
+
+  ok (bson_append_javascript (b, "js", "print();", -1) == FALSE,
+      "Appending to a finished element should fail");
+
+  bson_free (b);
+}
+
+RUN_TEST (12, bson_js_code);