From 5a478ec127cb6e72006cac6319fae7d3d3b3278e Mon Sep 17 00:00:00 2001
From: Tom Moor <tom.moor@gmail.com>
Date: Tue, 9 Feb 2021 21:29:24 -0800
Subject: [PATCH] fix: Incorrect policy returned after document create/import

---
 server/api/documents.js      | 2 ++
 server/api/documents.test.js | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/server/api/documents.js b/server/api/documents.js
index 778677005..b8d5f3f66 100644
--- a/server/api/documents.js
+++ b/server/api/documents.js
@@ -1205,6 +1205,7 @@ router.post("documents.import", auth(), async (ctx) => {
     user,
     ip: ctx.request.ip,
   });
+  document.collection = collection;
 
   return (ctx.body = {
     data: await presentDocument(document),
@@ -1275,6 +1276,7 @@ router.post("documents.create", auth(), async (ctx) => {
     editorVersion,
     ip: ctx.request.ip,
   });
+  document.collection = collection;
 
   return (ctx.body = {
     data: await presentDocument(document),
diff --git a/server/api/documents.test.js b/server/api/documents.test.js
index f314680f5..9b8dacf65 100644
--- a/server/api/documents.test.js
+++ b/server/api/documents.test.js
@@ -1656,6 +1656,7 @@ describe("#documents.create", () => {
     expect(res.status).toEqual(200);
     expect(newDocument.parentDocumentId).toBe(null);
     expect(newDocument.collectionId).toBe(collection.id);
+    expect(body.policies[0].abilities.update).toEqual(true);
   });
 
   it("should not allow very long titles", async () => {
@@ -1688,6 +1689,7 @@ describe("#documents.create", () => {
 
     expect(res.status).toEqual(200);
     expect(body.data.title).toBe("new document");
+    expect(body.policies[0].abilities.update).toEqual(true);
   });
 
   it("should error with invalid parentDocument", async () => {
@@ -1722,6 +1724,7 @@ describe("#documents.create", () => {
 
     expect(res.status).toEqual(200);
     expect(body.data.title).toBe("new document");
+    expect(body.policies[0].abilities.update).toEqual(true);
   });
 });