EPMRPP-89644 || Implement attachments support

src/index.ts

@@ -16,7 +16,8 @@
  */
 
 import { RPReporter } from './reporter';
+import { ReportingApi } from './reportingApi';
+import { LOG_LEVELS, STATUSES } from './constants';
 
-export { RPReporter };
-
+export { RPReporter, ReportingApi, LOG_LEVELS, STATUSES };
 export default RPReporter;

src/models/index.ts

@@ -21,6 +21,8 @@ import {
   FinishTestItemObjType,
   LogRQ,
   Attachment,
+  RPTaskMeta,
+  ReportingApi,
 } from './reporting';
 import { ReportPortalConfig } from './configs';
 import { Attribute } from './common';
@@ -32,5 +34,7 @@ export {
   ReportPortalConfig,
   Attachment,
   Attribute,
+  RPTaskMeta,
+  ReportingApi,
   LogRQ,
 };

src/models/reporting.ts

@@ -14,7 +14,7 @@
  *  limitations under the License.
  *
  */
-
+import { Task, TaskMeta } from "vitest";
 import { Attribute, Issue } from './common';
 import { TEST_ITEM_TYPES, LOG_LEVELS, LAUNCH_MODES } from '../constants';
 
@@ -61,3 +61,13 @@ export interface LogRQ {
   time?: number;
   file?: Attachment;
 }
+
+export interface RPTaskMeta extends TaskMeta {
+  test: {
+    logs: LogRQ[];
+  }
+}
+
+export interface ReportingApi {
+  attachment: (context: Task, data: Attachment) => void;
+}

src/reporter.ts

@@ -25,6 +25,7 @@ import {
   ReportPortalConfig,
   StartLaunchObjType,
   StartTestObjType,
+  RPTaskMeta,
 } from './models';
 import {
   getAgentInfo,
@@ -33,6 +34,7 @@ import {
   getCodeRef,
   getBasePath,
   isErrorLog,
+  isRPTaskMeta,
 } from './utils';
 import {
   LAUNCH_MODES,
@@ -165,13 +167,19 @@ export class RPReporter implements Reporter {
     const packsReversed = [...packs];
     packsReversed.reverse();
 
-    for (const [id, taskResult] of packsReversed) {
+    for (const [id, taskResult, meta] of packsReversed) {
       const testItem = this.testItems.get(id);
       const { id: testItemId, finishSend } = testItem || {};
       if (!testItemId || finishSend || !FINISHED_STATES.includes(taskResult?.state)) {
         continue;
       }
 
+      if (isRPTaskMeta(meta)) {
+        meta.test.logs.forEach(logRq => {
+          this.sendLog(testItemId, logRq);
+        })
+      }
+
       const finishTestItemObj = this.getFinishTestItemObj(taskResult);
 
       if (taskResult?.errors?.length) {

src/reportingApi.ts

@@ -0,0 +1,25 @@
+import { Task } from "vitest";
+import * as Models from './models';
+import { isRPTaskMeta } from './utils';
+
+const injectRPTaskMeta = (task: Task) => {
+  if (isRPTaskMeta(task.meta)) {
+    return
+  }
+
+  (task.meta as Models.RPTaskMeta) = {
+    ...task.meta,
+    test: {
+      logs: [],
+    }
+  }
+}
+
+const attachment = (task: Task, data: Models.Attachment) => {
+  injectRPTaskMeta(task);
+  (task.meta as Models.RPTaskMeta).test.logs.push({ file: data, time: Date.now() });
+}
+
+export const ReportingApi: Models.ReportingApi = {
+  attachment
+}

src/types/interfaces.d.ts

@@ -14,6 +14,7 @@
  *  limitations under the License.
  *
  */
+import { Task } from "vitest";
 
 declare namespace Interfaces {
   interface Attribute {
@@ -38,4 +39,8 @@ declare namespace Interfaces {
   interface ObjUniversal {
     [name: string]: string;
   }
+
+  interface ReportingApi {
+    attachment: (task: Task, data: Attachment) => void;
+  }
 }

src/utils.ts

@@ -16,9 +16,10 @@
  */
 
 import { normalize, sep } from 'node:path';
+import { TaskMeta } from 'vitest';
 // @ts-ignore
 import { name as pjsonName, version as pjsonVersion } from '../package.json';
-import { Attribute } from './models';
+import { Attribute, RPTaskMeta } from './models';
 
 export const isFalse = (value: string | boolean | undefined): boolean =>
   [false, 'false'].includes(value);
@@ -63,3 +64,5 @@ export const getCodeRef = (basePath: string, itemTitle: string): string =>
 export const isErrorLog = (message: string): boolean => {
   return message.toLowerCase().includes('error');
 };
+
+export const isRPTaskMeta = (meta: TaskMeta | RPTaskMeta): meta is RPTaskMeta => 'test' in meta;