From 86f919e54a386ce4f0e97a423765867dbd172044 Mon Sep 17 00:00:00 2001
From: SachaG <358832+SachaG@users.noreply.github.com>
Date: Sun, 17 Mar 2019 10:40:13 +0900
Subject: [PATCH] Make queryOne support any selector

---
 packages/vulcan-lib/lib/server/query.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/vulcan-lib/lib/server/query.js b/packages/vulcan-lib/lib/server/query.js
index 15f60eb0e..ae573769b 100644
--- a/packages/vulcan-lib/lib/server/query.js
+++ b/packages/vulcan-lib/lib/server/query.js
@@ -93,9 +93,10 @@ Meteor.startup(() => {
   Collections.forEach(collection => {
     const typeName = collection.options.typeName;
 
-    collection.queryOne = async (documentId, { fragmentName, fragmentText, context }) => {
+    collection.queryOne = async (documentIdOrSelector, { fragmentName, fragmentText, context }) => {
+      const selector = typeof documentIdOrSelector === 'string' ? { documentId: documentIdOrSelector } : documentIdOrSelector;
       const query = buildQuery(collection, { fragmentName, fragmentText });
-      const result = await runQuery(query, { input: { selector: { documentId } } }, context);
+      const result = await runQuery(query, { input: { selector } }, context);
       return result.data[Utils.camelCaseify(typeName)].result;
     };
   });