update unit test

Author: gijoe0295

tests/unit/ReportPrimaryActionUtilsTest.ts

@@ -25,6 +25,7 @@ const PERSONAL_DETAILS = {
 const REPORT_ID = 1;
 const CHAT_REPORT_ID = 2;
 const POLICY_ID = 3;
+const INVOICE_SENDER_ACCOUNT_ID = 4;
 
 // This keeps the error "@rnmapbox/maps native code not available." from causing the tests to fail
 jest.mock('@components/ConfirmedRoute.tsx');
@@ -126,23 +127,33 @@ describe('getPrimaryAction', () => {
         expect(getReportPrimaryAction({report, chatReport, reportTransactions: [transaction], violations: {}, policy: policy as Policy})).toBe('');
     });
 
-    it('should return PAY for submitted invoice report', async () => {
+    it('should return PAY for submitted invoice report if paid as personal', async () => {
         const report = {
             reportID: REPORT_ID,
             type: CONST.REPORT.TYPE.INVOICE,
-            ownerAccountID: CURRENT_USER_ACCOUNT_ID,
+            ownerAccountID: INVOICE_SENDER_ACCOUNT_ID,
+            parentReportID: CHAT_REPORT_ID,
             statusNum: CONST.REPORT.STATUS_NUM.SUBMITTED,
             stateNum: CONST.REPORT.STATE_NUM.SUBMITTED,
         } as unknown as Report;
+        const parentReport = {
+            reportID: CHAT_REPORT_ID,
+            invoiceReceiver: {
+                type: CONST.REPORT.INVOICE_RECEIVER_TYPE.INDIVIDUAL,
+                accountID: CURRENT_USER_ACCOUNT_ID,
+            },
+        } as unknown as Report;
         await Onyx.merge(`${ONYXKEYS.COLLECTION.REPORT}${REPORT_ID}`, report);
-        const policy = {
+        await Onyx.merge(`${ONYXKEYS.COLLECTION.REPORT}${CHAT_REPORT_ID}`, parentReport);
+        const policy = {} as Policy;
+        const invoiceReceiverPolicy = {
             role: CONST.POLICY.ROLE.ADMIN,
-        };
+        } as Policy;
         const transaction = {
             reportID: `${REPORT_ID}`,
         } as unknown as Transaction;
 
-        expect(getReportPrimaryAction({report, chatReport, reportTransactions: [transaction], violations: {}, policy: policy as Policy})).toBe(CONST.REPORT.PRIMARY_ACTIONS.PAY);
+        expect(getReportPrimaryAction({report, chatReport, reportTransactions: [transaction], violations: {}, policy, invoiceReceiverPolicy})).toBe(CONST.REPORT.PRIMARY_ACTIONS.PAY);
     });
 
     it('should return PAY for expense report with payments enabled', async () => {
@@ -560,12 +571,14 @@ describe('getTransactionThreadPrimaryAction', () => {
         ).toBe('');
     });
 
-    it('should return PAY if paid as business and the payer is the policy admin', async () => {
+    it('should return PAY for submitted invoice report if paid as business and the payer is the policy admin', async () => {
         const report = {
             reportID: REPORT_ID,
             type: CONST.REPORT.TYPE.INVOICE,
-            ownerAccountID: CURRENT_USER_ACCOUNT_ID,
+            ownerAccountID: INVOICE_SENDER_ACCOUNT_ID,
             parentReportID: CHAT_REPORT_ID,
+            statusNum: CONST.REPORT.STATUS_NUM.SUBMITTED,
+            stateNum: CONST.REPORT.STATE_NUM.SUBMITTED,
         } as unknown as Report;
         const parentReport = {
             reportID: CHAT_REPORT_ID,