refactor db module exporting UsersCollection, unit-testing repository

index.test.js

@@ -1,17 +1,17 @@
 const test = require('ava')
 const sinon = require('sinon')
-const db = require('./lib/db')
+const { UsersCollection } = require('./lib/db')
 
 const userRepository = require('./lib/user-repository')
 const emailService = require('./lib/email-service')
 const newsletter = require('.')
 
 test.beforeEach(async () => {
- await db.get('users').remove({})
+ await UsersCollection.remove({})
 })
 
 test('sends a newsletter to users that did not yet receive it', async t => {
- await db.get('users').insert({ name: 'test', email: '[email protected]', lastEmailSentAt: null })
+ await UsersCollection.insert({ name: 'test', email: '[email protected]', lastEmailSentAt: null })
 
  sinon.spy(userRepository, 'findNotYetReceivedNewsletter')
  sinon.stub(emailService, 'sendTo')

lib/db.js

@@ -1,4 +1,6 @@
 const monk = require('monk')
 const db = monk(process.env.MONGO_URL)
 
-module.exports = db
+module.exports = {
+ UsersCollection: db.get('users')
+}

lib/user-repository.js

@@ -1,8 +1,8 @@
-const db = require('./db')
+const { UsersCollection } = require('./db')
 
 module.exports = {
  findNotYetReceivedNewsletter: () => {
- return db.get('users').find({
+ return UsersCollection.find({
  lastEmailSentAt: null
  })
  }

lib/user-repository.test.js

@@ -1,15 +1,21 @@
 const test = require('ava')
-const db = require('./db')
+const sinon = require('sinon')
 const userRepository = require('./user-repository')
+const { UsersCollection } = require('./db')
 
 test.beforeEach(async () => {
- await db.get('users').remove({})
+ await UsersCollection.remove({})
 })
 
 test('find users that did not yet receive the newsletter', async t => {
- await db.get('users').insert({ name: 'test', email: '[email protected]', lastEmailSentAt: null })
+ await UsersCollection.insert({ name: 'test', email: '[email protected]', lastEmailSentAt: null })
+
+ sinon.spy(UsersCollection, 'find')
 
  const users = await userRepository.findNotYetReceivedNewsletter()
  t.true(Array.isArray(users))
  t.is(users.length, 1)
+
+ t.true(UsersCollection.find.calledOnce)
+ t.true(UsersCollection.find.calledWith({ lastEmailSentAt: null }))
 })