more flexible google mock

Author: rrrooommmaaa

test/source/mock/google/google-data.ts

@@ -11,7 +11,7 @@ type GmailMsg$payload$part = { partId?: string, body?: GmailMsg$payload$body, fi
 type GmailMsg$payload = { partId?: string, filename?: string, parts?: GmailMsg$payload$part[], headers?: GmailMsg$header[], mimeType?: string, body?: GmailMsg$payload$body };
 type GmailMsg$labelId = 'INBOX' | 'UNREAD' | 'CATEGORY_PERSONAL' | 'IMPORTANT' | 'SENT' | 'CATEGORY_UPDATES' | 'DRAFT';
 type GmailThread = { historyId: string; id: string; snippet: string; };
-type Label = { id: string, name: "CATEGORY_SOCIAL", messageListVisibility: "hide", labelListVisibility: "labelHide", type: 'system' };
+type Label = { id: string, name: string, messageListVisibility: 'show' | 'hide', labelListVisibility: 'labelShow' | 'labelHide', type: 'system' };
 type AcctDataFile = { messages: GmailMsg[]; drafts: GmailMsg[], attachments: { [id: string]: { data: string, size: number, filename?: string } }, labels: Label[] };
 type ExportedMsg = { acctEmail: string, full: GmailMsg, raw: GmailMsg, attachments: { [id: string]: { data: string, size: number } } };
 
@@ -111,7 +111,13 @@ export class GoogleData {
 
   public static withInitializedData = async (acct: string): Promise<GoogleData> => {
     if (typeof DATA[acct] === 'undefined') {
-      const acctData: AcctDataFile = { drafts: [], messages: [], attachments: {}, labels: [] };
+      const acctData: AcctDataFile = {
+        drafts: [], messages: [], attachments: {}, labels:
+          [
+            { id: 'INBOX', name: 'Inbox', messageListVisibility: 'show', labelListVisibility: 'labelShow', type: 'system' },
+            { id: 'DRAFT', name: 'Drafts', messageListVisibility: 'show', labelListVisibility: 'labelShow', type: 'system' }
+          ]
+      };
       const dir = GoogleData.exportedMsgsPath;
       const filenames: string[] = await new Promise((res, rej) => readdir(dir, (e, f) => e ? rej(e) : res(f)));
       const filePromises = filenames.map(f => new Promise((res, rej) => readFile(dir + f, (e, d) => e ? rej(e) : res(d))));
@@ -204,11 +210,11 @@ export class GoogleData {
   }
 
   public getMessage = (id: string): GmailMsg | undefined => {
-    return DATA[this.acct].messages.find(m => m.id === id);
+    return this.getMessages().find(m => m.id === id);
   }
 
   public getMessageBySubject = (subject: string): GmailMsg | undefined => {
-    return DATA[this.acct].messages.find(m => {
+    return this.getMessages().find(m => {
       if (m.payload?.headers) {
         const subjectHeader = m.payload.headers.find(x => x.name === 'Subject');
         if (subjectHeader) {
@@ -220,7 +226,7 @@ export class GoogleData {
   }
 
   public getMessagesByThread = (threadId: string) => {
-    return DATA[this.acct].messages.filter(m => m.threadId === threadId);
+    return this.getMessages().filter(m => m.threadId === threadId);
   }
 
   public searchMessages = (q: string) => {
@@ -266,26 +272,33 @@ export class GoogleData {
     return DATA[this.acct].labels;
   }
 
-  public getThreads = () => {
+  public getThreads = (labelIds: string[]) => {
     const threads: GmailThread[] = [];
-    for (const thread of DATA[this.acct].messages.map(m => ({ historyId: m.historyId, id: m.threadId!, snippet: `MOCK SNIPPET: ${GoogleData.msgSubject(m)}` }))) {
+    for (const thread of this.getMessages().
+      filter(m => labelIds ? (m.labelIds || []).some(l => labelIds.includes(l)) : true).
+      map(m => ({ historyId: m.historyId, id: m.threadId!, snippet: `MOCK SNIPPET: ${GoogleData.msgSubject(m)}` }))) {
       if (thread.id && !threads.map(t => t.id).includes(thread.id)) {
         threads.push(thread);
       }
     }
     return threads;
   }
 
+  // returns ordinary messages and drafts
+  private getMessages = () => {
+    return DATA[this.acct].messages.concat(DATA[this.acct].drafts);
+  }
+
   private searchMessagesBySubject = (subject: string) => {
     subject = subject.trim().toLowerCase();
-    const messages = DATA[this.acct].messages.filter(m => GoogleData.msgSubject(m).toLowerCase().includes(subject));
+    const messages = this.getMessages().filter(m => GoogleData.msgSubject(m).toLowerCase().includes(subject));
     return messages;
   }
 
   private searchMessagesByPeople = (includePeople: string[], excludePeople: string[]) => {
     includePeople = includePeople.map(person => person.trim().toLowerCase());
     excludePeople = excludePeople.map(person => person.trim().toLowerCase());
-    return DATA[this.acct].messages.filter(m => {
+    return this.getMessages().filter(m => {
       const msgPeople = GoogleData.msgPeople(m).toLowerCase();
       let shouldInclude = false;
       let shouldExclude = false;

test/source/mock/google/google-endpoints.ts

@@ -173,10 +173,10 @@ export const mockGoogleEndpoints: HandlersDefinition = {
     }
     throw new HttpClientErr(`Method not implemented for ${req.url}: ${req.method}`);
   },
-  '/gmail/v1/users/me/threads': async ({ }, req) => {
+  '/gmail/v1/users/me/threads': async (parsedReq, req) => {
     const acct = oauth.checkAuthorizationHeaderWithAccessToken(req.headers.authorization);
     if (isGet(req)) {
-      const threads = (await GoogleData.withInitializedData(acct)).getThreads();
+      const threads = (await GoogleData.withInitializedData(acct)).getThreads([parsedReq.query.labelIds]); // todo: support arrays?
       return { threads, resultSizeEstimate: threads.length };
     }
     throw new HttpClientErr(`Method not implemented for ${req.url}: ${req.method}`);
@@ -218,7 +218,7 @@ export const mockGoogleEndpoints: HandlersDefinition = {
       const acct = oauth.checkAuthorizationHeaderWithAccessToken(req.headers.authorization);
       const body = parsedReq.body as DraftSaveModel;
       if (body && body.message && body.message.raw && typeof body.message.raw === 'string') {
-        if (body.message.threadId && !(await GoogleData.withInitializedData(acct)).getThreads().find(t => t.id === body.message.threadId)) {
+        if (body.message.threadId && !(await GoogleData.withInitializedData(acct)).getThreads(['DRAFT']).find(t => t.id === body.message.threadId)) {
           throw new HttpClientErr('The thread you are replying to not found', 404);
         }
         const decoded = await Parse.convertBase64ToMimeMsg(body.message.raw);