1 files changed, 216 insertions, 0 deletions
diff --git a/node_modules/mongoose/test/shard.test.js b/node_modules/mongoose/test/shard.test.js
new file mode 100644
index 0000000..2df04c3
--- /dev/null
+++ b/node_modules/mongoose/test/shard.test.js
@@ -0,0 +1,216 @@
+
+var start = require('./common')
+  , should = require('should')
+  , random = require('../lib/utils').random
+  , mongoose = start.mongoose
+  , Mongoose = mongoose.Mongoose
+  , Schema = mongoose.Schema;
+
+var uri = process.env.MONGOOSE_SHARD_TEST_URI;
+
+if (!uri) {
+  console.log('\033[30m', '\n', 'You\'re not testing shards!'
+            , '\n', 'Please set the MONGOOSE_SHARD_TEST_URI env variable.', '\n'
+            , 'e.g: `mongodb://localhost:27017/database', '\n'
+            , 'Sharding must already be enabled on your database'
+            , '\033[39m');
+
+  exports.r = function expressoHack(){}
+  return;
+}
+
+var schema = new Schema({
+    name: String
+  , age: Number
+  , likes: [String]
+}, { shardkey: { name: 1, age: 1 }});
+
+var collection = 'shardperson_' + random();
+mongoose.model('ShardPerson', schema, collection);
+
+var db = start({ uri: uri });
+db.on('open', function () {
+
+  // set up a sharded test collection
+  var P = db.model('ShardPerson', collection);
+
+  var cmd = {};
+  cmd.shardcollection = db.name + '.' + collection;
+  cmd.key = P.schema.options.shardkey;
+
+  P.db.db.executeDbAdminCommand(cmd,function (err, res) {
+    db.close();
+
+    if (err) throw err;
+
+    if (!(res && res.documents && res.documents[0] && res.documents[0].ok)) {
+      throw new Error('could not shard test collection ' + collection);
+    }
+
+    // assign exports to tell expresso to begin
+    Object.keys(tests).forEach(function (test) {
+      exports[test] = tests[test];
+    });
+
+  });
+});
+
+var tests = {
+
+  'can read and write to a shard': function () {
+    var db = start({ uri:  uri })
+    var P = db.model('ShardPerson', collection);
+
+    P.create({ name: 'ryu', age: 25, likes: ['street fighting']}, function (err, ryu) {
+      should.strictEqual(null, err);
+      P.findById(ryu._id, function (err, doc) {
+        db.close();
+        should.strictEqual(null, err);
+        doc.id.should.equal(ryu.id);
+      });
+    });
+  },
+
+  'save() and remove() works with shard keys transparently': function () {
+    var db = start({ uri:  uri })
+    var P = db.model('ShardPerson', collection);
+
+    var zangief = new P({ name: 'Zangief', age: 33 });
+    zangief.save(function (err) {
+      should.strictEqual(null, err);
+
+      zangief._shardval.name.should.equal('Zangief');
+      zangief._shardval.age.should.equal(33);
+
+      P.findById(zangief._id, function (err, zang) {
+        should.strictEqual(null, err);
+
+        zang._shardval.name.should.equal('Zangief');
+        zang._shardval.age.should.equal(33);
+
+        zang.likes = ['spinning', 'laughing'];
+        zang.save(function (err) {
+          should.strictEqual(null, err);
+
+          zang._shardval.name.should.equal('Zangief');
+          zang._shardval.age.should.equal(33);
+
+          zang.likes.addToSet('winning');
+          zang.save(function (err) {
+            should.strictEqual(null, err);
+            zang._shardval.name.should.equal('Zangief');
+            zang._shardval.age.should.equal(33);
+            zang.remove(function (err) {
+              db.close();
+              should.strictEqual(null, err);
+            });
+          });
+        });
+      });
+    });
+  },
+
+  'inserting to a sharded collection without the full shard key fails': function () {
+    var db = start({ uri:  uri })
+    var P = db.model('ShardPerson', collection);
+
+    var pending = 6;
+
+    P.create({ name: 'ryu', likes: ['street fighting']}, function (err, ryu) {
+      --pending || db.close();
+      should.exist(err);
+      err.message.should.equal('tried to insert object with no valid shard key');
+    });
+
+    P.create({ likes: ['street fighting']}, function (err, ryu) { should.exist(err);
+      --pending || db.close();
+      should.exist(err);
+      err.message.should.equal('tried to insert object with no valid shard key');
+    });
+
+    P.create({ name: 'ryu' }, function (err, ryu) { should.exist(err);
+      --pending || db.close();
+      should.exist(err);
+      err.message.should.equal('tried to insert object with no valid shard key');
+    });
+
+    P.create({ age: 49 }, function (err, ryu) { should.exist(err);
+      --pending || db.close();
+      should.exist(err);
+      err.message.should.equal('tried to insert object with no valid shard key');
+    });
+
+    P.create({ likes: ['street fighting'], age: 8 }, function (err, ryu) {
+      --pending || db.close();
+      should.exist(err);
+      err.message.should.equal('tried to insert object with no valid shard key');
+    });
+
+    var p = new P;
+    p.save(function (err) {
+      --pending || db.close();
+      should.exist(err);
+      err.message.should.equal('tried to insert object with no valid shard key');
+    });
+
+  },
+
+  'updating a sharded collection without the full shard key fails': function () {
+    var db = start({ uri:  uri })
+    var P = db.model('ShardPerson', collection);
+
+    P.create({ name: 'ken', age: 27 }, function (err, ken) {
+      should.strictEqual(null, err);
+
+      P.update({ _id: ken._id }, { likes: ['kicking', 'punching'] }, function (err) {
+        should.exist(err);
+        "right object doesn't have full shard key".should.equal(err.message);
+
+        P.update({ _id: ken._id, name: 'ken' }, { likes: ['kicking', 'punching'] }, function (err) {
+          should.exist(err);
+
+          P.update({ _id: ken._id, age: 27 }, { likes: ['kicking', 'punching'] }, function (err) {
+            should.exist(err);
+
+            P.update({ age: 27 }, { likes: ['kicking', 'punching'] }, function (err) {
+              db.close();
+              should.exist(err);
+            });
+          });
+        });
+      });
+    });
+  },
+
+  'updating shard key values fails': function () {
+    var db = start({ uri:  uri })
+    var P = db.model('ShardPerson', collection);
+    P.create({ name: 'chun li', age: 19, likes: ['street fighting']}, function (err, chunli) {
+      should.strictEqual(null, err);
+
+      chunli._shardval.name.should.equal('chun li');
+      chunli._shardval.age.should.equal(19);
+
+      chunli.age = 20;
+      chunli.save(function (err) {
+        /^Can't modify shard key's value field/.test(err.message).should.be.true;
+
+        chunli._shardval.name.should.equal('chun li');
+        chunli._shardval.age.should.equal(19);
+
+        P.findById(chunli._id, function (err, chunli) {
+          should.strictEqual(null, err);
+
+          chunli._shardval.name.should.equal('chun li');
+          chunli._shardval.age.should.equal(19);
+
+          chunli.name='chuuuun liiiii';
+          chunli.save(function (err) {
+            db.close();
+            /^Can't modify shard key's value field/.test(err.message).should.be.true;
+          });
+        });
+      });
+    });
+  }
+}