From 2afebf955f630dc7e49dcf3c1cc547cf49dd098d Mon Sep 17 00:00:00 2001
From: Nikita Lutsenko <nlutsenko@me.com>
Date: Wed, 2 Mar 2016 21:33:33 -0800
Subject: [PATCH] Completely migrate SchemasRouter to new MongoCollection API.

---
 spec/schemas.spec.js         |  4 ++--
 src/Routers/SchemasRouter.js | 20 ++++++++++----------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/spec/schemas.spec.js b/spec/schemas.spec.js
index 24589a38..af5dabc1 100644
--- a/spec/schemas.spec.js
+++ b/spec/schemas.spec.js
@@ -175,7 +175,7 @@ describe('schemas', () => {
         expect(response.statusCode).toEqual(400);
         expect(body).toEqual({
           code: 103,
-          error: 'class HASALLPOD does not exist',
+          error: 'Class HASALLPOD does not exist.',
         });
         done();
       });
@@ -733,7 +733,7 @@ describe('schemas', () => {
             //Expect _SCHEMA entry to be gone.
             expect(response.statusCode).toEqual(400);
             expect(body.code).toEqual(Parse.Error.INVALID_CLASS_NAME);
-            expect(body.error).toEqual('class MyOtherClass does not exist');
+            expect(body.error).toEqual('Class MyOtherClass does not exist.');
             done();
           });
         });
diff --git a/src/Routers/SchemasRouter.js b/src/Routers/SchemasRouter.js
index 70b3157e..ec6e134b 100644
--- a/src/Routers/SchemasRouter.js
+++ b/src/Routers/SchemasRouter.js
@@ -45,16 +45,16 @@ function getAllSchemas(req) {
 }
 
 function getOneSchema(req) {
-  return req.config.database.collection('_SCHEMA')
-    .then(coll => coll.findOne({'_id': req.params.className}))
-    .then(schema => ({response: mongoSchemaToSchemaAPIResponse(schema)}))
-    .catch(() => ({
-      status: 400,
-      response: {
-        code: 103,
-        error: 'class ' + req.params.className + ' does not exist',
+  const className = req.params.className;
+  return req.config.database.adaptiveCollection('_SCHEMA')
+    .then(collection => collection.find({ '_id': className }, { limit: 1 }))
+    .then(results => {
+      if (results.length != 1) {
+        throw new Parse.Error(Parse.Error.INVALID_CLASS_NAME, `Class ${className} does not exist.`);
       }
-    }));
+      return results[0];
+    })
+    .then(schema => ({ response: mongoSchemaToSchemaAPIResponse(schema) }));
 }
 
 function createSchema(req) {
@@ -70,7 +70,7 @@ function createSchema(req) {
       response: {
         code: 135,
         error: 'POST ' + req.path + ' needs class name',
-      },
+      }
     });
   }
   return req.config.database.loadSchema()