initial part-tested edition

tests/count.test.ts

@@ -0,0 +1,88 @@
+import { Database } from 'arangojs'
+import { DocumentMetadata } from 'arangojs/documents'
+import { expect } from 'chai'
+import lib from '../lib'
+
+const testData: Record<string, (Pick<DocumentMetadata, '_key'> & Record<string, unknown>)[]> = {
+  pets: [
+    { _key: 'greedo', name: 'Greedo', age: 5, species: 'cat' },
+    { _key: 'haribo', name: 'Haribo', age: 1.5, species: 'dog' },
+    { _key: 'iguana', name: 'Iguana', age: 3, species: 'dog' },
+    { _key: 'jerkins', name: 'Jerkins', age: 15, species: 'cat' },
+    { _key: 'kahlua', name: 'Kahlua', age: 0.5, species: 'hamster' },
+    { _key: 'lemonade', name: 'Lemonade', age: 9, species: 'dog' }
+  ],
+  staff: [
+    { _key: 'aaron', name: 'Aaron', age: 38 },
+    { _key: 'benedict', name: 'Benedict', age: 25 },
+    { _key: 'cheryl', name: 'Cheryl', age: 34 },
+    { _key: 'davide', name: 'Davide', age: 52 },
+    { _key: 'emma', name: 'Emma', age: 23 },
+    { _key: 'frank', name: 'Frank', age: 47 }
+  ]
+}
+
+const getDatabase = async () => {
+  let db = new Database({
+    url: 'arangodb://127.0.0.1:8529'
+  })
+  if (!await db.database('arangosearch').exists()) {
+    await db.createDatabase('arangosearch')
+  }
+  db = db.database('arangosearch')
+  await installTestData(db)
+  return db
+}
+
+const installTestData = async (db: Database) => {
+  for (const c in testData) {
+    const collection = db.collection(c)
+    if (!await collection.exists()) {
+      await db.createCollection(c)
+      await collection.saveAll(testData[c], { overwriteMode: 'ignore' })
+    }
+  }
+}
+
+describe('count()', () => {
+  for (const c in testData) {
+    describe(c, () => {
+      const expected = testData[c].length
+      it(`should match ${expected} documents`, async () => {
+        const db = await getDatabase()
+        const count = await lib.count(db, db.collection(c))()
+        expect(count).to.equal(expected)
+      })
+    })
+  }
+})
+
+describe('count({ age: { gt: 5 }})', () => {
+  for (const c in testData) {
+    describe(c, () => {
+      const expected = testData[c].filter(i => i.age && i.age > 5).length
+      it(`should match ${expected} documents`, async () => {
+        const db = await getDatabase()
+        const count = await lib.count(db, db.collection(c))({
+          age: { gt: 5 }
+        })
+        expect(count).to.equal(expected)
+      })
+    })
+  }
+})
+
+describe('count({ name: { eq: "Iguana" }})', () => {
+  for (const c in testData) {
+    describe(c, () => {
+      const expected = testData[c].filter(i => i.name === 'Iguana').length
+      it(`should match ${expected} documents`, async () => {
+        const db = await getDatabase()
+        const count = await lib.count(db, db.collection(c))({
+          name: { eq: 'Iguana' }
+        })
+        expect(count).to.equal(expected)
+      })
+    })
+  }
+})