fix collection name detection

lib/model.js

@@ -31,6 +31,10 @@ function Model(fields) {
 	this.set(initialFields);
 }
 
+Model.prototype.collection = function () {
+	return pluralize(this.constructor.displayName || this.constructor.name).toLowerCase();
+};
+
 Model.prototype.getConnection = function () {
 	return this.constructor.getConnection();
 };
@@ -117,10 +121,6 @@ Model.prototype.increment = function (key, value = 1) {
 	return this.store.dispatch(increment(fields));
 };
 
-Model.collection = function () {
-	return pluralize(this.displayName || this.name).toLowerCase();
-};
-
 Model.getConnection = function () {
 	if (!this.database) {
 		return Promise.reject(new Error('Model is not registered in a database.'));
@@ -132,7 +132,7 @@ Model.getConnection = function () {
 Model.getCollection = function () {
 	return this.getConnection()
 		.then(db => {
-			const name = result(this, 'collection');
+			const name = result(this.prototype, 'collection');
 
 			return db.collection(name);
 		});

test/models.js

@@ -221,5 +221,23 @@ test('remove all documents', async t => {
 });
 
 test('automatically set collection name', async t => {
-	t.is(Post.collection(), 'posts');
+	t.is(Post.prototype.collection(), 'posts');
+});
+
+test('override collection name', async t => {
+	const {db} = t.context;
+
+	class CustomPost extends mongorito.Model {
+		collection() {
+			return 'awesome_posts';
+		}
+	}
+
+	db.register(CustomPost);
+
+	const post = new CustomPost({title: 'Greatness'});
+	await post.save();
+
+	const collections = await db._connection.collections();
+	t.deepEqual(collections.map(c => c.collectionName).sort(), ['comments', 'posts', 'awesome_posts'].sort());
 });