chore

Hendrixer · Hendrixer · commit d2defad11a91 · 2018-10-11T07:41:25.000-05:00
diff --git a/exercises/connect.js b/exercises/connect.js
@@ -1,4 +1,5 @@
 const mongoose = require('mongoose')
+mongoose.Promise = global.Promise
 
 const connect = (url) => mongoose.connect(url, {
   useNewUrlParser: true
diff --git a/exercises/hooks/__test__/org.spec.js b/exercises/hooks/__test__/org.spec.js
@@ -0,0 +1,17 @@
+const Project = require('../project')
+const Org = require('../org')
+const mongoose = require('mongoose')
+
+describe('Org model', () => {
+  test('removes projects when org is remove', async () => {
+    const org = await Org.create({name: 'org'})
+    await Project.create([
+      {name: 'project1', org: org.id},
+      {name: 'project', org: org.id}
+    ])
+    
+    await org.remove()
+    const matchedProjects = await Project.find({org: org._id}).exec()
+    expect(matchedProjects).toHaveLength(0)
+  })
+})
diff --git a/exercises/hooks/__test__/project.spec.js b/exercises/hooks/__test__/project.spec.js
@@ -0,0 +1,30 @@
+const Project = require('../project')
+const mongoose = require('mongoose')
+
+describe('Project model', () => {
+  test('project names are unique per org', async () => {
+    expect.assertions(1)
+
+    const org = mongoose.Types.ObjectId()
+    const name = 'name'
+    try {
+      await Project.init()
+      await Project.create([
+        {org, name},
+        {org, name}
+      ])
+    } catch (e) {
+      expect(e).toBeTruthy()
+    }
+  })
+  test('budgetLeft virtual should calculate budget left', async () => {
+    const project = await Project.create({
+      name: 'p1',
+      org: mongoose.Types.ObjectId(),
+      budget: 4000,
+      spent: 1000
+    })
+
+    expect(project.budgetLeft).toBe(3000)
+  })
+})
diff --git a/exercises/hooks/org.js b/exercises/hooks/org.js
@@ -0,0 +1,37 @@
+const mongoose = require('mongoose')
+const Project = require('./project')
+const cdnUrl = 'https://cdn.adminapp.com'
+
+const orgSchema = new mongoose.Schema({
+  name: {
+    type: String,
+    required: true,
+    unique: true
+  },
+  subscription: {
+    status: {
+      type: String,
+      required: true,
+      default: ['active'],
+      enum: ['active', 'trialing', 'overdue', 'canceled']
+    },
+    last4: {
+      type: Number,
+      min: 4,
+      max: 4
+    }
+  }
+})
+
+
+orgSchema.post('remove', async (doc, next) => {
+  console.log('hello')
+  await Project.remove({org: doc._id}).exec()
+  next()
+})
+
+orgSchema.virtual('avatar').get(function() {
+  return `${cdnUrl}/${this._id.toString()}`
+})
+
+module.exports = mongoose.model('org', orgSchema)
diff --git a/exercises/hooks/project.js b/exercises/hooks/project.js
@@ -0,0 +1,37 @@
+const mongoose = require('mongoose')
+
+const projectSchema = new mongoose.Schema({
+  name: {
+    type: String,
+    required: true
+  },
+  org: {
+    type: mongoose.Schema.Types.ObjectId,
+    ref: 'org',
+    required: true
+  },
+  dueOn: Date,
+  budget: {
+    type: Number,
+    default: 0
+  },
+  spent: {
+    type: Number,
+    default: 0
+  },
+  onTrack: {
+    type: Boolean,
+    default: false
+  }
+})
+
+projectSchema.index({
+  org: 1,
+  name: 1
+}, {unique: true})
+
+
+projectSchema.virtual('budgetLeft').get(function() {
+  return this.budget - this.spent
+})
+module.exports = mongoose.model('project', projectSchema)
diff --git a/exercises/queries/__test__/queries.spec.js b/exercises/queries/__test__/queries.spec.js
@@ -100,12 +100,6 @@ describe('queries', () => {
   describe('addRelatedPosts', () => {
     test('should not overrided related posts that are there', async () => {
       const author = mongoose.Types.ObjectId()
-
-      // const posts = await Post.create([
-      //   {title: 'learn things', content: createConent(100), contentLength: 100, author},
-      //   {title: 'lean more things', content: createConent(100), contentLength: 100, author},
-      //   {title: 'lean more things++', content: createConent(100), contentLength: 100, author}
-      // ])
       const post = await Post.create({
         author,
         title: 'Post',

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`const mongoose = require('mongoose')`
	`2`	`+mongoose.Promise = global.Promise`
`2`	`3`
`3`	`4`	`const connect = (url) => mongoose.connect(url, {`
`4`	`5`	`useNewUrlParser: true`