Add insert(), insertOne()

baumandm · May 23, 2017 · c4fe378 · c4fe378
1 parent db00d8e
commit c4fe378
Show file tree

Hide file tree

Showing 4 changed files with 52 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## 0.7.0
+
+- Add db.collection.insert() and db.collection.insertOne() functions
+
 ## 0.6.0
 
 - Add Date() and ISODate() functions 

diff --git a/project.clj b/project.clj
@@ -1,4 +1,4 @@
-(defproject monglorious "0.6.0"
+(defproject monglorious "0.7.0"
   :author "Dave Bauman"
   :description "Query MongoDB using strings!"
   :url "https://github.com/baumandm/monglorious"

diff --git a/src/monglorious/transforms.clj b/src/monglorious/transforms.clj
@@ -86,6 +86,16 @@
         "aggregate"
         (fn [_ db] (apply (partial mg-coll/aggregate db collection-name) args))
 
+        "insert"
+        (let [document-or-documents (first args)]
+          (if (sequential? document-or-documents)
+            (fn [_ db] (mg-coll/insert-batch db collection-name document-or-documents))
+            (fn [_ db] (apply (partial mg-coll/insert-and-return db collection-name) args))
+            ))
+
+        "insertone"
+        (fn [_ db] (apply (partial mg-coll/insert-and-return db collection-name) args))
+
         (throw (Exception. (format "Unsupported function: %s." function-name)))))
 
     ;; More than one function

diff --git a/test/monglorious/transforms_test.clj b/test/monglorious/transforms_test.clj
@@ -7,7 +7,7 @@
             [clj-time.format :as f]
             [clj-time.coerce :as c])
   (:use midje.sweet)
-  (:import (com.mongodb MongoQueryException)))
+  (:import (com.mongodb MongoQueryException WriteResult)))
 
 ;; Test run-command-transform()
 (against-background
@@ -285,3 +285,39 @@
         (execute {} "testdb" "db.timeDocs.find({ time: { $lt: new ISODate('2017-02-15T00:00:00.000Z') } }).count()") => 1
         (execute {} "testdb" "db.timeDocs.find({ time: { $lte: new ISODate('2017-02-15T00:00:00.000Z') } }).count()") => 2
         (execute {} "testdb" "db.timeDocs.find({ time: { $lt: new ISODate('2017-02-17') } }).count()") => 2))
+
+;; Test collect-command-transform() with insert()
+(against-background
+  [(before :contents
+           (let [conn (mg/connect)
+                 db (mg/get-db conn "testdb")]
+             (if db (mc/drop db "documents"))))
+   (after :contents
+          (let [conn (mg/connect)]
+            (mg/drop-db conn "testdb")))]
+
+  (fact "Monglorious inserts a document"
+        (execute {} "testdb" "db.documents.insert({name: 'TEST'})") => #(= "TEST" (get % "name"))
+        (execute {} "testdb" "db.documents.insert({name: 'DOCUMENT'})") => (fn [_]
+                                                                             (let [conn (mg/connect)
+                                                                                   db (mg/get-db conn "testdb")
+                                                                                   docs (mc/find-maps db "documents")]
+                                                                               (and (= 2 (count docs))
+                                                                                    (= #{"TEST" "DOCUMENT"} (set (map :name docs)))))))
+
+  (fact "Monglorious inserts several documents"
+        (execute {} "testdb" "db.documents.insert([{name: 'BATCH 1'}, {name: 'BATCH 2'}])") => (fn [_]
+                                                                                                 (let [conn (mg/connect)
+                                                                                                       db (mg/get-db conn "testdb")
+                                                                                                       docs (mc/find-maps db "documents")]
+                                                                                                   (and (= 4 (count docs))
+                                                                                                        (= #{"TEST" "DOCUMENT" "BATCH 1" "BATCH 2"} (set (map :name docs)))))))
+
+  (fact "Monglorious insertOnes a document"
+        (execute {} "testdb" "db.documents.insertOne({name: 'ONE'})") => #(= "ONE" (get % "name"))
+        (execute {} "testdb" "db.documents.insertOne({name: 'TWO'})") => (fn [_]
+                                                                             (let [conn (mg/connect)
+                                                                                   db (mg/get-db conn "testdb")
+                                                                                   docs (mc/find-maps db "documents")]
+                                                                               (and (= 6 (count docs))
+                                                                                    (= #{"TEST" "DOCUMENT" "BATCH 1" "BATCH 2" "ONE" "TWO"} (set (map :name docs))))))))