custom query tests added

izelnakri · Oct 26, 2017 · a4ad69a · a4ad69a
1 parent be381a4
commit a4ad69a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/test/mem-server.fixtures.js b/test/mem-server.fixtures.js
@@ -166,6 +166,8 @@ describe('MemServer fixture constraint feature', function() {
   });
 
   it('should throw error if any of the uuid fixtures have an incorrect type', function() {
+    this.timeout(5000);
+
     fs.mkdirSync(`./memserver/fixtures`);
     fs.writeFileSync(`${process.cwd()}/memserver/fixtures/photos.js`, `export default [
       {

diff --git a/test/mem-server.model.js b/test/mem-server.model.js
@@ -16,6 +16,9 @@ describe('MemServer.Model Interface', function() {
           name() {
             return faker.name.title();
           }
+        },
+        publicPhotos() {
+          return this.findAll({ is_public: true });
         }
       });
     `);
@@ -29,6 +32,9 @@ describe('MemServer.Model Interface', function() {
             return moment().toJSON();
           },
           is_important: true
+        },
+        forPhoto(photo) {
+          return this.findAll({ photo_id: photo.id });
         }
       });
     `);
@@ -184,8 +190,16 @@ describe('MemServer.Model Interface', function() {
       assert.equal(PhotoComment.count(), 4);
     });
 
-    // TODO: it('can have custom queries for a Model', function() {
-    //
-    // });
+    it('can have custom queries for a Model', function() {
+      const MemServer = require('../index.js');
+      const { Photo, PhotoComment } = MemServer.Models;
+
+      MemServer.start();
+
+      const photo = Photo.find(1);
+
+      assert.deepEqual(PhotoComment.forPhoto(photo), PhotoComent.findAll({ photo_id: photo.id }));
+      assert.deepEqual(Photo.publicPhotos(), Photo.findAll({ is_public: true }));
+    });
   });
 });